Update scenario_builder.py

scenario_builder.py

@@ -6,7 +6,7 @@ from typing import Dict, Any, List
 class ScenarioNode:
     def __init__(self, question: str, options: Dict[str, Any], correct_answer: str = None, feedback: str = None):
         self.question = question
-        self.options = options  # {label: next_node_id or action_fn_name}
+        self.options = options
         self.correct_answer = correct_answer
         self.feedback = feedback
 
@@ -23,75 +23,72 @@ class ScenarioBuilder:
         self.ai_engine = ai_engine
         self.scenarios = {}
 
-    import json
-    
     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.
-    """
+        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.
+        """
 
-    try:
-        response = self.ai_engine.generate_raw_text(prompt)
-        scenario_data_str = response.strip()
+        try:
+            response = self.ai_engine.generate_raw_text(prompt)
+            scenario_data_str = response.strip()
 
-        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()
+            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()
 
-        print("🧪 Raw AI response preview:", scenario_data_str[:200])
+            print("🧪 Raw AI response preview:", scenario_data_str[:200])
 
-        scenario_questions = json.loads(scenario_data_str)
+            scenario_questions = json.loads(scenario_data_str)
 
-        scenario_nodes = []
-        for q in scenario_questions:
-            node = ScenarioNode(
-                question=q['question'],
-                options=q['options'],
-                correct_answer=q['correct_answer'],
-                feedback=q['feedback']
-            )
-            scenario_nodes.append(node)
+            scenario_nodes = []
+            for q in scenario_questions:
+                node = ScenarioNode(
+                    question=q['question'],
+                    options=q['options'],
+                    correct_answer=q['correct_answer'],
+                    feedback=q['feedback']
+                )
+                scenario_nodes.append(node)
 
-        scenario_id = hashlib.md5(query.encode()).hexdigest()
-        self.scenarios[scenario_id] = scenario_nodes
+            scenario_id = hashlib.md5(query.encode()).hexdigest()
+            self.scenarios[scenario_id] = scenario_nodes
 
-        return {
-            "scenario_id": scenario_id,
-            "questions": [node.to_dict() for node in scenario_nodes]
-        }
-
-    except json.JSONDecodeError as e:
-        print(f"❌ JSON decode failed: {e}")
-        print(f"📦 Full AI Output:\n{scenario_data_str[: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}")
-        return {
-            "error": "Failed to generate scenario.",
-            "details": str(e)
-        }
+            return {
+                "scenario_id": scenario_id,
+                "questions": [node.to_dict() for node in scenario_nodes]
+            }
 
+        except json.JSONDecodeError as e:
+            print(f"❌ JSON decode failed: {e}")
+            print(f"📦 Full AI Output:\n{scenario_data_str[: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}")
+            return {
+                "error": "Failed to generate scenario.",
+                "details": str(e)
+            }
 
     def get_scenario(self, scenario_id: str) -> List[ScenarioNode]:
         return self.scenarios.get(scenario_id)