Update app.py

app.py

@@ -26,6 +26,20 @@ from ui_generators import (
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
+# --- Helper function to load HTML animation ---
+def get_sync_animation_html():
+    """Loads the HTML content for the sync animation."""
+    try:
+        # Ensure this path is correct relative to where app.py is run
+        with open("sync_animation.html", "r", encoding="utf-8") as f:
+            return f.read()
+    except FileNotFoundError:
+        logging.error("sync_animation.html not found. Please ensure it's in the same directory as app.py.")
+        return "<p style='text-align:center; color: red;'>Animation file not found. Syncing...</p>"
+    except Exception as e:
+        logging.error(f"Error loading sync_animation.html: {e}")
+        return f"<p style='text-align:center; color: red;'>Error loading animation: {e}. Syncing...</p>"
+
 # --- Guarded Analytics Fetch ---
 def guarded_fetch_analytics(token_state):
     """Guarded call to fetch_and_render_analytics, ensuring token and basic data structures."""
@@ -64,6 +78,9 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
         "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
         "url_user_token_temp_storage": None
     })
 
@@ -82,14 +99,17 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
         # 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)
+        dashboard_content = display_main_dashboard(new_state) # Ensure this function handles potentially empty dfs
         return status_msg, new_state, btn_update, dashboard_content
 
     with gr.Tabs():
         with gr.TabItem("1️⃣ Dashboard & Sync"):
             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>")
+            
             dashboard_display_html = gr.HTML("<p style='text-align:center;'>Dashboard loading...</p>")
 
             # Chain of events for initial load:
@@ -99,26 +119,31 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
                 outputs=[status_box, token_state, sync_data_btn, dashboard_display_html],
                 show_progress="full"
             )
-            # Also trigger initial_load_sequence if url_user_token_display changes (e.g. if it loads after org_urn)
-            # This helps ensure it runs once both are potentially available.
-            # Note: `org_urn_display.change` might be sufficient if `get_url_user_token` updates both nearly simultaneously.
-            # Adding this for robustness, but ensure it doesn't cause unwanted multiple runs if state isn't managed carefully.
-            # Consider using a flag in token_state if multiple triggers become an issue.
-            # For now, relying on org_urn_display.change as the primary trigger post-load.
+            # 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=sync_all_linkedin_data_orchestrator,
+                fn=lambda: get_sync_animation_html(), # Step 1: Show animation
+                inputs=None,
+                outputs=[sync_status_html_output],
+                show_progress=False # Animation is its own progress indicator
+            ).then(
+                fn=sync_all_linkedin_data_orchestrator, # Step 2: Run the actual sync
                 inputs=[token_state],
-                outputs=[sync_status_html_output, token_state], # token_state is updated here
-                show_progress="full"
+                # 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
             ).then(
-                fn=process_and_store_bubble_token, # Re-check sync status and update button
-                inputs=[url_user_token_display, org_urn_display, token_state], # Pass current token_state
-                outputs=[status_box, token_state, sync_data_btn], # token_state updated again
-                show_progress=False # Typically "full" for user-initiated actions, "minimal" or False for quick updates
+                fn=process_and_store_bubble_token, # Step 3: Re-check sync status and update button
+                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, # Refresh dashboard display
+                fn=display_main_dashboard, # Step 4: Refresh dashboard display
                 inputs=[token_state],
                 outputs=[dashboard_display_html],
                 show_progress=False
@@ -182,4 +207,5 @@ if __name__ == "__main__":
         logging.error("Matplotlib is not installed. Plots will not be generated. Please install it: pip install matplotlib")
 
     # Launch the Gradio app
-    app.launch(server_name="0.0.0.0", server_port=7860, debug=True)
+    # 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)