Create app.py

app.py

@@ -0,0 +1,30 @@
+import gradio as gr
+from civitai_downloader import download_file
+
+def download_model(url, token, output_path='/content'):
+    try:
+        # Download the file
+        downloaded_file_path = download_file(url, output_path, token)
+        return downloaded_file_path  # Return the path to the downloaded file
+    except Exception as e:
+        return str(e)  # Return error message if an exception occurs
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Civitai Model Downloader")
+
+    with gr.Row():
+        url_input = gr.Textbox(label="Model URL", placeholder="Enter the Civitai model download URL here")
+        token_input = gr.Textbox(label="Token", placeholder="Enter your Civitai token here", type="password")
+
+    output_path = gr.Textbox(label="Output Path", value="/content", interactive=True)
+    download_button = gr.Button("Download Model")
+    result_file = gr.File(label="Downloaded File", interactive=False)  # For file output
+
+    download_button.click(
+        download_model,
+        inputs=[url_input, token_input, output_path],
+        outputs=result_file  # Update to output the downloaded file
+    )
+
+demo.launch()
+