Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from huggingface_hub import HfApi, snapshot_download
-from huggingface_hub.utils import HfHubHTTPError, RepositoryNotFoundError
+from huggingface_hub.utils import HfHubHTTPError
 import os
 import uuid
 import shutil
@@ -12,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 ---
+# --- Core Logic Functions ---
 
 def handle_token_change(token):
     """
@@ -20,88 +20,84 @@ def handle_token_change(token):
     and auto-fills the author fields in both tabs.
     """
     if not token:
-        # No token, disable write actions and clear user-specific info
-        update_dict = {
-            manage_files_btn: gr.update(interactive=False),
-            delete_repo_btn: gr.update(interactive=False),
-            commit_btn: gr.update(interactive=False),
-            author_input: gr.update(value=""),
-            download_author_input: gr.update(value=""),
-            whoami_output: gr.update(value=None, visible=False)
+        updates = {
+            manage_files_btn: gr.update(interactive=False), delete_repo_btn: gr.update(interactive=False),
+            commit_btn: gr.update(interactive=False), author_input: gr.update(value=""),
+            download_author_input: gr.update(value=""), whoami_output: gr.update(value=None, visible=False)
         }
-        return (None, *update_dict.values())
-
+        return (None, *updates.values())
     try:
         api = get_hf_api(token)
         user_info = api.whoami()
         username = user_info.get('name')
-        
-        # Token is valid, enable write actions and set author everywhere
-        update_dict = {
-            manage_files_btn: gr.update(interactive=True),
-            delete_repo_btn: gr.update(interactive=True),
-            commit_btn: gr.update(interactive=True),
-            author_input: gr.update(value=username),
-            download_author_input: gr.update(value=username),
-            whoami_output: gr.update(value=user_info, visible=True)
+        updates = {
+            manage_files_btn: gr.update(interactive=True), delete_repo_btn: gr.update(interactive=True),
+            commit_btn: gr.update(interactive=True), author_input: gr.update(value=username),
+            download_author_input: gr.update(value=username), whoami_output: gr.update(value=user_info, visible=True)
         }
-        return (token, *update_dict.values())
-
+        return (token, *updates.values())
     except HfHubHTTPError as e:
         gr.Warning(f"Invalid Token: {e}. You can only perform read-only actions.")
-        update_dict = {
-            manage_files_btn: gr.update(interactive=False),
-            delete_repo_btn: gr.update(interactive=False),
-            commit_btn: gr.update(interactive=False),
-            whoami_output: gr.update(value=None, visible=False)
+        updates = {
+            manage_files_btn: gr.update(interactive=False), delete_repo_btn: gr.update(interactive=False),
+            commit_btn: gr.update(interactive=False), whoami_output: gr.update(value=None, visible=False)
         }
-        return (token, *update_dict.values())
-
+        return (token, *updates.values())
 
 def list_repos(token, author, repo_type):
-    """Lists repositories for a given author and type into a dropdown."""
+    """Backend function to fetch repository IDs."""
     if not author:
-        gr.Info("Please enter an author (username or organization) to list repositories.")
-        return gr.update(choices=[], value=None)
+        gr.Info("Please enter an author (username or organization).")
+        return []
     try:
         api = get_hf_api(token)
         list_fn = getattr(api, f"list_{repo_type}s")
         repos = list_fn(author=author)
         repo_ids = [repo.id for repo in repos]
         gr.Info(f"Found {len(repo_ids)} {repo_type}s for '{author}'.")
-        return gr.update(choices=repo_ids, value=None)
+        return repo_ids
     except HfHubHTTPError as e:
         gr.Error(f"Could not list repositories: {e}")
-        return gr.update(choices=[], value=None)
+        return []
+
+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)
+    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
+    )
+
+def list_repos_for_download(token, author, repo_type):
+    """Gradio wrapper to update the download dropdown."""
+    repo_ids = list_repos(token, author, repo_type)
+    return repo_type, gr.update(choices=repo_ids, value=None)
+
 
 def on_manage_repo_select(repo_id):
-    """Called when a repo is selected in the Manage tab. Makes action buttons visible."""
-    if repo_id:
-        return gr.update(visible=True) # Show actions
-    return gr.update(visible=False) # Hide actions
+    """Shows action buttons when a repo is selected in the Manage tab."""
+    return gr.update(visible=bool(repo_id))
 
 def delete_repo(token, repo_id, repo_type):
     """Deletes the selected repository."""
     if not token:
         gr.Error("A write-enabled Hugging Face token is required to delete a repository.")
-        return repo_id
-    if not repo_id:
-        gr.Warning("No repository selected to delete.")
-        return repo_id
+        return repo_id, gr.update(visible=True), gr.update(visible=False)
     try:
         api = get_hf_api(token)
         api.delete_repo(repo_id=repo_id, repo_type=repo_type)
         gr.Info(f"Successfully deleted '{repo_id}'.")
-        # After deletion, hide panels and return None to clear selection
-        return None
+        return None, gr.update(visible=False), gr.update(visible=False)
     except HfHubHTTPError as e:
         gr.Error(f"Failed to delete repository: {e}")
-        return repo_id # Keep repo selected on failure
+        return repo_id, gr.update(visible=True), gr.update(visible=False)
 
 # --- File Management Functions ---
 
 def show_files_and_load_first(token, repo_id, repo_type):
-    """Lists files and pre-loads the first file's content for editing."""
+    """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()
@@ -111,106 +107,76 @@ def show_files_and_load_first(token, repo_id, repo_type):
         filtered_files = [f for f in repo_files if not f.startswith('.')]
 
         if not filtered_files:
-            return (
-                gr.update(visible=True),
-                gr.update(choices=[], value=None),
-                gr.update(value="## This repository is empty or contains only hidden files.", language='markdown')
-            )
+            return (gr.update(visible=True), gr.update(choices=[], value=None),
+                    gr.update(value="## This repository is empty.", language='markdown'))
         
-        # Load the content of the first file automatically
         first_file_path = filtered_files[0]
         content, language = load_file_content_backend(token, repo_id, repo_type, first_file_path)
 
-        return (
-            gr.update(visible=True),
-            gr.update(choices=filtered_files, value=first_file_path),
-            gr.update(value=content, language=language)
-        )
+        return (gr.update(visible=True), gr.update(choices=filtered_files, value=first_file_path),
+                gr.update(value=content, language=language))
     except Exception as e:
-        gr.Error(f"Could not list files: {e}")
+        gr.Error(f"Could not manage files: {e}")
         return gr.update(visible=False), gr.update(), gr.update()
 
 def load_file_content_backend(token, repo_id, repo_type, filepath):
-    """Backend logic to fetch and format file content. Returns content and language."""
-    if not filepath:
-        return "## Select a file to view its content.", 'markdown'
+    """Backend logic to fetch and format file content."""
+    if not filepath: return "## Select a file to view.", 'markdown'
     try:
         api = get_hf_api(token)
         local_path = api.hf_hub_download(repo_id=repo_id, repo_type=repo_type, filename=filepath, token=token)
-        with open(local_path, 'r', encoding='utf-8') as f:
-            content = f.read()
+        with open(local_path, 'r', encoding='utf-8') as f: content = f.read()
         
-        language = os.path.splitext(filepath)[1].lstrip('.').lower()
-        if language in ['py', 'python']: language = 'python'
-        elif language == 'js': language = 'javascript'
-        elif language == 'md': language = 'markdown'
-        else: language = 'plaintext' # Default
-            
+        ext = os.path.splitext(filepath)[1].lstrip('.').lower()
+        lang_map = {'py': 'python', 'js': 'javascript', 'md': 'markdown'}
+        language = lang_map.get(ext, 'plaintext')
         return content, language
     except Exception as e:
         return f"Error loading file: {e}", 'plaintext'
 
 def load_file_content_for_editor(token, repo_id, repo_type, filepath):
-    """Gradio wrapper to update the code editor when a new file is selected."""
+    """Gradio wrapper to update the code editor when a file is selected."""
     content, language = load_file_content_backend(token, repo_id, repo_type, filepath)
     return gr.update(value=content, language=language)
 
-
 def commit_file(token, repo_id, repo_type, filepath, content, commit_message):
+    """Commits a file to the repository."""
     if not token: gr.Error("A write-enabled token is required."); return
     if not filepath: gr.Warning("No file selected."); return
     if not commit_message: gr.Warning("Commit message cannot be empty."); return
     try:
         api = get_hf_api(token)
-        api.upload_file(
-            path_or_fileobj=bytes(content, 'utf-8'), path_in_repo=filepath,
-            repo_id=repo_id, repo_type=repo_type, commit_message=commit_message,
-        )
+        api.upload_file(bytes(content, 'utf-8'), path_in_repo=filepath,
+                        repo_id=repo_id, repo_type=repo_type, commit_message=commit_message)
         gr.Info(f"Successfully committed '{filepath}' to '{repo_id}'!")
-    except Exception as e:
-        gr.Error(f"Failed to commit file: {e}")
+    except Exception as e: gr.Error(f"Failed to commit file: {e}")
 
 # --- Download Tab Functions ---
 
 def download_repos_as_zip(token, selected_repo_ids, repo_type, progress=gr.Progress()):
-    """Downloads selected repos of a given type and zips them up."""
+    """Downloads selected repos and zips them."""
     if not selected_repo_ids:
-        gr.Warning("No repositories selected for download.")
-        return gr.update(visible=False, value=None)
+        gr.Warning("No repositories selected for download."); return
     if not repo_type:
-        gr.Warning("Please list a repository type (Spaces, Models, etc.) before downloading.")
-        return gr.update(visible=False, value=None)
+        gr.Warning("Please list a repository type (Spaces, etc.) before downloading."); return
 
     download_root_dir = tempfile.mkdtemp()
     try:
-        total_repos = len(selected_repo_ids)
-        progress(0, desc="Starting download...")
-
         for i, repo_id in enumerate(selected_repo_ids):
-            progress((i) / total_repos, desc=f"Downloading {repo_id} ({i+1}/{total_repos})")
-            folder_name = repo_id.replace("/", "__")
-            target_path = os.path.join(download_root_dir, folder_name)
-
+            progress((i) / len(selected_repo_ids), desc=f"Downloading {repo_id}")
             try:
-                snapshot_download(
-                    repo_id=repo_id, repo_type=repo_type, local_dir=target_path,
-                    token=token, local_dir_use_symlinks=False, resume_download=True,
-                )
-            except Exception as e:
-                gr.Error(f"Failed to download {repo_id}: {e}")
-                continue
-
-        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(base_name=zip_base_name, format='zip', root_dir=download_root_dir)
+                snapshot_download(repo_id=repo_id, repo_type=repo_type, local_dir=os.path.join(download_root_dir, repo_id.replace("/", "__")),
+                                  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}")
+        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)
 
-
 # --- Gradio UI Layout ---
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Hugging Face Hub Toolkit") as demo:
     # State management
@@ -219,119 +185,74 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Hugging Face Hub
     download_repo_type_state = gr.State(None)
 
     gr.Markdown("# Hugging Face Hub Toolkit")
-    gr.Markdown("An intuitive interface to manage and download your Hugging Face repositories.")
-
     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)
+        hf_token = gr.Textbox(label="Hugging Face API Token", type="password", placeholder="hf_...")
+        whoami_output = gr.JSON(label="Authenticated User", visible=False)
 
     with gr.Tabs():
         with gr.TabItem("Manage Repositories"):
             with gr.Row():
                 with gr.Column(scale=1):
-                    gr.Markdown("### 1. Select Author & Repo Type")
-                    author_input = gr.Textbox(label="Author (Username or Org)", interactive=True)
+                    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=lambda rt: (rt, None, gr.update(visible=False), gr.update(visible=False)),
-                                inputs=gr.State(repo_type),
-                                outputs=[manage_repo_type_state, gr.Dropdown(), gr.Column(), gr.Column()] # Reset dropdown and hide panels
-                            ).then(
-                                fn=list_repos,
-                                inputs=[hf_token_state, author_input, gr.State(repo_type)],
-                                outputs=gr.Dropdown() # This is manage_repo_dropdown
-                            )
-
-                    gr.Markdown("### 2. Select Repository")
+                            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_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("### 3. Choose an Action")
-                        with gr.Row():
-                            manage_files_btn = gr.Button("Manage Files", interactive=False)
-                            delete_repo_btn = gr.Button("Delete This Repo", variant="stop", interactive=False)
-                    
+                        gr.Markdown("### 2. Choose an Action")
+                        manage_files_btn = gr.Button("Manage Files", interactive=False)
+                        delete_repo_btn = gr.Button("Delete This Repo", variant="stop", interactive=False)
                     with gr.Column(visible=False) as editor_panel:
-                        gr.Markdown("### 4. Edit Files")
+                        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)
+                        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)
 
         with gr.TabItem("Bulk Download (ZIP)"):
-            gr.Markdown("## Bulk Download Repositories as a ZIP Archive")
-            gr.Markdown("Select one or more repositories from an author to download them as a single ZIP file.")
-            
+            gr.Markdown("## Download Multiple Repositories as a ZIP")
             with gr.Row():
-                download_author_input = gr.Textbox(label="Author (Username or Org)", interactive=True)
+                download_author_input = gr.Textbox(label="Author (Username or Org)")
             with gr.Row():
-                 for repo_type, label in [("space", "Spaces"), ("model", "Models"), ("dataset", "Datasets")]:
+                for repo_type, label in [("space", "Spaces"), ("model", "Models"), ("dataset", "Datasets")]:
                     btn = gr.Button(f"List {label}")
-                    btn.click(
-                        fn=lambda rt: rt,
-                        inputs=gr.State(repo_type),
-                        outputs=download_repo_type_state
-                    ).then(
-                        fn=list_repos,
-                        inputs=[hf_token_state, download_author_input, gr.State(repo_type)],
-                        outputs=gr.Dropdown() # This is download_repo_dropdown
-                    )
-
-            download_repo_dropdown = gr.Dropdown(label="Available Repositories", info="Select the items you want to download.", multiselect=True, interactive=True)
+                    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_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)
 
     # --- Event Handlers ---
-    
-    # Token Authentication
-    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]
-    )
+    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])
 
-    # --- Manage Tab Handlers ---
-    manage_repo_dropdown.select(
-        fn=on_manage_repo_select,
-        inputs=manage_repo_dropdown,
-        outputs=action_panel
-    )
+    manage_repo_dropdown.select(fn=on_manage_repo_select, inputs=manage_repo_dropdown, outputs=action_panel)
     
-    manage_files_btn.click(
-        fn=show_files_and_load_first,
-        inputs=[hf_token_state, manage_repo_dropdown, manage_repo_type_state],
-        outputs=[editor_panel, file_selector, code_editor]
-    )
+    manage_files_btn.click(fn=show_files_and_load_first,
+                           inputs=[hf_token_state, manage_repo_dropdown, manage_repo_type_state],
+                           outputs=[editor_panel, file_selector, code_editor])
 
-    delete_repo_btn.click(
-        fn=delete_repo,
-        inputs=[hf_token_state, manage_repo_dropdown, manage_repo_type_state],
-        outputs=[manage_repo_dropdown],
-        js="() => confirm('Are you sure you want to permanently delete this repository? This action cannot be undone.')"
-    ).then( # After deletion, hide panels
-        lambda: (gr.update(visible=False), gr.update(visible=False)),
-        outputs=[action_panel, editor_panel]
-    )
-    
-    file_selector.change(
-        fn=load_file_content_for_editor,
-        inputs=[hf_token_state, manage_repo_dropdown, manage_repo_type_state, file_selector],
-        outputs=code_editor
-    )
+    delete_repo_btn.click(fn=delete_repo, inputs=[hf_token_state, manage_repo_dropdown, manage_repo_type_state],
+                          outputs=[manage_repo_dropdown, action_panel, editor_panel],
+                          js="() => confirm('Are you sure you want to permanently delete this repository?')")
 
-    commit_btn.click(
-        fn=commit_file,
-        inputs=[hf_token_state, manage_repo_dropdown, manage_repo_type_state, file_selector, code_editor, commit_message_input]
-    )
+    file_selector.change(fn=load_file_content_for_editor,
+                         inputs=[hf_token_state, manage_repo_dropdown, manage_repo_type_state, file_selector],
+                         outputs=code_editor)
 
-    # --- Download Tab Handlers ---
-    download_btn.click(
-        fn=download_repos_as_zip,
-        inputs=[hf_token_state, download_repo_dropdown, download_repo_type_state],
-        outputs=[download_output_file]
-    )
+    commit_btn.click(fn=commit_file,
+                     inputs=[hf_token_state, manage_repo_dropdown, manage_repo_type_state, file_selector, code_editor, commit_message_input])
+
+    download_btn.click(fn=download_repos_as_zip,
+                       inputs=[hf_token_state, download_repo_dropdown, download_repo_type_state],
+                       outputs=[download_output_file])
 
 if __name__ == "__main__":
     demo.launch(debug=True)