Update app.py

app.py

@@ -7,36 +7,32 @@ import uuid
 # --- State Management and API Client ---
 
 def get_hf_api(token):
-    if not token:
-        # Allow read-only operations without a token
-        return HfApi()
-    return HfApi(token=token)
+    """Initializes the HfApi client. Allows read-only operations if no token is provided."""
+    return HfApi(token=token if token else None)
 
 # --- UI Functions ---
 
-def handle_token_change(token):
-    """Called when the token is entered. Fetches user info and enables/disables UI elements."""
+def handle_token_change(token, current_author):
+    """
+    Called when the token is entered. Fetches user info and updates UI interactivity.
+    """
     if not token:
-        # No token, so disable write actions and clear username
+        # No token, disable write actions and clear user-specific info
         update_dict = {
-            # In 'Actions' panel
             manage_files_btn: gr.update(interactive=False),
             delete_repo_btn: gr.update(interactive=False),
-            # In 'Editor' panel
             commit_btn: gr.update(interactive=False),
-            # In 'List Repos' panel
-            author_input: gr.update(value=""),
-            # User info output
             whoami_output: gr.update(value=None, visible=False)
         }
-        return (None, "") + (gr.update(),) * (len(update_dict))
+        # Do not clear the author field if the user typed it manually
+        return (None, current_author, *update_dict.values())
 
     try:
         api = get_hf_api(token)
         user_info = api.whoami()
         username = user_info.get('name')
         
-        # Token is valid, enable write actions
+        # Token is valid, enable write actions and set author
         update_dict = {
             manage_files_btn: gr.update(interactive=True),
             delete_repo_btn: gr.update(interactive=True),
@@ -44,7 +40,6 @@ def handle_token_change(token):
             author_input: gr.update(value=username),
             whoami_output: gr.update(value=user_info, visible=True)
         }
-        # The first two return values update gr.State objects
         return (token, username, *update_dict.values())
 
     except HfHubHTTPError as e:
@@ -56,23 +51,25 @@ def handle_token_change(token):
             commit_btn: gr.update(interactive=False),
             whoami_output: gr.update(value=None, visible=False)
         }
-        # Clear username but keep the invalid token for read-only API calls
-        return (token, "", *update_dict.values())
+        # Clear username but keep author field as is
+        return (token, current_author, *update_dict.values())
 
 
 def list_repos(token, author, repo_type):
     """Lists repositories for a given author and type."""
     if not author:
         gr.Info("Please enter an author (username or organization) to list repositories.")
-        return gr.update(choices=[], value=None), gr.update(visible=False)
+        return gr.update(choices=[], value=None), gr.update(visible=False), gr.update(visible=False)
     try:
         api = get_hf_api(token)
-        repos = api.list_repos(author=author, repo_type=repo_type)
+        # Use the dedicated list functions for clarity
+        list_fn = getattr(api, f"list_{repo_type}s")
+        repos = list_fn(author=author)
         repo_ids = [repo.id for repo in repos]
-        return gr.update(choices=repo_ids, value=None), gr.update(visible=False)
+        return gr.update(choices=repo_ids, value=None), gr.update(visible=False), gr.update(visible=False)
     except HfHubHTTPError as e:
         gr.Error(f"Could not list repositories: {e}")
-        return gr.update(choices=[], value=None), gr.update(visible=False)
+        return gr.update(choices=[], value=None), gr.update(visible=False), gr.update(visible=False)
 
 def handle_repo_selection(repo_id):
     """Called when a repo is selected. Makes action buttons visible."""
@@ -84,19 +81,18 @@ 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
+        return repo_id, gr.update(visible=True), gr.update(visible=False)
     if not repo_id:
         gr.Warning("No repository selected to delete.")
-        return
+        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}'. Please re-list repositories.")
-        # Clear selection and hide action/editor panels
+        gr.Info(f"Successfully deleted '{repo_id}'. Listing updated repositories.")
         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, gr.update(visible=True), gr.update(visible=False) # Keep state on failure
+        return repo_id, gr.update(visible=True), gr.update(visible=False)
 
 # --- File Editor Functions ---
 
@@ -104,20 +100,17 @@ def show_file_manager(token, repo_id, repo_type):
     """Lists files in the selected repo and shows the editor panel."""
     if not repo_id:
         gr.Warning("No repository selected.")
-        return gr.update(visible=False)
+        return gr.update(visible=False), gr.update(), gr.update(), gr.update()
     try:
         api = get_hf_api(token)
         repo_files = api.list_repo_files(repo_id=repo_id, repo_type=repo_type)
-        
-        # Don't show .gitattributes or other hidden files by default
         filtered_files = [f for f in repo_files if not f.startswith('.')]
         
-        # Update UI components for the editor
         return (
-            gr.update(visible=True), # Show editor panel
-            gr.update(choices=filtered_files, value=None), # Update file dropdown
-            gr.update(value=f"## Select a file from the dropdown to view or edit its content.", language=None), # Clear code view
-            "" # Clear commit message
+            gr.update(visible=True),
+            gr.update(choices=filtered_files, value=None),
+            gr.update(value="## Select a file to view or edit.", language='markdown'),
+            ""
         )
     except RepositoryNotFoundError:
         gr.Error(f"Repository '{repo_id}' not found. It might be private and require a token.")
@@ -133,22 +126,19 @@ def load_file_content(token, repo_id, repo_type, filepath):
         return gr.update(value="## Select a file to view its content.", language='markdown')
     try:
         api = get_hf_api(token)
-        # Download the file to a temporary local path
         local_path = api.hf_hub_download(
-            repo_id=repo_id,
-            repo_type=repo_type,
-            filename=filepath,
-            token=token,
+            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()
         
-        # Determine language for syntax highlighting
-        language = os.path.splitext(filepath)[1].lstrip('.')
-        if language in ['py', 'js', 'html', 'css', 'json', 'md']:
-             return gr.update(value=content, language=language)
-        else:
-             return gr.update(value=content, language='plaintext')
+        language = os.path.splitext(filepath)[1].lstrip('.').lower()
+        supported_langs = ['python', 'typescript', 'css', 'json', 'markdown', 'html', 'javascript']
+        if language == 'py': language = 'python'
+        if language == 'js': language = 'javascript'
+        if language == 'md': language = 'markdown'
+        
+        return gr.update(value=content, language=language if language in supported_langs else 'plaintext')
 
     except Exception as e:
         gr.Error(f"Could not load file '{filepath}': {e}")
@@ -167,10 +157,8 @@ def commit_file(token, repo_id, repo_type, filepath, content, commit_message):
         return
 
     try:
-        # Write content to a temporary file to upload it
         temp_dir = "hf_temp_files"
         os.makedirs(temp_dir, exist_ok=True)
-        # Use a unique filename to avoid conflicts
         temp_file_path = os.path.join(temp_dir, f"{uuid.uuid4()}_{os.path.basename(filepath)}")
         
         with open(temp_file_path, "w", encoding="utf-8") as f:
@@ -184,22 +172,19 @@ def commit_file(token, repo_id, repo_type, filepath, content, commit_message):
             repo_type=repo_type,
             commit_message=commit_message,
         )
-        
-        # Clean up the temporary file
         os.remove(temp_file_path)
-        
         gr.Info(f"Successfully committed '{filepath}' to '{repo_id}'!")
     except Exception as e:
         gr.Error(f"Failed to commit file: {e}")
 
 
 # --- Gradio UI Layout ---
-with gr.Blocks(theme=gr.themes.Soft(), title="Hugging Face Hub Toolkit") as demo:
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue"), title="Hugging Face Hub Toolkit") as demo:
     # State management
     hf_token_state = gr.State(None)
-    username_state = gr.State("")
+    author_state = gr.State("")
     selected_repo_id = gr.State(None)
-    selected_repo_type = gr.State("space") # Default to spaces
+    selected_repo_type = gr.State("space") # Default
 
     gr.Markdown("# Hugging Face Hub Dashboard")
     gr.Markdown("An intuitive interface to manage your Hugging Face repositories. **Enter a write-token for full access.**")
@@ -207,44 +192,47 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Hugging Face Hub Toolkit") as demo
     with gr.Row():
         hf_token = gr.Textbox(
             label="Hugging Face API Token (write permission recommended)",
-            type="password",
-            placeholder="hf_...",
-            scale=3,
+            type="password", placeholder="hf_...", scale=3,
         )
         whoami_output = gr.JSON(label="Authenticated User", visible=False, scale=1)
 
-    with gr.Row():
+    with gr.Row(equal_height=False):
         # PANEL 1: List and Select Repos
         with gr.Column(scale=1):
             gr.Markdown("### 1. Select a Repository")
-            author_input = gr.Textbox(label="Author (Username or Org)")
+            author_input = gr.Textbox(label="Author (Username or Org)", interactive=True)
             
-            with gr.Tabs() as repo_type_tabs:
-                # This helper function creates the list button and radio selector for a repo type
-                def create_repo_lister(repo_type, label):
-                    with gr.Tab(label, id=repo_type):
-                        gr.Button(f"List {label}").click(
-                            fn=list_repos,
-                            inputs=[hf_token_state, author_input, gr.State(repo_type)],
-                            outputs=[repo_selector, editor_panel]
-                        )
-                
-                create_repo_lister("space", "Spaces")
-                create_repo_lister("model", "Models")
-                create_repo_lister("dataset", "Datasets")
+            repo_selector = gr.Radio(label="Select a Repository", interactive=True, value=None)
+            
+            # This is the corrected function definition. It now takes UI elements as arguments.
+            def create_repo_lister(repo_type, label, repo_selector_el, action_panel_el, editor_panel_el):
+                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_el, action_panel_el, editor_panel_el]
+                    ).then(
+                        # After listing, update the author state
+                        fn=lambda author: author,
+                        inputs=author_input,
+                        outputs=author_state
+                    )
 
-            repo_selector = gr.Radio(label="Select a Repository", interactive=True)
+            with gr.Tabs() as repo_type_tabs:
+                # The action and editor panels are defined later but referenced here.
+                # We will define placeholder variables for them now.
+                action_panel_ref = gr.Column()
+                editor_panel_ref = gr.Column()
 
         # PANEL 2 & 3: Actions and Editor
         with gr.Column(scale=3):
             with gr.Row():
-                # PANEL 2: Action Buttons
                 with gr.Column(scale=1, visible=False) as action_panel:
                     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)
                 
-                # PANEL 3: File Editor
                 with gr.Column(scale=3, visible=False) as editor_panel:
                     gr.Markdown("### 3. Edit Files")
                     file_selector = gr.Dropdown(label="Select File", interactive=True)
@@ -252,41 +240,34 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Hugging Face Hub Toolkit") as demo
                     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 Wiring ---
+    # --- Post-Layout UI Wiring ---
+    # Now that action_panel and editor_panel are fully defined, we can wire them up
+    # in the create_repo_lister function calls within the Tabs context.
+    with repo_type_tabs:
+        create_repo_lister("space", "Spaces", repo_selector, action_panel, editor_panel)
+        create_repo_lister("model", "Models", repo_selector, action_panel, editor_panel)
+        create_repo_lister("dataset", "Datasets", repo_selector, action_panel, editor_panel)
+
+    # --- Event Handlers ---
 
-    # When token changes, update auth state and UI
     hf_token.change(
         fn=handle_token_change,
-        inputs=hf_token,
-        outputs=[
-            hf_token_state,
-            username_state,
-            manage_files_btn,
-            delete_repo_btn,
-            commit_btn,
-            author_input,
-            whoami_output
-        ]
+        inputs=[hf_token, author_state],
+        outputs=[hf_token_state, author_state, manage_files_btn, delete_repo_btn, commit_btn, author_input, whoami_output]
     )
-
-    # When repo type tab is changed, store the new type and clear selection
-    def on_tab_change(repo_type):
-        return repo_type, None, gr.update(choices=[], value=None), gr.update(visible=False), gr.update(visible=False)
-
+    
     repo_type_tabs.select(
-        fn=on_tab_change,
+        fn=lambda rt: (rt, None, gr.update(choices=[], value=None), gr.update(visible=False), gr.update(visible=False)),
         inputs=repo_type_tabs,
         outputs=[selected_repo_type, selected_repo_id, repo_selector, action_panel, editor_panel]
     )
 
-    # When a repo is selected, update state and show the action panel
     repo_selector.select(
         fn=lambda repo_id: (repo_id, *handle_repo_selection(repo_id)),
         inputs=repo_selector,
         outputs=[selected_repo_id, action_panel, editor_panel]
     )
     
-    # Action button clicks
     manage_files_btn.click(
         fn=show_file_manager,
         inputs=[hf_token_state, selected_repo_id, selected_repo_type],
@@ -297,11 +278,14 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Hugging Face Hub Toolkit") as demo
         fn=delete_repo,
         inputs=[hf_token_state, selected_repo_id, selected_repo_type],
         outputs=[selected_repo_id, action_panel, editor_panel],
-        # Add a confirmation popup before deleting
         js="() => confirm('Are you sure you want to permanently delete this repository? This action cannot be undone.')"
+    ).then(
+        # After attempting deletion, refresh the list
+        fn=list_repos,
+        inputs=lambda: (hf_token_state.value, author_state.value, selected_repo_type.value),
+        outputs=[repo_selector, action_panel, editor_panel]
     )
     
-    # Editor interactions
     file_selector.change(
         fn=load_file_content,
         inputs=[hf_token_state, selected_repo_id, selected_repo_type, file_selector],
@@ -315,4 +299,4 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Hugging Face Hub Toolkit") as demo
     )
 
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch()