Update app.py

app.py

@@ -44,7 +44,7 @@ def handle_token_change(token):
         }
         return (token, *updates.values())
 
-def list_repos(token, author, repo_type):
+def list_repos_backend(token, author, repo_type):
     """Backend function to fetch repository IDs."""
     if not author:
         gr.Info("Please enter an author (username or organization).")
@@ -62,20 +62,19 @@ def list_repos(token, author, repo_type):
 
 def list_repos_for_management(token, author, repo_type):
     """Gradio wrapper to update the management dropdown and reset the UI."""
-    repo_ids = list_repos(token, author, repo_type)
+    repo_ids = list_repos_backend(token, author, repo_type)
     return (
         repo_type,
-        gr.update(choices=repo_ids, value=None), # Update dropdown choices
-        gr.update(visible=False), # Hide action panel
-        gr.update(visible=False)  # Hide editor panel
+        gr.update(choices=repo_ids, value=None),
+        gr.update(visible=False),
+        gr.update(visible=False)
     )
 
 def list_repos_for_download(token, author, repo_type):
     """Gradio wrapper to update the download dropdown."""
-    repo_ids = list_repos(token, author, repo_type)
+    repo_ids = list_repos_backend(token, author, repo_type)
     return repo_type, gr.update(choices=repo_ids, value=None)
 
-
 def on_manage_repo_select(repo_id):
     """Shows action buttons when a repo is selected in the Manage tab."""
     return gr.update(visible=bool(repo_id))
@@ -100,7 +99,7 @@ def show_files_and_load_first(token, repo_id, repo_type):
     """Lists files in a dropdown and pre-loads the first file's content."""
     if not repo_id:
         gr.Warning("No repository selected.")
-        return gr.update(visible=False), gr.update(), gr.update()
+        return gr.update(visible=False), gr.update(choices=[], value=None), gr.update(value="")
     try:
         api = get_hf_api(token)
         repo_files = api.list_repo_files(repo_id=repo_id, repo_type=repo_type)
@@ -117,7 +116,7 @@ def show_files_and_load_first(token, repo_id, repo_type):
                 gr.update(value=content, language=language))
     except Exception as e:
         gr.Error(f"Could not manage files: {e}")
-        return gr.update(visible=False), gr.update(), gr.update()
+        return gr.update(visible=False), gr.update(choices=[], value=None), gr.update(value="")
 
 def load_file_content_backend(token, repo_id, repo_type, filepath):
     """Backend logic to fetch and format file content."""
@@ -156,23 +155,26 @@ def commit_file(token, repo_id, repo_type, filepath, content, commit_message):
 def download_repos_as_zip(token, selected_repo_ids, repo_type, progress=gr.Progress()):
     """Downloads selected repos and zips them."""
     if not selected_repo_ids:
-        gr.Warning("No repositories selected for download."); return
+        gr.Warning("No repositories selected for download."); return gr.update(value=None, visible=False)
     if not repo_type:
-        gr.Warning("Please list a repository type (Spaces, etc.) before downloading."); return
+        gr.Warning("Please list a repository type (Spaces, etc.) before downloading."); return gr.update(value=None, visible=False)
 
     download_root_dir = tempfile.mkdtemp()
     try:
+        total_repos = len(selected_repo_ids)
         for i, repo_id in enumerate(selected_repo_ids):
-            progress((i) / len(selected_repo_ids), desc=f"Downloading {repo_id}")
+            progress((i) / total_repos, desc=f"Downloading {repo_id} ({i+1}/{total_repos})")
             try:
-                snapshot_download(repo_id=repo_id, repo_type=repo_type, local_dir=os.path.join(download_root_dir, repo_id.replace("/", "__")),
+                folder_name = repo_id.replace("/", "__")
+                snapshot_download(repo_id=repo_id, repo_type=repo_type, local_dir=os.path.join(download_root_dir, folder_name),
                                   token=token, local_dir_use_symlinks=False, resume_download=True)
             except Exception as e: gr.Error(f"Failed to download {repo_id}: {e}")
         
-        progress(0.95, desc="Creating ZIP file...")
-        zip_base_name = os.path.join(tempfile.gettempdir(), f"hf_{repo_type}s_{uuid.uuid4().hex}")
+        progress(0.95, desc="All items downloaded. Creating ZIP file...")
+        zip_base_name = os.path.join(tempfile.gettempdir(), f"hf_{repo_type}s_archive_{uuid.uuid4().hex}")
         zip_path = shutil.make_archive(zip_base_name, 'zip', download_root_dir)
         progress(1, desc="Download ready!")
+        gr.Info("ZIP file created successfully!")
         return gr.update(value=zip_path, visible=True)
     finally:
         shutil.rmtree(download_root_dir, ignore_errors=True)
@@ -191,18 +193,14 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Hugging Face Hub
 
     with gr.Tabs():
         with gr.TabItem("Manage Repositories"):
+            # Define all components first
             with gr.Row():
                 with gr.Column(scale=1):
                     gr.Markdown("### 1. Select a Repository")
                     author_input = gr.Textbox(label="Author (Username or Org)")
                     with gr.Row():
-                        for repo_type, label in [("space", "Spaces"), ("model", "Models"), ("dataset", "Datasets")]:
-                            btn = gr.Button(f"List {label}")
-                            btn.click(fn=list_repos_for_management,
-                                      inputs=[hf_token_state, author_input, gr.State(repo_type)],
-                                      outputs=[manage_repo_type_state, gr.Dropdown(), action_panel, editor_panel])
+                        manage_buttons = [gr.Button(f"List {label}") for label in ["Spaces", "Models", "Datasets"]]
                     manage_repo_dropdown = gr.Dropdown(label="Select a Repository", interactive=True)
-
                 with gr.Column(scale=2):
                     with gr.Column(visible=False) as action_panel:
                         gr.Markdown("### 2. Choose an Action")
@@ -214,22 +212,29 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Hugging Face Hub
                         code_editor = gr.Code(label="File Content", interactive=True)
                         commit_message_input = gr.Textbox(label="Commit Message", placeholder="e.g., Update README.md")
                         commit_btn = gr.Button("Commit Changes", variant="primary", interactive=False)
+            # Now, attach event handlers
+            repo_types = ["space", "model", "dataset"]
+            for i, btn in enumerate(manage_buttons):
+                btn.click(fn=list_repos_for_management,
+                          inputs=[hf_token_state, author_input, gr.State(repo_types[i])],
+                          outputs=[manage_repo_type_state, manage_repo_dropdown, action_panel, editor_panel])
 
         with gr.TabItem("Bulk Download (ZIP)"):
+            # Define all components first
             gr.Markdown("## Download Multiple Repositories as a ZIP")
+            download_author_input = gr.Textbox(label="Author (Username or Org)")
             with gr.Row():
-                download_author_input = gr.Textbox(label="Author (Username or Org)")
-            with gr.Row():
-                for repo_type, label in [("space", "Spaces"), ("model", "Models"), ("dataset", "Datasets")]:
-                    btn = gr.Button(f"List {label}")
-                    btn.click(fn=list_repos_for_download,
-                              inputs=[hf_token_state, download_author_input, gr.State(repo_type)],
-                              outputs=[download_repo_type_state, gr.Dropdown()])
+                download_buttons = [gr.Button(f"List {label}") for label in ["Spaces", "Models", "Datasets"]]
             download_repo_dropdown = gr.Dropdown(label="Select Repositories", multiselect=True, interactive=True)
             download_btn = gr.Button("Download Selected as ZIP", variant="primary")
             download_output_file = gr.File(label="Your Downloaded ZIP File", visible=False)
+            # Now, attach event handlers
+            for i, btn in enumerate(download_buttons):
+                btn.click(fn=list_repos_for_download,
+                          inputs=[hf_token_state, download_author_input, gr.State(repo_types[i])],
+                          outputs=[download_repo_type_state, download_repo_dropdown])
 
-    # --- Event Handlers ---
+    # --- Global and Cross-Tab Event Handlers ---
     hf_token.change(fn=handle_token_change, inputs=hf_token,
                     outputs=[hf_token_state, manage_files_btn, delete_repo_btn, commit_btn, author_input, download_author_input, whoami_output])