Update app.py

app.py

@@ -76,15 +76,15 @@ except ImportError as e:
         # matching the `outputs` in the .then() call later.
         return (
             gr.update(value="Modules not loaded."), # agentic_pipeline_status_md (0)
-            gr.update(choices=[], value=None),      # report_selector_dd (1)
-            gr.update(choices=[], value=[]),        # key_results_cbg (2) - KEPT HIDDEN for compatibility
+            gr.update(choices=[], value=None),       # report_selector_dd (1)
+            gr.update(choices=[], value=[]),         # key_results_cbg (2) - KEPT HIDDEN for compatibility
             gr.update(value="Modules not loaded."), # okr_detail_display_md (3) - KEPT HIDDEN for compatibility
-            None,                                   # orchestration_raw_results_st (4)
-            [],                                     # selected_key_result_ids_st (5) - KEPT HIDDEN for compatibility
-            [],                                     # key_results_for_selection_st (6) - KEPT HIDDEN for compatibility
-            gr.update(value=empty_header_html),     # report_header_html_display (7)
+            None,                                    # orchestration_raw_results_st (4)
+            [],                                      # selected_key_result_ids_st (5) - KEPT HIDDEN for compatibility
+            [],                                      # key_results_for_selection_st (6) - KEPT HIDDEN for compatibility
+            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)
+            {},                                      # reconstruction_cache_st (9)
             gr.update(value=get_initial_okr_display()), # NEW: enhanced_okr_display_html (10)
             gr.update(value={}) # NEW: actionable_okrs_data_st (11)
         )
@@ -96,7 +96,7 @@ except ImportError as e:
 
 
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
-               title="LinkedIn Organization Dashboard") as app:
+              title="LinkedIn Organization Dashboard") as app:
     # --- STATE MANAGEMENT ---
     token_state = gr.State(value={
         "token": None, "client_id": None, "org_urn": None,
@@ -130,17 +130,28 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
     gr.Markdown("# 🚀 LinkedIn Organization Dashboard")
     url_user_token_display = gr.Textbox(label="User Token (Hidden)", interactive=False, visible=False)
     org_urn_display = gr.Textbox(label="Org URN (Hidden)", interactive=False, visible=False)
-    status_box = gr.Textbox(label="Status", interactive=False, value="Initializing...")
+    # MODIFIED: Initially hide the status_box and make its value empty
+    status_box = gr.Textbox(label="Status", interactive=False, value="", visible=False)
 
     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)
 
     def initial_data_load_sequence(url_token, org_urn_val, current_state):
         """
-        Handles the initial data loading from Bubble.
-        No longer generates dashboard HTML as the Home tab is now static.
+        Handles the initial data loading from Bubble and controls status_box visibility.
         """
         status_msg, new_state = load_data_from_bubble(url_token, org_urn_val, current_state)
-        return status_msg, new_state
+
+        # Determine if the status box should be visible
+        # It should be visible if the token is not available (None)
+        # OR if the status message indicates an error or warning, even if a token *might* be present.
+        show_status_box = False
+        if new_state.get("token") is None:
+            show_status_box = True
+        elif "Error" in status_msg or "Warning" in status_msg:
+            show_status_box = True
+        
+        # Return the gr.update object to control value and visibility
+        return gr.update(value=status_msg, visible=show_status_box), new_state
 
     analytics_icons = {'bomb': BOMB_ICON, 'explore': EXPLORE_ICON, 'formula': FORMULA_ICON, 'active': ACTIVE_ICON}
     analytics_tab_instance = AnalyticsTab(
@@ -281,24 +292,30 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
     # Ensure agentic_display_outputs correctly maps to the newly created components
     # This list must match the outputs of load_and_display_agentic_results
     agentic_display_outputs = [
-        agentic_pipeline_status_md,         # 0: Status Markdown (hidden)
-        report_selector_dd,                 # 1: Dropdown for selecting reports
-        key_results_cbg,                    # 2: Checkbox group for OKRs (kept hidden)
-        okr_detail_display_md,              # 3: Markdown for detailed OKR display (kept hidden)
-        orchestration_raw_results_st,       # 4: Raw results state
-        selected_key_result_ids_st,         # 5: Selected KR IDs state (kept hidden)
-        key_results_for_selection_st,       # 6: All KRs for selection state (kept hidden)
-        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
-        actionable_okrs_data_st             # 11: NEW: The actionable_okrs dictionary state
+        agentic_pipeline_status_md,            # 0: Status Markdown (hidden)
+        report_selector_dd,                    # 1: Dropdown for selecting reports
+        key_results_cbg,                       # 2: Checkbox group for OKRs (kept hidden)
+        okr_detail_display_md,                 # 3: Markdown for detailed OKR display (kept hidden)
+        orchestration_raw_results_st,          # 4: Raw results state
+        selected_key_result_ids_st,            # 5: Selected KR IDs state (kept hidden)
+        key_results_for_selection_st,          # 6: All KRs for selection state (kept hidden)
+        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
+        actionable_okrs_data_st                # 11: NEW: The actionable_okrs dictionary state
     ]
 
+    # MODIFIED: Chain the initial_load_event to first show "Loading...", then perform data load.
     initial_load_event = org_urn_display.change(
+        fn=lambda: gr.update(value="Loading data...", visible=True), # Show loading message immediately
+        inputs=[],
+        outputs=[status_box],
+        show_progress="full"
+    ).then(
         fn=initial_data_load_sequence,
         inputs=[url_user_token_display, org_urn_display, token_state],
-        outputs=[status_box, token_state],
+        outputs=[status_box, token_state], # status_box is updated here with final message/visibility
         show_progress="full"
     )
 
@@ -332,4 +349,4 @@ if __name__ == "__main__":
     if not os.environ.get("GEMINI_API_KEY"):
         logging.warning("WARNING: 'GEMINI_API_KEY' is not set.")
 
-    app.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)), debug=True)
+    app.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)), debug=True)