Update app.py

app.py

@@ -1036,16 +1036,25 @@ def create_ui() -> gr.Blocks:
                 return gr.update(visible=False), hf_url
             return gr.update(visible=False), ""
 
-        def handle_explore_repo(repo_id: str) -> Tuple[Any, Any]:
-            """Handle navigating to the repo explorer."""
+        def handle_explore_repo(repo_id: str) -> Tuple[Any, Any, Any]:
+            """Handle navigating to the repo explorer and populate the repo ID."""
             logger.info(f"DEBUG: handle_explore_repo called with repo_id: '{repo_id}'")
             logger.info(f"DEBUG: repo_id type: {type(repo_id)}")
             logger.info(f"DEBUG: repo_id length: {len(repo_id) if repo_id else 'None'}")
             
-            return (
-                gr.update(visible=False),  # close modal
-                gr.update(selected="repo_explorer_tab")  # switch tab
-            )
+            if repo_id and repo_id.strip() and repo_id.strip() != 'nan':
+                clean_repo_id = repo_id.strip()
+                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
+                )
+            else:
+                return (
+                    gr.update(visible=False),  # close modal
+                    gr.update(selected="repo_explorer_tab"),  # switch tab
+                    gr.update()  # don't change repo explorer input
+                )
 
         def handle_cancel_modal() -> Any:
             """Handle closing the modal."""
@@ -1195,56 +1204,14 @@ def create_ui() -> gr.Blocks:
             inputs=[selected_repo_display],
             outputs=[
                 repo_action_modal, 
-                tabs
+                tabs,
+                repo_components["repo_explorer_input"]
             ],
             js="""(repo_id) => { 
-                console.log('DEBUG: JS called with repo_id:', repo_id);
-                console.log('DEBUG: repo_id type:', typeof repo_id);
-                
-                // Try multiple ways to get the repo_id
-                let actualRepoId = repo_id;
-                
-                // Check if repo_id is valid
-                if (!actualRepoId || actualRepoId === 'None' || actualRepoId === null || actualRepoId === 'null') {
-                    console.log('DEBUG: repo_id is null/undefined, trying alternatives...');
-                    
-                    // Try to get from global variable
-                    if (window.selectedRepoId) {
-                        actualRepoId = window.selectedRepoId;
-                        console.log('DEBUG: Found repo_id in global variable:', actualRepoId);
-                    }
-                    
-                    // Try to get from the modal text input
-                    if (!actualRepoId) {
-                        const modalTextboxes = document.querySelectorAll('label:contains("Selected Repository") + textarea, label:contains("Selected Repository") + input');
-                        console.log('DEBUG: Found modal textboxes:', modalTextboxes.length);
-                        modalTextboxes.forEach((input, index) => {
-                            console.log('DEBUG: Modal input', index, 'value:', input.value);
-                            if (input.value && input.value.trim()) {
-                                actualRepoId = input.value.trim();
-                            }
-                        });
-                    }
-                }
-                
-                console.log('DEBUG: Final repo_id to use:', actualRepoId);
-                
+                console.log('DEBUG: Navigate to repo explorer for:', repo_id);
                 setTimeout(() => { 
                     window.scrollTo({top: 0, behavior: 'smooth'}); 
-                    window.dispatchEvent(new Event('repoExplorerNavigation'));
-                    
-                    // Find and set the repo explorer input
-                    const inputs = document.querySelectorAll('input[placeholder*="microsoft/DialoGPT-medium"], input[placeholder*="Repository ID"], label:contains("Repository ID") + input');
-                    console.log('DEBUG: Found repo explorer inputs:', inputs.length);
-                    inputs.forEach((input, index) => {
-                        console.log('DEBUG: Setting repo explorer input', index, 'to value:', actualRepoId);
-                        if (actualRepoId) {
-                            input.value = actualRepoId;
-                            input.dispatchEvent(new Event('input', { bubbles: true }));
-                            input.dispatchEvent(new Event('change', { bubbles: true }));
-                        }
-                    });
-                }, 300); 
+                }, 200); 
             }"""
         )
         cancel_modal_btn.click(
@@ -1262,44 +1229,14 @@ def create_ui() -> gr.Blocks:
         df_output.select(
             fn=handle_dataframe_select,
             inputs=[df_output],
-            outputs=[selected_repo_display, repo_action_modal, tabs, expanded_content_title, expanded_content_text, text_expansion_modal],
-            js="""(evt, df) => {
-                console.log('DEBUG: Dataframe selection event:', evt);
-                if (evt && evt.index && evt.index.length >= 2) {
-                    const row = evt.index[0];
-                    const col = evt.index[1];
-                    console.log('DEBUG: Selected row', row, 'col', col);
-                    
-                    // Store selected repo_id globally for backup
-                    if (col === 0 && df && df.length > row) {
-                        const repoId = df[row][0];
-                        console.log('DEBUG: Storing global repo_id:', repoId);
-                        window.selectedRepoId = repoId;
-                    }
-                }
-            }"""
+            outputs=[selected_repo_display, repo_action_modal, tabs, expanded_content_title, expanded_content_text, text_expansion_modal]
         )
         
         # Add selection event for top repositories dataframe too
         top_repos_df.select(
             fn=handle_dataframe_select,
             inputs=[top_repos_df],
-            outputs=[selected_repo_display, repo_action_modal, tabs, expanded_content_title, expanded_content_text, text_expansion_modal],
-            js="""(evt, df) => {
-                console.log('DEBUG: Top repos dataframe selection event:', evt);
-                if (evt && evt.index && evt.index.length >= 2) {
-                    const row = evt.index[0];
-                    const col = evt.index[1];
-                    console.log('DEBUG: Selected row', row, 'col', col);
-                    
-                    // Store selected repo_id globally for backup
-                    if (col === 0 && df && df.length > row) {
-                        const repoId = df[row][0];
-                        console.log('DEBUG: Storing global repo_id:', repoId);
-                        window.selectedRepoId = repoId;
-                    }
-                }
-            }"""
+            outputs=[selected_repo_display, repo_action_modal, tabs, expanded_content_title, expanded_content_text, text_expansion_modal]
         )
         
         # Reset button event