Update scenario_builder.py

scenario_builder.py

@@ -25,66 +25,80 @@ class ScenarioBuilder:
 
     async def create_scenario_from_query(self, query: str, num_questions: int = 5) -> Dict[str, Any]:
         prompt = f"""
-        Based on the medical query: '{query}', generate a {num_questions}-question interactive medical training scenario.
-        Each question must have 4 options (A, B, C, D), a correct answer (A|B|C|D), and a feedback explanation.
-        Format the response as a **valid JSON array** of objects, like:
-        [
-            {{
-                "question": "string",
-                "options": {{
-                    "A": "string",
-                    "B": "string",
-                    "C": "string",
-                    "D": "string"
-                }},
-                "correct_answer": "A",
-                "feedback": "string"
-            }}
-        ]
-        ⚠️ Respond ONLY with valid JSON.
-        ❌ DO NOT include markdown (like ```json) or extra explanation or commentary.
-        ✅ Output MUST be a raw JSON array. No formatting, no headings, no intro text.
-        """
+You are a medical training assistant.
+
+Based on the topic: "{query}", generate a {num_questions}-question interactive medical scenario. Each question must have:
+
+- a `question` (string)
+- an `options` object with keys "A"–"D"
+- a `correct_answer` ("A", "B", "C", or "D")
+- a `feedback` string
+
+Respond ONLY with this exact JSON format:
+
+[
+  {{
+    "question": "...",
+    "options": {{
+      "A": "...",
+      "B": "...",
+      "C": "...",
+      "D": "..."
+    }},
+    "correct_answer": "A|B|C|D",
+    "feedback": "..."
+  }},
+  ...
+]
+
+⚠️ No markdown. No prose. No headings. Just valid JSON array. Your job is to act like a JSON API.
+"""
 
         try:
-            response = self.ai_engine.generate_raw_text(prompt)
-            scenario_data_str = response.strip()
+            # Use Groq for accurate structure
+            raw_response = self.ai_engine._generate_with_groq(
+                query=query,
+                refined_prompt=prompt
+            )
 
-            if scenario_data_str.startswith("```"):
-                scenario_data_str = re.sub(r"^```(?:json)?", "", scenario_data_str)
-                scenario_data_str = re.sub(r"```$", "", scenario_data_str).strip()
+            # Clean any wrapping or garbage
+            start = raw_response.find("[")
+            end = raw_response.rfind("]") + 1
+            json_block = raw_response[start:end].strip()
 
-            print("🧪 Raw AI response preview:", scenario_data_str[:200])
+            print("🧪 Raw Groq output preview:", json_block[:200])
 
-            scenario_questions = json.loads(scenario_data_str)
+            scenario_data = json.loads(json_block)
 
-            scenario_nodes = []
-            for q in scenario_questions:
+            # Turn into ScenarioNode objects
+            nodes = []
+            for q in scenario_data:
                 node = ScenarioNode(
-                    question=q['question'],
-                    options=q['options'],
-                    correct_answer=q['correct_answer'],
-                    feedback=q['feedback']
+                    question=q["question"],
+                    options=q["options"],
+                    correct_answer=q["correct_answer"],
+                    feedback=q["feedback"]
                 )
-                scenario_nodes.append(node)
+                nodes.append(node)
 
             scenario_id = hashlib.md5(query.encode()).hexdigest()
-            self.scenarios[scenario_id] = scenario_nodes
+            self.scenarios[scenario_id] = nodes
 
             return {
                 "scenario_id": scenario_id,
-                "questions": [node.to_dict() for node in scenario_nodes]
+                "questions": [n.to_dict() for n in nodes]
             }
 
         except json.JSONDecodeError as e:
-            print(f"❌ JSON decode failed: {e}")
-            print(f"📦 Full AI Output:\n{scenario_data_str[:500]}")
+            print("❌ JSON parse error:", e)
+            print("⚠️ Raw Groq output:", raw_response[:500])
             return {
                 "error": "Failed to parse AI-generated scenario. Invalid JSON format.",
                 "details": str(e)
             }
+
         except Exception as e:
-            print(f"❌ Scenario generation error: {e}")
+            print("❌ Scenario generation error:", e)
             return {
                 "error": "Failed to generate scenario.",
                 "details": str(e)