Update backend/agents/puzzle_agent.py

backend/agents/puzzle_agent.py

@@ -31,7 +31,16 @@ class PuzzleAgent:
         }
 
         self.high_rated_levels = [] 
-
+        
+        # Load the 500 Logic Scenarios
+        self.scenarios_path = os.path.join(os.getcwd(), "data", "maze_scenarios_500.json")
+        self.scenarios = []
+        try:
+            with open(self.scenarios_path, 'r') as f:
+                data = json.load(f)
+                self.scenarios = data.get("scenarios", [])
+        except Exception as e:
+            print(f"WARNING: Could not load maze_scenarios_500.json: {e}")
     def _generate_maze_grid(self, size: int) -> list:
         """Generates a perfect, guaranteed solvable maze using Recursive Backtracking."""
         # Force size to be odd for perfect wall alignment
@@ -82,10 +91,22 @@ class PuzzleAgent:
             if len(self.high_rated_levels) > 50: self.high_rated_levels.pop(0)
             self.high_rated_levels.append(level_data)
 
-    def _get_system_prompt(self, difficulty: str, specific_topic: str, generated_grids: list) -> str:
+    def _get_system_prompt(self, difficulty: str, specific_topic: str, generated_grids: list, scenario_context: Dict = None) -> str:
+        mission_context = ""
+        if scenario_context:
+            mission_context = f"""
+            MISSION SCENARIO:
+            - Goal: {scenario_context.get('title')}
+            - Objective: {scenario_context.get('story')}
+            - Logic Steps: {', '.join(scenario_context.get('required_steps', []))}
+            - Objective: You MUST incorporate these logic steps into the 5-level story arc.
+            """
+
         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.
+        Your task is to generate a UNIQUE MISSION ARC of EXACTLY 5 interconnected game levels in JSON format.
+        
+        {mission_context}
         
         CURRENT SETTINGS:
         - Mode: maze
@@ -100,43 +121,45 @@ class PuzzleAgent:
         Grid 4: {json.dumps(generated_grids[3])}
         Grid 5: {json.dumps(generated_grids[4])}
         
+        TILE ID SCHEMA:
         - 0 = Path
         - 1 = Wall
-        - 2 = Start
-        - 3 = Goal
-        - 4 = Hazard / Killing Point (MUST be avoided or jumped over!)
+        - 2 = Start (Captain Cody)
+        - 3 = Goal / Final Mission Point
+        - 4 = Hazard (☠️ Skull) - Must be avoided/jumped!
+        - 5 = Mission Item (Locker/Key/Terminal) - Must be visited!
+        - 8 = NPC / Expert (Manager/Scientist) - Must be visited!
         
-        You MUST assign Grid 1 to Level 1's grid_layout, Grid 2 to Level 2's grid_layout, and so on.
-        You must write engaging stories, hints, and goals explaining the hazards (the 4s).
+        LOGIC RULES:
+        1. Level 1 must introduce the mission.
+        2. Progressively place Tile IDs 5 and 8 in the grids for later levels to represent mission steps.
+        3. You MUST assign Grid 1 to Level 1, Grid 2 to Level 2, etc.
         
         DYNAMIC PROCEDURAL UI INSTRUCTION (THEME PALETTE):
-        You must act as a UI Designer. Based on the {specific_topic} story, generate a modern Glassmorphism hex color palette. 
-        It MUST be returned in the root JSON.
-        
-        OUTPUT FORMAT: RAW JSON ONLY (No Markdown).
+        You must act as a UI Designer. Generate a modern Glassmorphism hex color palette based on the {specific_topic} and {scenario_context.get('theme') if scenario_context else 'Space'} story.
         
         REQUIRED JSON STRUCTURE:
         {{
             "story_arc_title": "Epic Title",
             "theme_palette": {{
-                "background_gradient": ["#0F2027", "#203A43"], // 2 to 3 colors
-                "wall_color": "#00FFCC",
-                "path_color": "#121212",
-                "hazard_glow": "#FF0055"
+                "background_gradient": ["#HEX", "#HEX"],
+                "wall_color": "#HEX",
+                "path_color": "#HEX",
+                "hazard_glow": "#HEX"
             }},
             "levels": [
                 {{
                     "type": "maze",
-                    "level_id": "UUID_STRING",
-                    "title": "Level 1: The Breach",
-                    "story": "Engaging scenario explaining the objective and hazards.",
-                    "concept_tutorial": "EDUCATIONAL: Explain the CS concept (Sequencing, Loops) in 2 simple sentences.",
-                    "goal_description": "What to do.",
-                    "grid_layout": [[...]], // MUST USE THE EXACT GRID PROVIDED FOR THIS LEVEL (e.g. Grid 1 for Level 1)
+                    "level_id": "UUID",
+                    "title": "Level Name",
+                    "story": "Mission-driven story based on the scenario steps.",
+                    "concept_tutorial": "Explain the logic concept.",
+                    "goal_description": "Objective (e.g. 'Go to the Manager (8) first, then the Locker (5)')",
+                    "grid_layout": [[...]], // MUST USE PROVIDED GRIDS
                     "allowed_blocks": ["move_forward", "turn_left", "turn_right", "jump"],
                     "optimal_steps": 10,
-                    "hint_1": "General hint",
-                    "hint_2": "Specific hint about jumping the hazard (if any)"
+                    "hint_1": "Logic hint",
+                    "hint_2": "Hazard hint"
                 }}
             ]
         }}
@@ -144,18 +167,19 @@ class PuzzleAgent:
         return base_prompt
 
     def generate_level(self, difficulty: str, specific_topic: Optional[str] = None) -> Dict[str, Any]:
-        if not specific_topic:
-            category = random.choice(list(self.topics.keys()))
-            specific_topic = random.choice(self.topics[category])
+        # Pick a random Logic Scenario context from the JSON
+        scenario = {}
+        if self.scenarios:
+            filtered = [s for s in self.scenarios if s.get('difficulty', '').lower() == difficulty.lower()]
+            scenario = random.choice(filtered if filtered else self.scenarios)
+            specific_topic = scenario.get('theme', specific_topic)
 
         # Generate 5 progressively harder mathematical grids
-        base_size = 5
-        if difficulty == "Intermediate": base_size = 7
-        elif difficulty == "Expert": base_size = 9
+        base_size = scenario.get('grid_size', 5) if scenario else 5
         
         perfect_grids = []
         for i in range(5):
-            size = base_size + (2 if i >= 3 else 0) # Level 4 and 5 get larger
+            size = base_size + (2 if i >= 3 else 0)
             perfect_grids.append(self._generate_maze_grid(size))
 
         cache_key = f"maze_{difficulty}_{specific_topic}"
@@ -164,7 +188,7 @@ class PuzzleAgent:
             if time.time() - entry['timestamp'] < self.cache_duration:
                 return entry['data']
         
-        prompt = self._get_system_prompt(difficulty, specific_topic, perfect_grids)
+        prompt = self._get_system_prompt(difficulty, specific_topic, perfect_grids, scenario)
 
         try:
             if not self.model: raise Exception("No API Key")