Update app.py

app.py

@@ -1,39 +1,33 @@
 import os
-import shutil
-import zipfile
+import requests
 import gradio as gr
-from git import Repo
 
-def clone_and_zip_repo(repo_url):
-    # Directory settings
-    clone_dir = "cloned_repo"
-    zip_file = "repo.zip"
+def download_and_save_repo(repo_url):
+    # 対応するzip URLを生成（GitHubの場合に応用可能）
+    if not repo_url.endswith(".git"):
+        return "Error: The provided URL does not point to a Git repository."
 
-    # 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)
+    # .git を削除してzip形式のURLに変換
+    zip_url = repo_url.replace(".git", "/archive/refs/heads/main.zip")
 
+    zip_file = "repo.zip"
     try:
-        # Clone the repository
-        Repo.clone_from(repo_url, clone_dir)
+        # リポジトリのzipファイルをダウンロード
+        response = requests.get(zip_url, stream=True)
+        response.raise_for_status()  # HTTPエラーをチェック
 
-        # 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)
+        # ダウンロードしたファイルを保存
+        with open(zip_file, "wb") as f:
+            for chunk in response.iter_content(chunk_size=1024):
+                f.write(chunk)
 
         return zip_file
     except Exception as e:
         return f"Error: {str(e)}"
 
 # Gradio interface
-def download_repo(repo_url):
-    result = clone_and_zip_repo(repo_url)
+def handle_download(repo_url):
+    result = download_and_save_repo(repo_url)
     if result.endswith(".zip"):
         return result
     else:
@@ -46,12 +40,12 @@ with gr.Blocks() as demo:
         repo_url_input = gr.Textbox(label="Hugging Face Repo URL", placeholder="Enter the repository URL")
     
     with gr.Row():
-        download_button = gr.Button("Clone and Download")
-
+        download_button = gr.Button("Download Repository as Zip")
+    
     with gr.Row():
         output = gr.File(label="Download Zip File")
-
-    download_button.click(download_repo, inputs=[repo_url_input], outputs=[output])
+    
+    download_button.click(handle_download, inputs=[repo_url_input], outputs=[output])
 
 # Run the Gradio app
 demo.launch()