Update services/state_manager.py

services/state_manager.py

@@ -28,6 +28,8 @@ from config import (
     LOG_SUBJECT_POSTS, LOG_SUBJECT_MENTIONS, LOG_SUBJECT_FOLLOWER_STATS
 )
 
+from .report_data_handler import fetch_latest_agentic_analysis
+
 def check_token_status(token_state):
     """Checks the status of the LinkedIn token."""
     return "✅ Token available" if token_state and token_state.get("token") else "❌ Token not available"
@@ -147,6 +149,32 @@ def process_and_store_bubble_token(url_user_token, org_urn, token_state):
             except Exception as e:
                 logging.error(f"❌ Error fetching {table_name} from Bubble: {e}.", exc_info=True)
                 new_state[state_key] = pd.DataFrame()
+
+        new_state["agentic_pipeline_should_run_now"] = False
+        new_state["bubble_agentic_analysis_data"] = None
+        
+        # Fetch all report data
+        all_analysis_data, error = fetch_latest_agentic_analysis(current_org_urn)
+        
+        if all_analysis_data is not None and not all_analysis_data.empty:
+            new_state["bubble_agentic_analysis_data"] = all_analysis_data
+            
+            quarter_data = all_analysis_data[all_analysis_data['type'] == 'Quarter'].copy()
+            
+            if not quarter_data.empty:
+                latest_quarter_analysis = quarter_data.sort_values(by='Created Date', ascending=False).iloc[0]
+                should_run_due_to_no_data = pd.isna(latest_quarter_analysis)
+            else:
+                should_run_due_to_no_data = True  # No quarter data found
+        else:
+            should_run_due_to_no_data = True  # No data at all
+        
+        if should_run_due_to_no_data:
+            logging.info(f"No existing agentic analysis data found for org {current_org_urn}. Triggering run.")
+            new_state["agentic_pipeline_should_run_now"] = True
+        else:
+            logging.info(f"Agentic analysis data for org {current_org_urn} is up-to-date.")
+        
     else:
         logging.warning("Org URN not available in state. Cannot fetch data from Bubble.")
         for key in ["bubble_posts_df", "bubble_mentions_df", "bubble_follower_stats_df", "bubble_operations_log_df"]: