Update run_agentic_pipeline.py

run_agentic_pipeline.py

@@ -25,8 +25,6 @@ except ImportError as e:
         return {'header_html': '<h1>Agentic modules not loaded.</h1>', 'body_markdown': 'Report display unavailable.'}
     # These functions are now used by app.py directly for placeholder returns if modules are not loaded,
     # but their definitions might still be expected in some contexts, so providing minimal ones.
-    # We still need placeholders for the *return values* if the import fails, even if they're not called
-    # directly within this file, to maintain the correct function signature for load_and_display_agentic_results.
     def extract_key_results_for_selection(okrs_dict): return []
     def format_single_okr_for_display(okr_data, **kwargs): return "Agentic modules not loaded."
     # Placeholder for the new OKR UI display
@@ -68,10 +66,11 @@ def load_and_display_agentic_results(token_state: dict, session_cache: dict):
         session_cache: The session-specific cache for reconstructed data.
 
     Returns:
-        A tuple of Gradio updates, including the updated cache and the new enhanced OKR HTML.
-        This function now returns 11 values to match the expected outputs in app.py.
+        A tuple of Gradio updates, including the updated cache, the new enhanced OKR HTML,
+        and the extracted actionable_okrs_dict.
+        This function now returns 12 values to match the expected outputs in app.py.
     """
-    # Define placeholder content for empty or error states to match 11 outputs
+    # Define placeholder content for empty or error states to match 12 outputs
     empty_header_html = """
         <div class="report-title">📊 Comprehensive Analysis Report</div>
         <div class="report-subtitle">AI-Generated Insights from Your LinkedIn Data</div>
@@ -87,7 +86,7 @@ def load_and_display_agentic_results(token_state: dict, session_cache: dict):
         </div>
         """
 
-    # Default initial updates for 11 outputs
+    # Default initial updates for 12 outputs
     initial_updates = (
         gr.update(value="Status: No agentic analysis data found."), # 0: agentic_pipeline_status_md
         gr.update(choices=[], value=None, interactive=False),       # 1: report_selector_dd
@@ -99,12 +98,13 @@ def load_and_display_agentic_results(token_state: dict, session_cache: dict):
         gr.update(value=empty_header_html),                         # 7: report_header_html_display
         gr.update(value=empty_body_markdown_no_selection),          # 8: report_body_markdown_display
         gr.update(value=session_cache),                             # 9: reconstruction_cache_st
-        gr.update(value=get_initial_okr_display())                  # 10: enhanced_okr_display_html
+        gr.update(value=get_initial_okr_display()),                 # 10: enhanced_okr_display_html
+        gr.update(value={})                                         # 11: actionable_okrs_data_st (NEW)
     )
 
     if not AGENTIC_MODULES_LOADED:
         logger.error("Agentic modules not loaded, returning placeholder updates.")
-        # Ensure error updates match the 11-item signature
+        # Ensure error updates match the 12-item signature
         error_header_html = '<div class="report-title">Error Loading Report</div><div class="report-subtitle">Agentic modules not loaded.</div><div class="status-badge">Error</div>'
         error_body_markdown = '<div class="empty-state"><div class="empty-state-icon">❌</div><div class="empty-state-title">Module Error</div><div class="empty-state-description">Agentic analysis modules could not be loaded. Report display unavailable.</div></div>'
         return (
@@ -118,7 +118,8 @@ def load_and_display_agentic_results(token_state: dict, session_cache: dict):
             gr.update(value=error_header_html), # 7
             gr.update(value=error_body_markdown), # 8
             gr.update(value=session_cache), # 9
-            gr.update(value=get_initial_okr_display()) # 10: Placeholder for enhanced OKR display
+            gr.update(value=get_initial_okr_display()), # 10: Placeholder for enhanced OKR display
+            gr.update(value={}) # 11: Placeholder for actionable_okrs_data_st (NEW)
         )
 
     agentic_df = token_state.get("bubble_agentic_analysis_data")
@@ -165,6 +166,7 @@ def load_and_display_agentic_results(token_state: dict, session_cache: dict):
         raw_results_state = None
         okr_details_md = "No OKRs found in the latest report." # This is for the old, hidden component
         enhanced_okr_html_content = get_initial_okr_display() # Default to loading state or empty
+        actionable_okrs_dict_for_state = {} # NEW: Initialize the new state variable
         key_results_cbg_update = gr.update(choices=[], value=[], interactive=False)
         all_krs_state = []
 
@@ -174,6 +176,7 @@ def load_and_display_agentic_results(token_state: dict, session_cache: dict):
 
             # Refined check for actionable_okrs_dict content
             if actionable_okrs_dict and isinstance(actionable_okrs_dict, dict) and actionable_okrs_dict.get("okrs"):
+                actionable_okrs_dict_for_state = actionable_okrs_dict # Store for the new state
                 # Format for the new enhanced HTML display
                 enhanced_okr_html_content = format_okrs_for_enhanced_display(raw_results_state)
             else:
@@ -198,12 +201,13 @@ def load_and_display_agentic_results(token_state: dict, session_cache: dict):
             gr.update(value=report_header_content),      # 7: report_header_html_display
             gr.update(value=report_body_content),        # 8: report_body_markdown_display
             gr.update(value=updated_cache),              # 9: reconstruction_cache_st
-            gr.update(value=enhanced_okr_html_content)   # 10: NEW: The enhanced HTML display for OKRs
+            gr.update(value=enhanced_okr_html_content),   # 10: NEW: The enhanced HTML display for OKRs
+            gr.update(value=actionable_okrs_dict_for_state) # 11: NEW: Explicit actionable_okrs_data_st
         )
 
     except Exception as e:
         logger.error(f"Failed to process and display agentic results: {e}", exc_info=True)
-        # Ensure error returns match the 11-item signature
+        # Ensure error returns match the 12-item signature
         error_header_html = """
             <div class="report-title">⚠️ Error Loading Report</div>
             <div class="report-subtitle">An error occurred during data processing.</div>
@@ -230,5 +234,6 @@ def load_and_display_agentic_results(token_state: dict, session_cache: dict):
             gr.update(value=error_header_html), # 7
             gr.update(value=error_body_markdown), # 8
             gr.update(value=session_cache), # 9
-            gr.update(value=get_initial_okr_display()) # 10: Placeholder for enhanced OKR display on error
+            gr.update(value=get_initial_okr_display()), # 10: Placeholder for enhanced OKR display on error
+            gr.update(value={}) # 11: Placeholder for actionable_okrs_data_st on error
         )