Update app.py

app.py

@@ -6,20 +6,30 @@ import gradio as gr
 
 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 previous files
-    if os.path.exists(output_dir):
-        shutil.rmtree(output_dir)
-    if os.path.exists(zip_file):
-        os.remove(zip_file)
-    
     try:
+        if "huggingface.co/" not in repo_url:
+            raise ValueError("Invalid URL: The provided URL does not belong to Hugging Face.")
+        
+        repo_id = repo_url.replace("https://huggingface.co/", "").strip()
+        
+        # Detect repo type (spaces, models, or datasets)
+        if repo_id.startswith("spaces/"):
+            repo_type = "space"
+            repo_id = repo_id.replace("spaces/", "")
+        else:
+            repo_type = None  # Default type (models or datasets)
+
+        output_dir = "repo"
+        zip_file = "repo.zip"
+        
+        # Clean up previous files
+        if os.path.exists(output_dir):
+            shutil.rmtree(output_dir)
+        if os.path.exists(zip_file):
+            os.remove(zip_file)
+        
         # Download the repository using Hugging Face Hub
-        snapshot_download(repo_id, local_dir=output_dir)
+        snapshot_download(repo_id, local_dir=output_dir, repo_type=repo_type)
         
         # Create a zip file of the downloaded repository
         with zipfile.ZipFile(zip_file, "w", zipfile.ZIP_DEFLATED) as zipf: