Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import gradio as gr
 import pandas as pd
 import os
@@ -36,7 +35,8 @@ from features.chatbot.chatbot_handler import generate_llm_response
 try:
     # UPDATED: Using the new display function to show pre-computed results
     from run_agentic_pipeline import load_and_display_agentic_results
-    from ui.insights_ui_generator import format_single_okr_for_display
+    # ADDED: Import the new report formatter
+    from ui.insights_ui_generator import format_single_okr_for_display, format_report_for_display
     AGENTIC_MODULES_LOADED = True
 except ImportError as e:
     logging.error(f"Could not import agentic pipeline display modules: {e}. Tabs 3 and 4 will be disabled.")
@@ -44,9 +44,11 @@ except ImportError as e:
     # Placeholder for the new function name if imports fail
     def load_and_display_agentic_results(*args, **kwargs):
         # This tuple matches the expected number of outputs for the event handler
-        return "Modules not loaded.", "Modules not loaded.", "Modules not loaded.", None, [], [], "Error"
+        return "Modules not loaded.", gr.update(), "Modules not loaded.", "Modules not loaded.", None, [], [], "Error"
     def format_single_okr_for_display(okr_data, **kwargs):
         return "Agentic modules not loaded. OKR display unavailable."
+    def format_report_for_display(report_data):
+        return "Agentic modules not loaded. Report display unavailable."
 
 # --- ANALYTICS TAB MODULE IMPORT ---
 from services.analytics_tab_module import AnalyticsTab
@@ -122,6 +124,30 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
         fn_get_initial_insight=get_initial_insight_prompt_and_suggestions,
         fn_generate_llm_response=generate_llm_response
     )
+    
+    # --- NEW: Handler function for report selection ---
+    def display_selected_report(selected_report_id: str, current_token_state: dict):
+        """
+        Finds the selected report in the state and formats it for display.
+        """
+        if not selected_report_id:
+            return gr.update(value="*Please select a report from the library.*")
+        
+        agentic_df = current_token_state.get("bubble_agentic_analysis_data")
+        if agentic_df is None or agentic_df.empty:
+             return gr.update(value="*Report data not loaded. Cannot display report.*")
+
+        # The dropdown value is the unique ID of the report (e.g., Bubble _id)
+        selected_report_series_df = agentic_df[agentic_df['_id'] == selected_report_id]
+        
+        if selected_report_series_df.empty:
+            return gr.update(value=f"*Error: Could not find report with ID {selected_report_id}.*")
+            
+        selected_report_series = selected_report_series_df.iloc[0]
+        report_markdown = format_report_for_display(selected_report_series)
+        
+        return gr.update(value=report_markdown)
+
 
     with gr.Tabs() as tabs:
         with gr.TabItem("1️⃣ Dashboard", id="tab_dashboard"):
@@ -132,12 +158,24 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
         # Use the AnalyticsTab module to create Tab 2
         analytics_tab_instance.create_tab_ui()
 
-        # Tab 3: Agentic Analysis Report
+        # Tab 3: Agentic Analysis Report - MODIFIED
         with gr.TabItem("3️⃣ Agentic Analysis Report", id="tab_agentic_report", visible=AGENTIC_MODULES_LOADED):
             gr.Markdown("## 🤖 Comprehensive Analysis Report (from Bubble.io)")
             agentic_pipeline_status_md = gr.Markdown("Status: Loading report data...", visible=True)
-            gr.Markdown("Questo report è stato pre-generato da un agente AI e caricato da Bubble.io.")
-            agentic_report_display_md = gr.Markdown("The AI-generated report will be displayed here once loaded.")
+            gr.Markdown("Questo report è stato pre-generato da un agente AI e caricato da Bubble.io. Seleziona un report dalla libreria per visualizzarlo.")
+            
+            # NEW: Report library dropdown
+            with gr.Row():
+                report_selector_dd = gr.Dropdown(
+                    label="Report Library",
+                    choices=[],
+                    interactive=True,
+                    info="Select a previously generated report to view its details."
+                )
+            
+            # This Markdown component will now display the selected report
+            agentic_report_display_md = gr.Markdown("Please select a report from the library to view it.")
+            
             if not AGENTIC_MODULES_LOADED:
                 gr.Markdown("🔴 **Error:** Agentic pipeline display modules could not be loaded. This tab is disabled.")
 
@@ -192,11 +230,22 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
                     inputs=[key_results_cbg, orchestration_raw_results_st, key_results_for_selection_st],
                     outputs=[okr_detail_display_md]
                 )
+    
+    # --- NEW: Event binding for the report selector dropdown ---
+    if AGENTIC_MODULES_LOADED:
+        report_selector_dd.change(
+            fn=display_selected_report,
+            inputs=[report_selector_dd, token_state],
+            outputs=[agentic_report_display_md],
+            show_progress="minimal"
+        )
+
 
     # --- EVENT HANDLING (SIMPLIFIED) ---
-    # Define the output list for loading agentic results
+    # MODIFIED: Define the output list for loading agentic results
     agentic_display_outputs = [
         agentic_report_display_md,
+        report_selector_dd, # ADDED: The new dropdown
         key_results_cbg,
         okr_detail_display_md,
         orchestration_raw_results_st,
@@ -227,9 +276,9 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
         show_progress="full"
     # Then, load and display the pre-computed agentic results
     ).then(
-        fn=load_and_display_agentic_results, # UPDATED function call
+        fn=load_and_display_agentic_results, # This function's implementation needs to be updated
         inputs=[token_state, orchestration_raw_results_st, selected_key_result_ids_st, key_results_for_selection_st],
-        outputs=agentic_display_outputs,
+        outputs=agentic_display_outputs, # The output list is now updated
         show_progress="minimal"
     )