Update services/report_data_handler.py

services/report_data_handler.py

@@ -31,8 +31,6 @@ def fetch_latest_agentic_analysis(org_urn: str) -> Tuple[Optional[pd.DataFrame],
         return None, "org_urn is missing."
 
     try:
-        # We fetch all reports and will sort them later if needed, but typically the
-        # external process should manage providing the "latest" or "active" report.
         report_data_df, error = fetch_linkedin_posts_data_from_bubble(
             data_type=BUBBLE_REPORT_TABLE_NAME,
             constraint_value=org_urn,
@@ -56,33 +54,30 @@ def fetch_latest_agentic_analysis(org_urn: str) -> Tuple[Optional[pd.DataFrame],
         return None, str(e)
 
 
-def fetch_and_reconstruct_data_from_bubble(report_df: pd.DataFrame) -> Optional[Dict[str, Any]]:
+def fetch_and_reconstruct_data_from_bubble(report_series: pd.Series) -> Optional[Dict[str, Any]]:
     """
-    Takes a DataFrame of report data, fetches all related child items (OKRs, KRs, Tasks)
-    from Bubble, and reconstructs the full nested dictionary expected by the UI.
+    MODIFIED: Takes a pandas Series of a single report, fetches all related child items 
+    (OKRs, KRs, Tasks) from Bubble, and reconstructs the full nested dictionary.
 
     Args:
-        report_df: The DataFrame containing one or more reports, fetched previously.
+        report_series: A pandas Series representing a single report to be processed.
 
     Returns:
         A dictionary containing the reconstructed data ('report_str', 'actionable_okrs'),
-        or None if the report is not found or a critical error occurs.
+        or None if the report series is invalid or a critical error occurs.
     """
-    logger.info("Starting data reconstruction from fetched Bubble data.")
-    if report_df is None or report_df.empty:
-        logger.warning("Cannot reconstruct data, the provided report DataFrame is empty.")
+    logger.info("Starting data reconstruction from a single fetched Bubble report.")
+    if report_series is None or report_series.empty:
+        logger.warning("Cannot reconstruct data, the provided report Series is empty.")
         return None
 
     try:
-        # Assuming the most recent report is desired if multiple are returned.
-        # You might need more sophisticated logic here to select the "active" report.
-        latest_report = report_df.sort_values(by='Created Date', ascending=False).iloc[0]
-        report_id = latest_report.get('_id')
+        report_id = report_series.get('_id')
         if not report_id:
-            logger.error("Fetched report is missing a Bubble '_id', cannot reconstruct children.")
+            logger.error("Fetched report series is missing a Bubble '_id', cannot reconstruct children.")
             return None
 
-        logger.info(f"Reconstructing data for the latest report, ID: {report_id}")
+        logger.info(f"Reconstructing all related data for report, ID: {report_id}")
 
         # 1. Fetch all related OKRs using the report_id
         okrs_df, error = fetch_linkedin_posts_data_from_bubble(
@@ -137,13 +132,13 @@ def fetch_and_reconstruct_data_from_bubble(report_df: pd.DataFrame) -> Optional[
         # 5. Assemble the final payload for the UI
         actionable_okrs = {"okrs": reconstructed_okrs}
         final_reconstructed_data = {
-            "report_str": latest_report.get("report_text", "Report text not found."),
-            "quarter": latest_report.get("quarter"),
-            "year": latest_report.get("year"),
+            "report_str": report_series.get("report_text", "Report text not found."),
+            "quarter": report_series.get("quarter"),
+            "year": report_series.get("year"),
             "actionable_okrs": actionable_okrs,
             "report_id": report_id
         }
-        logger.info("Successfully reconstructed nested data structure for the UI.")
+        logger.info(f"Successfully reconstructed nested data structure for report {report_id}.")
         return final_reconstructed_data
 
     except Exception as e: