Update app.py

app.py

@@ -1,42 +1,41 @@
 import os
 import shutil
 import zipfile
+from huggingface_hub import snapshot_download
 import gradio as gr
 
-def clone_and_zip_with_git(repo_url):
-    # Directory and file settings
-    clone_dir = "cloned_repo"
+def download_huggingface_repo(repo_url):
+    # Extract the repo ID from the URL
+    repo_id = repo_url.replace("https://huggingface.co/", "").strip()
+    
+    output_dir = "repo"
     zip_file = "repo.zip"
-
-    # Clean up old directories and files if they exist
-    if os.path.exists(clone_dir):
-        shutil.rmtree(clone_dir)
+    
+    # Clean up previous files
+    if os.path.exists(output_dir):
+        shutil.rmtree(output_dir)
     if os.path.exists(zip_file):
         os.remove(zip_file)
-
+    
     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
+        # Download the repository using Hugging Face Hub
+        snapshot_download(repo_id, local_dir=output_dir)
+        
+        # Create a zip file of the downloaded repository
         with zipfile.ZipFile(zip_file, "w", zipfile.ZIP_DEFLATED) as zipf:
-            for root, _, files in os.walk(clone_dir):
+            for root, _, files in os.walk(output_dir):
                 for file in files:
                     file_path = os.path.join(root, file)
-                    arcname = os.path.relpath(file_path, clone_dir)
+                    arcname = os.path.relpath(file_path, output_dir)
                     zipf.write(file_path, arcname)
-
-        return zip_file, "Repository cloned and zipped successfully."
+        
+        return zip_file, "Repository downloaded and zipped successfully."
     except Exception as e:
         return None, f"Error: {str(e)}"
 
 # Gradio interface
 def handle_download(repo_url):
-    file_path, message = clone_and_zip_with_git(repo_url)
+    file_path, message = download_huggingface_repo(repo_url)
     return file_path, message
 
 with gr.Blocks() as demo: