Update app.py

app.py

@@ -82,33 +82,14 @@ except ImportError as e:
             gr.update(value=empty_header_html),     # report_header_html_display (7)
             gr.update(value=empty_body_markdown),   # report_body_markdown_display (8)
             {},                                     # reconstruction_cache_st (9)
-            gr.update(value=get_initial_okr_display()) # NEW: enhanced_okr_display_html (10)
+            gr.update(value=get_initial_okr_display()), # NEW: enhanced_okr_display_html (10)
+            gr.update(value={}) # NEW: actionable_okrs_data_st (11)
         )
     def fetch_and_reconstruct_data_from_bubble(*args, **kwargs):
         return None, {}
     def format_report_for_display(report_data):
         # Placeholder for when modules are not loaded, returns structure matching the new design
         return {'header_html': '<h1>Agentic modules not loaded.</h1>', 'body_markdown': 'Report display unavailable.'}
-    # REMOVED from insights_ui_generator.py, so also remove placeholder if not needed by other direct calls
-    # def extract_key_results_for_selection(okr_data):
-    #     return []
-    # def format_single_okr_for_display(okr_data, **kwargs):
-    #     return "Agentic modules not loaded. OKR display unavailable."
-
-
-# --- ANALYTICS TAB MODULE IMPORT ---
-from services.analytics_tab_module import AnalyticsTab
-
-# Configure logging
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(module)s - %(message)s')
-
-# API Key Setup
-user_provided_api_key = os.environ.get("GEMINI_API_KEY")
-if user_provided_api_key:
-    os.environ["GOOGLE_API_KEY"] = user_provided_api_key
-    logging.info("GOOGLE_API_KEY environment variable has been set from GEMINI_API_KEY.")
-else:
-    logging.error("CRITICAL ERROR: The API key environment variable 'GEMINI_API_KEY' was not found.")
 
 
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
@@ -138,6 +119,8 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
 
     # --- NEW: Session-specific cache for reconstructed OKR data ---
     reconstruction_cache_st = gr.State({})
+    # NEW: State to hold the actionable_okrs dictionary explicitly
+    actionable_okrs_data_st = gr.State({})
 
 
     # --- UI LAYOUT ---
@@ -305,7 +288,8 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
         report_header_html_display,         # 7: New HTML output for header
         report_body_markdown_display,       # 8: New Markdown output for body
         reconstruction_cache_st,            # 9: Reconstruction cache state
-        enhanced_okr_display_html           # 10: NEW: The enhanced HTML display for OKRs
+        enhanced_okr_display_html,          # 10: NEW: The enhanced HTML display for OKRs
+        actionable_okrs_data_st             # 11: NEW: The actionable_okrs dictionary state
     ]
 
     initial_load_event = org_urn_display.change(
@@ -324,12 +308,12 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
     ).then(
         fn=load_and_display_agentic_results,
         inputs=[token_state, reconstruction_cache_st],
-        # MODIFIED: Updated outputs to match all components returned by load_and_display_agentic_results
+        # MODIFIED: Updated outputs to match all components returned by load_and_display_agentic_results (now 12)
         outputs=agentic_display_outputs,
         show_progress="minimal"
     ).then( # NEW CHAIN: Update the enhanced OKR display after load_and_display_agentic_results runs
         fn=format_okrs_for_enhanced_display,
-        inputs=[orchestration_raw_results_st], # Take the raw results as input
+        inputs=[actionable_okrs_data_st], # Take the new explicit actionable_okrs_data_st as input
         outputs=[enhanced_okr_display_html],
         show_progress="minimal"
     )