Update data_planner.py

data_planner.py

@@ -6,17 +6,14 @@ def determine_data_requirements(user_prompt: str, company_code: str):
     system_msg = {
         "role": "system",
         "content": (
-            "You are a planning assistant for a Firestore chatbot. Your job is to analyze the user's question and return ONLY a JSON object with:
-"
-            "1. 'collections': a list of Firestore collections needed.
-"
-            "2. 'filters': a dictionary of filters to apply, such as companyCode.
-"
-            "3. 'instruction': a clear summary of what to do with the data (e.g., summarize interventions, count tasks).
-"
-            "Always include the filter 'companyCode': '<provided>' in your output.
-"
-            "Strictly return only a valid JSON object with keys: collections, filters, instruction."
+            "You are a planning assistant for a Firestore chatbot. Your job is to analyze the user's question and return ONLY a JSON object with:\n"
+            "1. 'collections': a list of Firestore collections needed.\n"
+            "2. 'filters': a dictionary of filters to apply, such as companyCode.\n"
+            "3. 'instruction': a clear summary of what to do with the data (e.g., summarize interventions, count tasks).\n"
+            "Always include the filter 'companyCode': '<provided>' in your output.\n"
+            "Strictly return only a valid JSON object with keys: collections, filters, instruction.\n"
+            "Never return markdown, explanations, or preamble.\n"
+            "Return ONLY a raw JSON object."
         )
     }
 
@@ -27,6 +24,13 @@ def determine_data_requirements(user_prompt: str, company_code: str):
 
     try:
         result = ask_gpt([system_msg, user_msg])
-        return json.loads(result)
+        print("🧠 Raw planning response from Gemini:")
+        print(result)
+
+        parsed = json.loads(result)
+        if "collections" not in parsed:
+            raise ValueError("Response JSON missing 'collections'")
+        return parsed
     except Exception as e:
-        return {"error": str(e)}
+        print("❌ Planning error:", str(e))
+        return {"error": f"Planning failed: {str(e)}"}