Update app.py

app.py

@@ -1,8 +1,10 @@
 import gradio as gr
-from huggingface_hub import HfApi
+from huggingface_hub import HfApi, snapshot_download
 from huggingface_hub.utils import HfHubHTTPError, RepositoryNotFoundError
 import os
 import uuid
+import shutil
+import tempfile
 
 # --- State Management and API Client ---
 
@@ -10,7 +12,7 @@ def get_hf_api(token):
     """Initializes the HfApi client. Allows read-only operations if no token is provided."""
     return HfApi(token=token if token else None)
 
-# --- UI Functions ---
+# --- UI Functions (Original Tab) ---
 
 def handle_token_change(token):
     """
@@ -92,7 +94,7 @@ def delete_repo(token, repo_id, repo_type):
         gr.Error(f"Failed to delete repository: {e}")
         return repo_id, gr.update(visible=True), gr.update(visible=False)
 
-# --- File Editor Functions ---
+# --- File Editor Functions (Original Tab) ---
 
 def show_file_manager(token, repo_id, repo_type):
     if not repo_id:
@@ -140,6 +142,86 @@ def commit_file(token, repo_id, repo_type, filepath, content, commit_message):
     except Exception as e:
         gr.Error(f"Failed to commit file: {e}")
 
+# --- Download Tab Functions ---
+
+def list_spaces_for_download(token, author):
+    """Lists spaces for a given author to populate the checkbox group."""
+    if not author:
+        gr.Info("Please enter an author (username or organization) to list spaces.")
+        return gr.update(choices=[], value=None)
+    try:
+        api = get_hf_api(token)
+        spaces = api.list_spaces(author=author)
+        repo_ids = [space.id for space in spaces]
+        if not repo_ids:
+            gr.Warning(f"No Spaces found for author '{author}'.")
+        return gr.update(choices=repo_ids, value=None)
+    except RepositoryNotFoundError:
+        gr.Warning(f"Author '{author}' not found or has no public spaces.")
+        return gr.update(choices=[], value=None)
+    except HfHubHTTPError as e:
+        gr.Error(f"Could not list spaces: {e}")
+        return gr.update(choices=[], value=None)
+
+def download_spaces_as_zip(token, selected_space_ids, progress=gr.Progress()):
+    """Downloads selected spaces and zips them up."""
+    if not selected_space_ids:
+        gr.Warning("No spaces selected for download.")
+        return gr.update(visible=False, value=None)
+
+    # Create a temporary directory for all the downloaded content
+    download_root_dir = tempfile.mkdtemp()
+    
+    try:
+        total_spaces = len(selected_space_ids)
+        progress(0, desc="Starting download...")
+
+        # 1. Download each space into a dedicated subfolder within the temp directory
+        for i, repo_id in enumerate(selected_space_ids):
+            progress((i) / total_spaces, desc=f"Downloading {repo_id} ({i+1}/{total_spaces})")
+            
+            # Sanitize repo_id to create a valid folder name for the zip
+            folder_name = repo_id.replace("/", "__")
+            target_path = os.path.join(download_root_dir, folder_name)
+
+            try:
+                # Use snapshot_download to get the entire repo efficiently
+                snapshot_download(
+                    repo_id=repo_id,
+                    repo_type="space",
+                    local_dir=target_path,
+                    token=token,
+                    local_dir_use_symlinks=False, # Crucial for zipping
+                    resume_download=True,
+                )
+            except Exception as e:
+                # Log the error and skip this repo
+                gr.Error(f"Failed to download {repo_id}: {e}")
+                continue
+
+        # 2. Create the zip archive from the directory of downloaded spaces
+        progress(0.95, desc="All spaces downloaded. Creating ZIP file...")
+        
+        # We create the zip file outside the download dir so we can clean up easily
+        zip_base_name = os.path.join(tempfile.gettempdir(), f"hf_spaces_archive_{uuid.uuid4().hex}")
+        
+        # shutil.make_archive returns the full path to the created archive
+        zip_path = shutil.make_archive(
+            base_name=zip_base_name,
+            format='zip',
+            root_dir=download_root_dir # This becomes the root of the zip
+        )
+        
+        progress(1, desc="Download ready!")
+        gr.Info("ZIP file created successfully!")
+        
+        # Return the path to the zip file and make the component visible
+        return gr.update(value=zip_path, visible=True)
+
+    finally:
+        # 3. Clean up the large download directory, regardless of success or failure
+        shutil.rmtree(download_root_dir, ignore_errors=True)
+
 # --- Gradio UI Layout ---
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Hugging Face Hub Toolkit") as demo:
     # State management
@@ -148,43 +230,74 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Hugging Face Hub
     selected_repo_id = gr.State(None)
     selected_repo_type = gr.State("space") # Default
 
-    gr.Markdown("# Hugging Face Hub Dashboard")
+    gr.Markdown("# Hugging Face Hub Toolkit")
     gr.Markdown("An intuitive interface to manage your Hugging Face repositories. **Enter a write-token for full access.**")
 
     with gr.Sidebar():
         hf_token = gr.Textbox(label="Hugging Face API Token", type="password", placeholder="hf_...", scale=3)
         whoami_output = gr.JSON(label="Authenticated User", visible=False, scale=1)
 
-    with gr.Row(equal_height=False):
-        with gr.Column(scale=1):
-            gr.Markdown("### 1. Select a Repository")
-            author_input = gr.Textbox(label="Author (Username or Org)", interactive=True)
-            repo_selector = gr.Radio(label="Select a Repository", interactive=True, value=None)
-            with gr.Tabs() as repo_type_tabs:
-                for repo_type, label in [("space", "Spaces"), ("model", "Models"), ("dataset", "Datasets")]:
-                    with gr.Tab(label, id=repo_type):
-                        btn = gr.Button(f"List {label}")
-                        btn.click(
-                            fn=list_repos,
-                            inputs=[hf_token_state, author_input, gr.State(repo_type)],
-                            outputs=[repo_selector, gr.Column(), gr.Column()] # Dummy outputs for panels to hide them
-                        ).then(fn=lambda author: author, inputs=author_input, outputs=author_state)
-        
-        with gr.Column(scale=3):
-            with gr.Column(visible=False) as action_panel:
-                gr.Markdown("### 2. Choose an Action")
-                with gr.Row():
-                    manage_files_btn = gr.Button("Manage Files", interactive=False, scale=1)
-                    delete_repo_btn = gr.Button("Delete This Repo", variant="stop", interactive=False, scale=1)
+    with gr.Tabs():
+        with gr.TabItem("Manage Repositories"):
+            with gr.Row(equal_height=False):
+                with gr.Column(scale=1):
+                    gr.Markdown("### 1. Select a Repository")
+                    author_input = gr.Textbox(label="Author (Username or Org)", interactive=True)
+                    repo_selector = gr.Radio(label="Select a Repository", interactive=True, value=None)
+                    with gr.Tabs() as repo_type_tabs:
+                        for repo_type, label in [("space", "Spaces"), ("model", "Models"), ("dataset", "Datasets")]:
+                            with gr.Tab(label, id=repo_type):
+                                btn = gr.Button(f"List {label}")
+                                btn.click(
+                                    fn=list_repos,
+                                    inputs=[hf_token_state, author_input, gr.State(repo_type)],
+                                    outputs=[repo_selector, gr.Column(), gr.Column()] # Dummy outputs for panels to hide them
+                                ).then(fn=lambda author: author, inputs=author_input, outputs=author_state)
+                
+                with gr.Column(scale=3):
+                    with gr.Column(visible=False) as action_panel:
+                        gr.Markdown("### 2. Choose an Action")
+                        with gr.Row():
+                            manage_files_btn = gr.Button("Manage Files", interactive=False, scale=1)
+                            delete_repo_btn = gr.Button("Delete This Repo", variant="stop", interactive=False, scale=1)
+                    
+                    with gr.Column(visible=False) as editor_panel:
+                        gr.Markdown("### 3. Edit Files")
+                        file_selector = gr.Dropdown(label="Select File", interactive=True)
+                        code_editor = gr.Code(label="File Content", language="markdown", interactive=True)
+                        commit_message_input = gr.Textbox(label="Commit Message", placeholder="e.g., Update README.md", interactive=True)
+                        commit_btn = gr.Button("Commit Changes", variant="primary", interactive=False)
+
+        with gr.TabItem("Download Spaces (ZIP)"):
+            gr.Markdown("## Bulk Download Spaces as a ZIP Archive")
+            gr.Markdown("Select one or more Spaces from an author to download them as a single ZIP file. Each Space will be in its own folder inside the archive.")
+            
+            with gr.Row():
+                download_author_input = gr.Textbox(
+                    label="Author (Username or Org)", 
+                    interactive=True, 
+                    placeholder="e.g., huggingface-projects or osanseviero"
+                )
+                list_spaces_btn = gr.Button("List Spaces", variant="secondary")
+
+            spaces_checkboxes = gr.CheckboxGroup(label="Available Spaces", info="Select the spaces you want to download.")
+            download_btn = gr.Button("Download Selected Spaces as ZIP", variant="primary")
+            download_output_file = gr.File(label="Your Downloaded ZIP File", visible=False)
             
-            with gr.Column(visible=False) as editor_panel:
-                gr.Markdown("### 3. Edit Files")
-                file_selector = gr.Dropdown(label="Select File", interactive=True)
-                code_editor = gr.Code(label="File Content", language="markdown", interactive=True)
-                commit_message_input = gr.Textbox(label="Commit Message", placeholder="e.g., Update README.md", interactive=True)
-                commit_btn = gr.Button("Commit Changes", variant="primary", interactive=False)
-
-    # --- Event Handlers ---
+            # --- Event Handlers for Download Tab ---
+            list_spaces_btn.click(
+                fn=list_spaces_for_download,
+                inputs=[hf_token_state, download_author_input],
+                outputs=[spaces_checkboxes]
+            )
+
+            download_btn.click(
+                fn=download_spaces_as_zip,
+                inputs=[hf_token_state, spaces_checkboxes],
+                outputs=[download_output_file]
+            )
+
+    # --- Event Handlers (Original Tab) ---
     hf_token.change(
         fn=handle_token_change, inputs=hf_token,
         outputs=[hf_token_state, author_state, manage_files_btn, delete_repo_btn, commit_btn, author_input, whoami_output]
@@ -207,8 +320,6 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Hugging Face Hub
         outputs=[selected_repo_id, action_panel, editor_panel],
         js="() => confirm('Are you sure you want to permanently delete this repository? This action cannot be undone.')"
     ).then(
-        # On successful deletion, refresh the repo list.
-        # THIS IS THE CORRECTED PART: Pass Gradio components to inputs, not a lambda.
         fn=list_repos,
         inputs=[hf_token_state, author_state, selected_repo_type],
         outputs=[repo_selector, action_panel, editor_panel]