Update app.py

app.py

@@ -1,43 +1,57 @@
-import gradio as gr
 import os
 import shutil
-import tempfile
+import zipfile
+import gradio as gr
 from git import Repo
 
-def download_and_zip_repo(huggingface_url):
+def clone_and_zip_repo(repo_url):
+    # Directory settings
+    clone_dir = "cloned_repo"
+    zip_file = "repo.zip"
+
+    # Clean up old directories and files if they exist
+    if os.path.exists(clone_dir):
+        shutil.rmtree(clone_dir)
+    if os.path.exists(zip_file):
+        os.remove(zip_file)
+
     try:
-        # 一時ディレクトリを作成
-        with tempfile.TemporaryDirectory() as temp_dir:
-            # リポジトリをクローン
-            Repo.clone_from(huggingface_url, temp_dir)
-
-            # ZIPファイルのパスを指定
-            zip_path = os.path.join(temp_dir, "repository.zip")
-            
-            # クローンしたリポジトリをZIP形式で圧縮
-            shutil.make_archive(zip_path.replace(".zip", ""), 'zip', temp_dir)
-
-            # 圧縮したZIPファイルを返す
-            return zip_path
+        # Clone the repository
+        Repo.clone_from(repo_url, clone_dir)
+
+        # Create a zip file
+        with zipfile.ZipFile(zip_file, 'w', zipfile.ZIP_DEFLATED) as zipf:
+            for root, _, files in os.walk(clone_dir):
+                for file in files:
+                    file_path = os.path.join(root, file)
+                    arcname = os.path.relpath(file_path, clone_dir)
+                    zipf.write(file_path, arcname)
+
+        return zip_file
     except Exception as e:
-        return str(e)
+        return f"Error: {str(e)}"
+
+# Gradio interface
+def download_repo(repo_url):
+    result = clone_and_zip_repo(repo_url)
+    if result.endswith(".zip"):
+        return result
+    else:
+        return result
 
-# Gradioインターフェースの定義
-with gr.Blocks() as app:
-    gr.Markdown("## Hugging Face リポジトリ ダウンローダー")
-    gr.Markdown("Hugging FaceのリポジトリURLを入力して、ZIPファイルをダウンロードできます。")
+with gr.Blocks() as demo:
+    gr.Markdown("## Clone and Download Hugging Face Repository")
     
     with gr.Row():
-        input_url = gr.Textbox(label="Hugging Face リポジトリのURLを入力")
-        download_button = gr.Button("ダウンロード")
+        repo_url_input = gr.Textbox(label="Hugging Face Repo URL", placeholder="Enter the repository URL")
     
-    output_file = gr.File(label="ダウンロードされたZIPファイル")
+    with gr.Row():
+        download_button = gr.Button("Clone and Download")
+
+    with gr.Row():
+        output = gr.File(label="Download Zip File")
 
-    download_button.click(
-        fn=download_and_zip_repo,
-        inputs=input_url,
-        outputs=output_file
-    )
+    download_button.click(download_repo, inputs=[repo_url_input], outputs=[output])
 
-# アプリを起動
-app.launch()
+# Run the Gradio app
+demo.launch()