Update backend/agents/scenario_agent.py

backend/agents/scenario_agent.py

@@ -164,8 +164,8 @@ class ScenarioAgent:
         base_prompt = f"""
         You are the Lead Game Designer for 'CodeCracker', an educational coding game for ages 10-14.
         Your task is to generate a UNIQUE STORY ARC of EXACTLY 5 interconnected game levels in JSON format.
-        The story MUST be based on STEM topics or real-life practical scenarios.
-        The levels should progress the story narratively from level 1 to the final level 5.
+        The story MUST be a single cohesive narrative that progresses from level 1 to level 5.
+        Each level's 'story' field should build upon the previous one.
         
         CURRENT SETTINGS:
         - Mode: {mode}
@@ -180,7 +180,7 @@ class ScenarioAgent:
         {self._get_few_shot_examples(mode)}
         
         OUTPUT FORMAT: RAW JSON ONLY (No Markdown).
-        The JSON must be an object with "story_arc_title" and a "levels" array containing the individual levels.
+        The JSON must be an object with "story_arc_title" and a "levels" array containing EXACTLY 5 levels.
         """
         
         if mode == "maze":
@@ -370,22 +370,88 @@ class ScenarioAgent:
         elif mode == "blockly":
             return {
                 "story_arc_title": "Logic Repair Arc",
-                "levels": [{
-                    "type": "blockly",
-                    "level_id": "fallback_002",
-                    "title": "Logic Repair",
-                    "story": "The automated systems are down. We need manual logic configuration.",
-                    "concept_tutorial": "Loops: Use loops to repeat actions without writing code over and over.",
-                    "problem": "Create a loop that counts to 3.",
-                    "toolbox_categories": ["Loops", "Math", "Variables"],
-                    "initial_code": "",
-                    "validation_rules": {
-                        "required_concepts": ["loop"],
-                        "expected_output": "3"
+                "levels": [
+                    {
+                        "type": "blockly",
+                        "level_id": "fallback_b_01",
+                        "title": "System Reboot",
+                        "story": "The mainframe is unresponsive. We need to send a simple '3' signal to wake it up.",
+                        "concept_tutorial": "Variables & Output: We use code to send values to the system.",
+                        "problem": "Print the number 3.",
+                        "toolbox_categories": ["Text", "Math"],
+                        "initial_code": "",
+                        "validation_rules": {
+                            "required_concepts": ["print"],
+                            "expected_output": "3"
+                        },
+                        "hint_1": "Use the 'print' block.",
+                        "hint_2": "Snap a number block '3' inside it."
                     },
-                    "hint_1": "Use the 'repeat' block.",
-                    "hint_2": "Set the number to 3."
-                }]
+                    {
+                        "type": "blockly",
+                        "level_id": "fallback_b_02",
+                        "title": "Data Stream",
+                        "story": "Mainframe is awake! Now we need to repeat the signal 3 times to stabilize the connection.",
+                        "concept_tutorial": "Loops: Use loops to repeat actions without writing code over and over.",
+                        "problem": "Create a loop that prints '3' three times.",
+                        "toolbox_categories": ["Loops", "Text", "Math"],
+                        "initial_code": "",
+                        "validation_rules": {
+                            "required_concepts": ["loop", "print"],
+                            "expected_output": "3,3,3"
+                        },
+                        "hint_1": "Use the 'repeat' block.",
+                        "hint_2": "Set the loop count to 3."
+                    },
+                    {
+                        "type": "blockly",
+                        "level_id": "fallback_b_03",
+                        "title": "Oxygen Check",
+                        "story": "Connection stable. Now check the oxygen levels. If they are below 20, we need to print 'Warning'.",
+                        "concept_tutorial": "Logic: Use 'If' blocks to make decisions based on values.",
+                        "problem": "Check if variable 'oxygen' (set to 15) is less than 20. If yes, print 'Warning'.",
+                        "toolbox_categories": ["Logic", "Variables", "Text", "Math"],
+                        "initial_code": "oxygen = 15",
+                        "validation_rules": {
+                            "required_concepts": ["if", "print"],
+                            "expected_output": "Warning"
+                        },
+                        "hint_1": "Use an 'if' block from Logic.",
+                        "hint_2": "Compare oxygen < 20."
+                    },
+                    {
+                        "type": "blockly",
+                        "level_id": "fallback_b_04",
+                        "title": "Security Key",
+                        "story": "Systems are green. Final step: generate a security key by adding 5 to the secret number 10.",
+                        "concept_tutorial": "Math: Computers are great at calculations.",
+                        "problem": "Add 5 to 10 and print the result.",
+                        "toolbox_categories": ["Math", "Text"],
+                        "initial_code": "",
+                        "validation_rules": {
+                            "required_concepts": ["print"],
+                            "expected_output": "15"
+                        },
+                        "hint_1": "Use the + block from Math.",
+                        "hint_2": ""
+                    },
+                    {
+                        "type": "blockly",
+                        "level_id": "fallback_b_05",
+                        "title": "Mission Complete",
+                        "story": "Access granted! Welcome back, Commander. Print 'Ready' to finish.",
+                        "concept_tutorial": "Final Review: You've mastered the basics of logic and calculation.",
+                        "problem": "Print 'Ready'.",
+                        "toolbox_categories": ["Text"],
+                        "initial_code": "",
+                        "validation_rules": {
+                            "required_concepts": ["print"],
+                            "expected_output": "Ready"
+                        },
+                        "hint_1": "One last print block!",
+                        "hint_2": ""
+                    }
+                ]
             }
         elif mode == "time_challenge":
             return {
@@ -420,4 +486,4 @@ if __name__ == "__main__":
     load_dotenv()
     agent = ScenarioAgent()
     print("Testing Maze Generation...")
-    print(json.dumps(agent.generate_level("maze", "Easy"), indent=2))
+    print(json.dumps(agent.generate_level("maze", "Easy"), indent=2))