Update ui/ui_main_page_enhancements.py

ui/ui_main_page_enhancements.py

@@ -532,7 +532,9 @@ def build_main_app_ui(
     build_home_tab_ui_func,
     create_enhanced_report_tab_func,
     create_enhanced_okr_tab_func,
-    format_report_for_display_func # New argument to pass the formatting function
+    format_report_for_display_func, # New argument to pass the formatting function
+    AGENTIC_MODULES_LOADED, # ADD THIS PARAMETER
+    get_initial_okr_display_func=None # ADD THIS PARAMETER TOO
 ):
     """
     Builds the main Gradio application UI with enhanced styling and structure.
@@ -540,6 +542,8 @@ def build_main_app_ui(
     Args:
         (All necessary functions and constants are passed as arguments to ensure self-containment
          within the UI module without direct imports of non-UI related logic)
+        AGENTIC_MODULES_LOADED (bool): Whether agentic modules are loaded
+        get_initial_okr_display_func (callable): Function to get initial OKR display
 
     Returns:
         tuple: A tuple containing:
@@ -686,6 +690,20 @@ def build_main_app_ui(
                         <p>Agentic modules could not be loaded. This tab is currently unavailable.</p>
                     </div>
                     """)
+                    # Create dummy components to maintain compatibility
+                    enhanced_okr_display_html = gr.HTML("""
+                    <div class="card" style="text-align: center; padding: 2rem;">
+                        <div style="font-size: 2rem; margin-bottom: 1rem;">🔴</div>
+                        <h3 style="color: var(--error-color);">OKR Display Unavailable</h3>
+                        <p>Agentic modules are not loaded.</p>
+                    </div>
+                    """)
+                    # Create hidden old components for compatibility
+                    with gr.Column(visible=False):
+                        gr.Markdown("### Suggested Key Results (OLD UI - HIDDEN)")
+                        key_results_cbg = gr.CheckboxGroup(label="Select Key Results", choices=[], value=[], interactive=True)
+                        gr.Markdown("### Detailed OKRs and Tasks (OLD UI - HIDDEN)")
+                        okr_detail_display_md = gr.Markdown("I dettagli OKR appariranno qui.")
                 else:
                     # Keep the old components but make them invisible to maintain load_and_display_agentic_results signature
                     with gr.Column(visible=False):
@@ -721,7 +739,7 @@ def build_main_app_ui(
         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 if AGENTIC_MODULES_LOADED else gr.State(get_initial_okr_display()), # NEW: The enhanced HTML display for OKRs
+        enhanced_okr_display_html,             # 10: The enhanced HTML display for OKRs
         actionable_okrs_data_st                # 11: NEW: The actionable_okrs dictionary state
     ]
 
@@ -730,4 +748,4 @@ def build_main_app_ui(
             tabs, report_selector_dd, agentic_display_outputs,
             analytics_tab_instance, chat_histories_st,
             current_chat_plot_id_st, plot_data_for_chatbot_st, # Return these as well
-            format_report_for_display_func) # Return format_report_for_display_func as well
+            format_report_for_display_func) # Return format_report_for_display_func as well