Update ui/okr_ui_generator.py

ui/okr_ui_generator.py

@@ -456,20 +456,29 @@ def get_initial_okr_display() -> str:
     </div>
     """
 
-def format_okrs_for_enhanced_display(raw_results: dict) -> str:
+def format_okrs_for_enhanced_display(reconstruction_cache: dict) -> str:
     """
     Enhanced formatting function that creates beautiful HTML for OKR display
-    from the raw results dictionary.
+    from the reconstruction cache dictionary.
 
     Args:
-        raw_results (dict): The dictionary containing the 'actionable_okrs' data.
-                            Expected structure: {'actionable_okrs': {'okrs': List[OKR_dict]}}
+        reconstruction_cache (dict): The reconstruction cache containing reconstructed data.
+                                   Expected to contain report data with 'actionable_okrs' key.
 
     Returns:
         str: A comprehensive HTML string representing the OKRs, or an empty state HTML.
     """
-    if not raw_results or not raw_results.get("actionable_okrs"):
-        logger.warning("No raw results or 'actionable_okrs' found for display.")
+    if not reconstruction_cache:
+        logger.warning("No reconstruction cache found for display.")
+        return get_empty_okr_state()
+
+    # Extract actionable_okrs from the first (and typically only) report in cache
+    for report_id, report_data in reconstruction_cache.items():
+        if isinstance(report_data, dict) and 'actionable_okrs' in report_data:
+            raw_results = {'actionable_okrs': report_data['actionable_okrs']}
+            break
+    else:
+        logger.warning("No 'actionable_okrs' found in reconstruction cache for display.")
         return get_empty_okr_state()
 
     actionable_okrs = raw_results.get("actionable_okrs", {})
@@ -479,6 +488,8 @@ def format_okrs_for_enhanced_display(raw_results: dict) -> str:
         logger.info("No OKRs found in 'actionable_okrs' list.")
         return get_empty_okr_state()
 
+    # Rest of the function remains the same...
+
     # Calculate statistics for the stats bar
     total_objectives = len(okrs_list)
     total_key_results = sum(len(okr.get('key_results', [])) for okr in okrs_list)