Update app.py

app.py

@@ -553,6 +553,7 @@ def create_ui() -> gr.Blocks:
         user_requirements_state = gr.State("")  # Store user requirements from chatbot
         loaded_repo_content_state = gr.State("")  # Store loaded repository content
         current_repo_id_state = gr.State("")  # Store current repository ID
+        selected_repo_id_state = gr.State("")  # Store selected repository ID for modal actions
 
         gr.Markdown(
             """
@@ -868,14 +869,14 @@ def create_ui() -> gr.Blocks:
             status = "Status: Keywords extracted. User requirements saved for analysis."
             return final_keywords_str, status, user_requirements
 
-        def handle_dataframe_select(evt: gr.SelectData, df_data) -> Tuple[str, Any, Any, str, str, Any]:
+        def handle_dataframe_select(evt: gr.SelectData, df_data) -> Tuple[str, Any, Any, str, str, Any, str]:
             """Handle dataframe row selection - only repo ID (column 0) shows modal since full text is now displayed directly."""
             print(f"DEBUG: Selection event triggered!")
             print(f"DEBUG: evt = {evt}")
             print(f"DEBUG: df_data type = {type(df_data)}")
             
             if evt is None:
-                return "", gr.update(visible=False), gr.update(), "", "", gr.update(visible=False)
+                return "", gr.update(visible=False), gr.update(), "", "", gr.update(visible=False), ""
             
             try:
                 # Get the selected row and column from the event
@@ -893,12 +894,12 @@ def create_ui() -> gr.Blocks:
                         if repo_id and str(repo_id).strip() and str(repo_id).strip() != 'nan':
                             clean_repo_id = str(repo_id).strip()
                             logger.info(f"Showing modal for repository: {clean_repo_id}")
-                            return clean_repo_id, gr.update(visible=True), gr.update(), "", "", gr.update(visible=False)
+                            return clean_repo_id, gr.update(visible=True), gr.update(), "", "", gr.update(visible=False), clean_repo_id
                     
                     # For content columns (1,2,3) and relevance (4), do nothing since full text is shown directly
                     else:
                         print(f"DEBUG: Clicked on column {col_idx}, full text already shown in table")
-                        return "", gr.update(visible=False), gr.update(), "", "", gr.update(visible=False)
+                        return "", gr.update(visible=False), gr.update(), "", "", gr.update(visible=False), ""
                 else:
                     print(f"DEBUG: df_data is not a DataFrame or row_idx {row_idx} out of range")
                 
@@ -906,7 +907,7 @@ def create_ui() -> gr.Blocks:
                 print(f"DEBUG: Exception occurred: {e}")
                 logger.error(f"Error handling dataframe selection: {e}")
             
-            return "", gr.update(visible=False), gr.update(), "", "", gr.update(visible=False)
+            return "", gr.update(visible=False), gr.update(), "", "", gr.update(visible=False), ""
 
         def handle_analyze_all_repos(repo_ids: List[str], user_requirements: str, progress=gr.Progress()) -> Tuple[pd.DataFrame, str, pd.DataFrame, Any]:
             """Analyzes all repositories in the CSV file with progress tracking."""
@@ -1036,14 +1037,14 @@ 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, Any]:
+        def handle_explore_repo(selected_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'}")
+            logger.info(f"DEBUG: handle_explore_repo called with selected_repo_id: '{selected_repo_id}'")
+            logger.info(f"DEBUG: selected_repo_id type: {type(selected_repo_id)}")
+            logger.info(f"DEBUG: selected_repo_id length: {len(selected_repo_id) if selected_repo_id else 'None'}")
             
-            if repo_id and repo_id.strip() and repo_id.strip() != 'nan':
-                clean_repo_id = repo_id.strip()
+            if selected_repo_id and selected_repo_id.strip() and selected_repo_id.strip() != 'nan':
+                clean_repo_id = selected_repo_id.strip()
                 return (
                     gr.update(visible=False),  # close modal
                     gr.update(selected="repo_explorer_tab"),  # switch tab
@@ -1201,7 +1202,7 @@ def create_ui() -> gr.Blocks:
         )
         explore_repo_btn.click(
             fn=handle_explore_repo,
-            inputs=[selected_repo_display],
+            inputs=[selected_repo_id_state],
             outputs=[
                 repo_action_modal, 
                 tabs,
@@ -1229,14 +1230,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]
+            outputs=[selected_repo_display, repo_action_modal, tabs, expanded_content_title, expanded_content_text, text_expansion_modal, selected_repo_id_state]
         )
         
         # 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]
+            outputs=[selected_repo_display, repo_action_modal, tabs, expanded_content_title, expanded_content_text, text_expansion_modal, selected_repo_id_state]
         )
         
         # Reset button event