Create data_planner.py

data_planner.py

@@ -0,0 +1,32 @@
+# 🔧 data_planner.py
+import json
+from openai_client import ask_gpt
+
+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."
+        )
+    }
+
+    user_msg = {
+        "role": "user",
+        "content": f"User question: {user_prompt}\n\nCompanyCode: {company_code}"
+    }
+
+    try:
+        result = ask_gpt([system_msg, user_msg])
+        return json.loads(result)
+    except Exception as e:
+        return {"error": str(e)}