Update app.py

app.py

@@ -16,7 +16,7 @@ from analyzer import (
 )
 from hf_utils import download_filtered_space_files, search_top_spaces
 from chatbot_page import chat_with_user, extract_keywords_from_conversation
-from repo_explorer import create_repo_explorer_tab, setup_repo_explorer_events
+from repo_explorer import create_repo_explorer_tab, setup_repo_explorer_events, initialize_repo_chatbot
 
 # --- Configuration ---
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -918,8 +918,8 @@ def create_ui() -> gr.Blocks:
                 return gr.update(visible=False), hf_url
             return gr.update(visible=False), ""
 
-        def handle_explore_repo(selected_repo_id: str) -> Tuple[Any, Any, Any]:
-            """Handle navigating to the repo explorer and populate the repo ID."""
+        def handle_explore_repo(selected_repo_id: str) -> Tuple[Any, Any, Any, str, str]:
+            """Handle navigating to the repo explorer and automatically load the repository."""
             logger.info(f"DEBUG: handle_explore_repo called with selected_repo_id: '{selected_repo_id}'")
             
             if selected_repo_id and selected_repo_id.strip() and selected_repo_id.strip() != 'nan':
@@ -927,13 +927,17 @@ def create_ui() -> gr.Blocks:
                 return (
                     gr.update(visible=False),  # close modal
                     gr.update(selected="repo_explorer_tab"),  # switch tab
-                    gr.update(value=clean_repo_id)  # populate repo explorer input
+                    gr.update(value=clean_repo_id),  # populate repo explorer input
+                    clean_repo_id,  # trigger repository loading with the repo ID
+                    "auto_load"  # signal to auto-load the repository
                 )
             else:
                 return (
                     gr.update(visible=False),  # close modal
                     gr.update(selected="repo_explorer_tab"),  # switch tab
-                    gr.update()  # don't change repo explorer input
+                    gr.update(),  # don't change repo explorer input
+                    "",  # no repo ID to load
+                    ""  # no auto-load signal
                 )
 
         def handle_cancel_modal() -> Any:
@@ -1335,7 +1339,9 @@ def create_ui() -> gr.Blocks:
             outputs=[
                 repo_action_modal, 
                 tabs,
-                repo_components["repo_explorer_input"]
+                repo_components["repo_explorer_input"],
+                repo_states["current_repo_id"],  # Set the current repo ID
+                status_box_input  # Use for auto-load signal
             ],
             js="""(repo_id) => { 
                 console.log('DEBUG: Navigate to repo explorer for:', repo_id);
@@ -1343,6 +1349,19 @@ def create_ui() -> gr.Blocks:
                     window.scrollTo({top: 0, behavior: 'smooth'}); 
                 }, 200); 
             }"""
+        ).then(
+            # Auto-load the repository if the signal indicates to do so
+            fn=lambda repo_id, signal: handle_load_repository(repo_id) if signal == "auto_load" and repo_id else ("", ""),
+            inputs=[repo_states["current_repo_id"], status_box_input],
+            outputs=[repo_components["repo_status_display"], repo_states["repo_context_summary"]]
+        ).then(
+            # Initialize the chatbot with welcome message after auto-loading
+            fn=lambda repo_status, repo_id, repo_context, signal: (
+                initialize_repo_chatbot(repo_status, repo_id, repo_context) 
+                if signal == "auto_load" and repo_id else []
+            ),
+            inputs=[repo_components["repo_status_display"], repo_states["current_repo_id"], repo_states["repo_context_summary"], status_box_input],
+            outputs=[repo_components["repo_chatbot"]]
         )
         cancel_modal_btn.click(
             fn=handle_cancel_modal,