Update app.py

app.py

@@ -18,9 +18,6 @@ from config import (
     BUBBLE_API_ENDPOINT_ENV_VAR
 )
 
-# analytics_tab_module is now imported locally inside ui/ui_enhancements.py's build_main_app_ui
-# from services.analytics_tab_module import AnalyticsTab # REMOVED global import here
-
 # UPDATED: Using the new data loading function from the refactored state manager
 from services.state_manager import load_data_from_bubble
 
@@ -116,6 +113,7 @@ def initial_data_load_sequence(url_token, org_urn_val, current_state):
     return status_msg, new_state
 
 # Build the main UI using the function from ui_enhancements
+# This function now also sets up the main event listeners.
 (app, url_user_token_display, org_urn_display, status_box,
  token_state, reconstruction_cache_st, enhanced_okr_display_html,
  tabs, report_selector_dd, agentic_display_outputs,
@@ -136,46 +134,18 @@ def initial_data_load_sequence(url_token, org_urn_val, current_state):
         create_enhanced_okr_tab_func=create_enhanced_okr_tab,
         format_report_for_display_func=format_report_for_display, # Pass the imported function
         AGENTIC_MODULES_LOADED=AGENTIC_MODULES_LOADED, # ADD THIS PARAMETER
-        get_initial_okr_display_func=get_initial_okr_display if AGENTIC_MODULES_LOADED else None # ADD THIS TOO
-    )
-
-# Event handlers (re-establishing them now that components are defined)
-app.load(fn=get_url_user_token, inputs=None, outputs=[url_user_token_display, org_urn_display], api_name="get_url_params", show_progress=False)
-
-if AGENTIC_MODULES_LOADED:
-    report_selector_dd.change(
-        fn=lambda sr_id, c_state: update_report_display_enhanced(sr_id, c_state, format_report_for_display),
-        inputs=[report_selector_dd, token_state],
-        outputs=[agentic_display_outputs[7], agentic_display_outputs[8]], # report_header_html_display, report_body_markdown_display
-        show_progress="minimal"
+        get_initial_okr_display_func=get_initial_okr_display if AGENTIC_MODULES_LOADED else None, # ADD THIS TOO
+        # Pass the initial data load sequence function and others that need to be bound
+        # within the Gradio Blocks context.
+        initial_data_load_sequence_func=initial_data_load_sequence,
+        get_url_user_token_func=get_url_user_token,
+        load_and_display_agentic_results_func=load_and_display_agentic_results,
+        format_okrs_for_enhanced_display_func=format_okrs_for_enhanced_display,
+        update_report_display_enhanced_func=update_report_display_enhanced # Pass this as well
     )
 
-# Initial load sequence with enhanced status updates
-initial_load_event = org_urn_display.change(
-    fn=initial_data_load_sequence,
-    inputs=[url_user_token_display, org_urn_display, token_state],
-    outputs=[status_box, token_state],
-    show_progress="full"
-)
-
-# Chain the loading events
-initial_load_event.then(
-    fn=analytics_tab_instance.refresh_analytics_graphs_ui,
-    inputs=[token_state, analytics_tab_instance.date_filter_selector, analytics_tab_instance.custom_start_date_picker,
-            analytics_tab_instance.custom_end_date_picker, chat_histories_st_returned],
-    outputs=analytics_tab_instance.graph_refresh_outputs_list,
-    show_progress="full"
-).then(
-    fn=load_and_display_agentic_results,
-    inputs=[token_state, reconstruction_cache_st],
-    outputs=agentic_display_outputs,
-    show_progress="minimal"
-).then(
-    fn=format_okrs_for_enhanced_display,
-    inputs=[reconstruction_cache_st],
-    outputs=[enhanced_okr_display_html],
-    show_progress="minimal"
-)
+# The event handlers are now set up *inside* build_main_app_ui,
+# so these calls are no longer needed here.
 
 if __name__ == "__main__":
     # Enhanced startup logging
@@ -208,10 +178,10 @@ if __name__ == "__main__":
     
     print("🌐 Launching dashboard on http://0.0.0.0:7860")
     print("🎯 Dashboard features:")
-    print("   • 📊 Advanced LinkedIn Analytics")
-    print("   • 🤖 AI-Powered Insights")
-    print("   • 🎯 OKR Generation & Tracking") 
-    print("   • ☁️  Bubble.io Integration")
+    print("    • 📊 Advanced LinkedIn Analytics")
+    print("    • 🤖 AI-Powered Insights")
+    print("    • 🎯 OKR Generation & Tracking") 
+    print("    • ☁️  Bubble.io Integration")
     
     app.launch(
         server_name="0.0.0.0", 
@@ -221,3 +191,4 @@ if __name__ == "__main__":
         show_tips=True,
         enable_queue=True
     )
+