Update app.py

app.py

@@ -45,10 +45,8 @@ def guarded_fetch_analytics(token_state):
     """Guarded call to fetch_and_render_analytics, ensuring token and basic data structures."""
     if not token_state or not token_state.get("token"):
         logging.warning("Analytics fetch: Access denied. No token.")
-        # Ensure the number of returned Nones matches the expected number of outputs for the plots
         return ("❌ Access denied. No token.", None, None, None, None, None, None, None)
 
-    # Ensure DataFrames are passed, even if empty, to avoid errors in the analytics function
     posts_df_analytics = token_state.get("bubble_posts_df", pd.DataFrame())
     mentions_df_analytics = token_state.get("bubble_mentions_df", pd.DataFrame())
     follower_stats_df_analytics = token_state.get("bubble_follower_stats_df", pd.DataFrame())
@@ -72,34 +70,28 @@ def guarded_fetch_analytics(token_state):
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
                title="LinkedIn Organization Dashboard") as app:
 
-    # Central state for holding token, client_id, org_urn, and fetched dataframes
     token_state = gr.State(value={
         "token": None, "client_id": None, "org_urn": None,
         "bubble_posts_df": pd.DataFrame(), "fetch_count_for_api": 0,
         "bubble_mentions_df": pd.DataFrame(),
         "bubble_follower_stats_df": pd.DataFrame(),
-        "bubble_operations_log_df": pd.DataFrame(), # Added from your state_manager logic
-        "mentions_should_sync_now": False, # Added from your state_manager logic
-        "fs_should_sync_now": False, # Added from your state_manager logic
+        "bubble_operations_log_df": pd.DataFrame(), 
+        "mentions_should_sync_now": False, 
+        "fs_should_sync_now": False, 
         "url_user_token_temp_storage": None
     })
 
     gr.Markdown("# 🚀 LinkedIn Organization Dashboard")
-    # Hidden textboxes to capture URL parameters
     url_user_token_display = gr.Textbox(label="User Token (from URL - Hidden)", interactive=False, visible=False)
     status_box = gr.Textbox(label="Overall LinkedIn Token Status", interactive=False, value="Initializing...")
     org_urn_display = gr.Textbox(label="Organization URN (from URL - Hidden)", interactive=False, visible=False)
 
-    # Load URL parameters when the Gradio app loads
     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)
 
-    # This function will run after URL params are loaded and org_urn_display changes
     def initial_load_sequence(url_token, org_urn_val, current_state):
         logging.info(f"Initial load sequence triggered. Org URN: {org_urn_val}, URL Token: {'Present' if url_token else 'Absent'}")
-        # Process token, fetch Bubble data, determine sync needs
         status_msg, new_state, btn_update = process_and_store_bubble_token(url_token, org_urn_val, current_state)
-        # Display initial dashboard content based on (potentially empty) Bubble data
-        dashboard_content = display_main_dashboard(new_state) # Ensure this function handles potentially empty dfs
+        dashboard_content = display_main_dashboard(new_state) 
         return status_msg, new_state, btn_update, dashboard_content
 
     with gr.Tabs():
@@ -107,43 +99,35 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
             gr.Markdown("System checks for existing data from Bubble. The 'Sync' button activates if new data needs to be fetched from LinkedIn based on the last sync times and data availability.")
             sync_data_btn = gr.Button("🔄 Sync LinkedIn Data", variant="primary", visible=False, interactive=False)
             
-            # This HTML component will now display the animation during sync
-            sync_status_html_output = gr.HTML("<p style='text-align:center;'>Sync status will appear here.</p>")
+            # MODIFIED: Added height=400 (or adjust as needed)
+            sync_status_html_output = gr.HTML("<p style='text-align:center;'>Sync status will appear here.</p>", height=400) 
             
             dashboard_display_html = gr.HTML("<p style='text-align:center;'>Dashboard loading...</p>")
 
-            # Chain of events for initial load:
             org_urn_display.change(
                 fn=initial_load_sequence,
                 inputs=[url_user_token_display, org_urn_display, token_state],
                 outputs=[status_box, token_state, sync_data_btn, dashboard_display_html],
                 show_progress="full"
             )
-            # Note: The logic for also triggering on url_user_token_display.change has been removed
-            # as org_urn_display.change should be sufficient if get_url_user_token updates both,
-            # or if org_urn is the primary trigger. If issues arise, this can be revisited.
 
-            # When Sync button is clicked:
             sync_data_btn.click(
-                fn=lambda: get_sync_animation_html(), # Step 1: Show animation
+                fn=lambda: get_sync_animation_html(), 
                 inputs=None,
                 outputs=[sync_status_html_output],
-                show_progress=False # Animation is its own progress indicator
+                show_progress=False 
             ).then(
-                fn=sync_all_linkedin_data_orchestrator, # Step 2: Run the actual sync
+                fn=sync_all_linkedin_data_orchestrator, 
                 inputs=[token_state],
-                # sync_all_linkedin_data_orchestrator must return:
-                # 1. HTML string for status (replaces animation)
-                # 2. Updated token_state
                 outputs=[sync_status_html_output, token_state], 
-                show_progress=False # Rely on animation; or set to "full" if you want Gradio's default too
+                show_progress=False 
             ).then(
-                fn=process_and_store_bubble_token, # Step 3: Re-check sync status and update button
+                fn=process_and_store_bubble_token, 
                 inputs=[url_user_token_display, org_urn_display, token_state], 
                 outputs=[status_box, token_state, sync_data_btn], 
                 show_progress=False 
             ).then(
-                fn=display_main_dashboard, # Step 4: Refresh dashboard display
+                fn=display_main_dashboard, 
                 inputs=[token_state],
                 outputs=[dashboard_display_html],
                 show_progress=False
@@ -191,7 +175,6 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
             )
 
 if __name__ == "__main__":
-    # Check for essential environment variables
     if not os.environ.get(LINKEDIN_CLIENT_ID_ENV_VAR):
         logging.warning(f"WARNING: '{LINKEDIN_CLIENT_ID_ENV_VAR}' environment variable not set. The app may not function correctly for LinkedIn API calls.")
     if not os.environ.get(BUBBLE_APP_NAME_ENV_VAR) or \
@@ -202,10 +185,7 @@ if __name__ == "__main__":
     try:
         import matplotlib
         logging.info(f"Matplotlib version: {matplotlib.__version__} found. Backend: {matplotlib.get_backend()}")
-        # The backend is now set in ui_generators.py, which is good practice.
     except ImportError:
         logging.error("Matplotlib is not installed. Plots will not be generated. Please install it: pip install matplotlib")
 
-    # Launch the Gradio app
-    # Ensure share=False for local testing unless you intend to share it via Gradio's link
     app.launch(server_name="0.0.0.0", server_port=7860, debug=True, share=False)