Update app.py

app.py

@@ -132,55 +132,28 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
         fn_generate_llm_response=generate_llm_response
     )
     
-    # --- MODIFIED: Handler now uses the session cache ---
-    def update_report_and_okr_display(selected_report_id: str, current_token_state: dict, session_cache: dict):
-        error_return_tuple = (
-            gr.update(value="*Please select a report to view its details.*"),
-            gr.update(choices=[], value=[], interactive=False),
-            gr.update(value="*Please select a report to see OKRs.*"),
-            None, [], [], session_cache # Pass cache back unchanged
-        )
+    # --- FIXED: New handler only updates the report display ---
+    def update_report_display(selected_report_id: str, current_token_state: dict):
+        """
+        Updates only the report display markdown when a new report is selected.
+        The OKR visualization remains unchanged as it's loaded initially.
+        """
         if not selected_report_id:
-            return error_return_tuple
+            return gr.update(value="*Please select a report to view its details.*")
         
         agentic_df = current_token_state.get("bubble_agentic_analysis_data")
         if agentic_df is None or agentic_df.empty:
-            return error_return_tuple
+            return gr.update(value="*Analysis data not loaded or is empty.*")
 
         selected_report_series_df = agentic_df[agentic_df['_id'] == selected_report_id]
         if selected_report_series_df.empty:
-            error_return_tuple[0] = gr.update(value=f"*Error: Report with ID {selected_report_id} not found.*")
-            return error_return_tuple
-            
+            return gr.update(value=f"*Error: Report with ID '{selected_report_id}' not found.*")
+        
+        # Extract the report data and format it for display
         selected_report_series = selected_report_series_df.iloc[0]
         report_markdown = format_report_for_display(selected_report_series)
         
-        # Use the session cache
-        reconstructed_data, updated_cache = fetch_and_reconstruct_data_from_bubble(selected_report_series, session_cache)
-
-        if reconstructed_data:
-            raw_results_state = reconstructed_data
-            actionable_okrs_dict = reconstructed_data.get("actionable_okrs", {})
-            all_krs_state = extract_key_results_for_selection(actionable_okrs_dict)
-            if all_krs_state:
-                kr_choices = [(kr['kr_description'], kr['unique_kr_id']) for kr in all_krs_state]
-                key_results_cbg_update = gr.update(choices=kr_choices, value=[], interactive=True)
-                okrs_list = actionable_okrs_dict.get("okrs", [])
-                output_md_parts = [format_single_okr_for_display(okr, okr_main_index=i) for i, okr in enumerate(okrs_list)]
-                okr_details_md = "\n\n---\n\n".join(output_md_parts) if output_md_parts else "No OKRs defined."
-            else:
-                key_results_cbg_update = gr.update(choices=[], value=[], interactive=False)
-                okr_details_md = "No Key Results found for this report."
-                all_krs_state = []
-        else:
-            key_results_cbg_update = gr.update(choices=[], value=[], interactive=False)
-            okr_details_md = "Error: Could not fetch or reconstruct OKR data."
-            raw_results_state, all_krs_state = None, []
-        
-        return (
-            report_markdown, key_results_cbg_update, okr_details_md,
-            raw_results_state, [], all_krs_state, updated_cache
-        )
+        return report_markdown
 
     with gr.Tabs() as tabs:
         with gr.TabItem("1️⃣ Dashboard", id="tab_dashboard"):
@@ -243,22 +216,19 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
                 )
     
     if AGENTIC_MODULES_LOADED:
-        report_selection_outputs = [
-            agentic_report_display_md, key_results_cbg, okr_detail_display_md,
-            orchestration_raw_results_st, selected_key_result_ids_st, 
-            key_results_for_selection_st, reconstruction_cache_st # Pass cache state back
-        ]
+        # FIXED: The change event for the report selector now only updates the report display markdown.
+        # The OKR visualization is not affected and remains static after the initial load.
         report_selector_dd.change(
-            fn=update_report_and_okr_display,
-            inputs=[report_selector_dd, token_state, reconstruction_cache_st], # Pass cache state in
-            outputs=report_selection_outputs,
+            fn=update_report_display,
+            inputs=[report_selector_dd, token_state],
+            outputs=[agentic_report_display_md],
             show_progress="minimal"
         )
 
     agentic_display_outputs = [
         agentic_report_display_md, report_selector_dd, key_results_cbg,
         okr_detail_display_md, orchestration_raw_results_st, selected_key_result_ids_st,
-        key_results_for_selection_st, agentic_pipeline_status_md, reconstruction_cache_st # Pass cache state back
+        key_results_for_selection_st, agentic_pipeline_status_md, reconstruction_cache_st
     ]
 
     initial_load_event = org_urn_display.change(
@@ -276,7 +246,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
         show_progress="full"
     ).then(
         fn=load_and_display_agentic_results,
-        inputs=[token_state, reconstruction_cache_st], # Pass cache state in
+        inputs=[token_state, reconstruction_cache_st],
         outputs=agentic_display_outputs,
         show_progress="minimal"
     )