Update app.py

app.py

@@ -1,29 +1,42 @@
 import os
-import requests
+import shutil
+import zipfile
 import gradio as gr
 
-def download_and_save_repo(repo_url):
-    if not repo_url.endswith(".git"):
-        return None, "Error: The provided URL does not point to a Git repository."
-
-    zip_url = repo_url.replace(".git", "/archive/refs/heads/main.zip")
+def clone_and_zip_with_git(repo_url):
+    # Directory and file settings
+    clone_dir = "cloned_repo"
     zip_file = "repo.zip"
 
-    try:
-        response = requests.get(zip_url, stream=True)
-        response.raise_for_status()
-
-        with open(zip_file, "wb") as f:
-            for chunk in response.iter_content(chunk_size=1024):
-                f.write(chunk)
+    # 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)
 
-        return zip_file, "Repository downloaded and zipped successfully."
+    try:
+        # Run git clone command
+        os.system(f"git clone {repo_url} {clone_dir}")
+
+        # Check if the clone_dir exists
+        if not os.path.exists(clone_dir):
+            return None, "Error: Cloning the repository failed. Please check the URL."
+
+        # Create a zip file from the cloned directory
+        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, "Repository cloned and zipped successfully."
     except Exception as e:
         return None, f"Error: {str(e)}"
 
 # Gradio interface
 def handle_download(repo_url):
-    file_path, message = download_and_save_repo(repo_url)
+    file_path, message = clone_and_zip_with_git(repo_url)
     return file_path, message
 
 with gr.Blocks() as demo:
@@ -33,7 +46,7 @@ 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("Download Repository as Zip")
+        download_button = gr.Button("Clone and Download")
     
     with gr.Row():
         output_file = gr.File(label="Download Zip File")