Update app.py

app.py

@@ -1,14 +1,17 @@
 import os
+import shutil
+import zipfile
 import gradio as gr
 from huggingface_hub import snapshot_download, model_info, dataset_info, space_info
 from huggingface_hub.utils import RepositoryNotFoundError, HFValidationError
 
-# Allow only safe folders
-BASE_DOWNLOAD_DIR = "/home/user/downloads"
-os.makedirs(BASE_DOWNLOAD_DIR, exist_ok=True)
+# Setup base directory
+BASE_DIR = "/home/user/downloads"
+ZIP_DIR = "/home/user/zips"
+os.makedirs(BASE_DIR, exist_ok=True)
+os.makedirs(ZIP_DIR, exist_ok=True)
 
 def detect_repo_type(repo_id):
-    """Try detecting the repo type automatically."""
     try:
         model_info(repo_id)
         return "model"
@@ -26,55 +29,60 @@ def detect_repo_type(repo_id):
         pass
     return None
 
-def download_hf_repo(repo_id, subfolder):
+def zip_folder(source_folder, output_zip):
+    shutil.make_archive(output_zip.replace('.zip', ''), 'zip', source_folder)
+    return output_zip
+
+def download_hf_repo(repo_id, folder_name):
     repo_id = repo_id.strip()
-    if not repo_id or "/" not in repo_id:
-        return "❌ Please enter a valid Hugging Face repo ID (e.g. `username/repo_name`)"
+    folder_name = folder_name.strip() or repo_id.replace("/", "_")
 
-    target_dir = os.path.join(BASE_DOWNLOAD_DIR, subfolder.strip() or repo_id.replace("/", "_"))
-    os.makedirs(target_dir, exist_ok=True)
+    if not repo_id or "/" not in repo_id:
+        return "❌ Please enter a valid Hugging Face repo ID.", None
 
     repo_type = detect_repo_type(repo_id)
     if not repo_type:
-        return f"❌ Repository '{repo_id}' not found as model, dataset, or space."
+        return f"❌ Repository '{repo_id}' not found.", None
+
+    target_dir = os.path.join(BASE_DIR, folder_name)
+    os.makedirs(target_dir, exist_ok=True)
 
     try:
         log = f"🔍 Detected repo type: {repo_type}\n"
         log += f"⬇️ Downloading to: {target_dir}...\n"
-        output_path = snapshot_download(
+
+        snapshot_download(
             repo_id=repo_id,
             repo_type=repo_type,
             local_dir=target_dir,
             local_dir_use_symlinks=False
         )
-        log += f"✅ Download complete!\n\nSaved at:\n`{output_path}`"
-        return log
+
+        # ZIP the folder
+        zip_path = os.path.join(ZIP_DIR, f"{folder_name}.zip")
+        zip_folder(target_dir, zip_path)
+
+        log += f"✅ Download complete! Repository saved and zipped.\n\n"
+        return log, zip_path
     except Exception as e:
-        return f"❌ Error occurred:\n{str(e)}"
+        return f"❌ Error occurred:\n{str(e)}", None
 
-with gr.Blocks(title="Hugging Face Repo Downloader") as demo:
-    gr.Markdown("## 🤖 Hugging Face Repository Downloader")
-    gr.Markdown("Download entire models, datasets, or spaces from the Hugging Face Hub with automatic type detection.")
+with gr.Blocks(title="HF RepoSnap") as demo:
+    gr.Markdown("## 🤖 HF Downloader")
+    gr.Markdown("Download any Hugging Face model, dataset, or space and export it as a .zip file for local use.")
 
     with gr.Row():
-        repo_id_input = gr.Textbox(
-            label="Hugging Face Repository",
-            placeholder="e.g. NorthSage/Facefusion-NSFW",
-            lines=1
-        )
-        subfolder_input = gr.Textbox(
-            label="Download Folder Name (optional)",
-            placeholder="e.g. facefusion",
-            lines=1
-        )
+        repo_input = gr.Textbox(label="Hugging Face Repository", placeholder="e.g. NorthSage/Facefusion-NSFW")
+        folder_input = gr.Textbox(label="Download Folder Name (optional)", placeholder="e.g. facefusion")
 
-    download_btn = gr.Button("🚀 Start Download")
-    output_log = gr.Textbox(label="Download Log", lines=15, interactive=False)
+    download_btn = gr.Button("🚀 Download & Zip Repository")
+    log_output = gr.Textbox(label="Status Log", lines=12, interactive=False)
+    file_output = gr.File(label="📦 Download Zipped Repository")
 
     download_btn.click(
         fn=download_hf_repo,
-        inputs=[repo_id_input, subfolder_input],
-        outputs=output_log
+        inputs=[repo_input, folder_input],
+        outputs=[log_output, file_output]
     )
 
-demo.launch()
+demo.launch()