Update backend/agents/puzzle_agent.py

backend/agents/puzzle_agent.py

@@ -97,6 +97,7 @@ class PuzzleAgent:
             self.model = genai.GenerativeModel("gemini-1.5-flash-latest")
 
         # Load the 500-scenario dataset
+        # Look for the dataset inside the backend folder (safer for HuggingFace deployment)
         base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
         path = os.path.join(base_dir, "maze_scenarios_500.json")
         self.scenarios: List[Dict] = []
@@ -286,8 +287,10 @@ class PuzzleAgent:
         progression_str = ""
         for i, (scen, cg) in enumerate(zip(scenarios, compiled_grids)):
             t = scen.get("theme", "space").capitalize()
+            title = scen.get("title", f"Level {i+1}")
+            base_story = scen.get("story", "Solve the maze.")
             s = " → ".join(cg["ordered_steps"]) or "reach the exit"
-            progression_str += f"Level {i+1}: Theme = {t}, Steps = {s}\n"
+            progression_str += f"Level {i+1}:\n- Theme: {t}\n- Basic Concept: {title} - {base_story}\n- Steps: {s}\n\n"
 
         prompt = f"""
 You are the Mission Architect for CodeCracker, a STEM education app for kids.
@@ -302,7 +305,7 @@ LEARNING GOAL: {concept['description']}
 STRICT RULES:
 1. The story MUST explain the DAA concept clearly to a student (age 10-16).
 2. Reference the exact mission steps for that level in its goal_description.
-3. CRITICAL: Provide a UNIQUE programming/logic problem in the `story` field for EACH level.
+3. CRITICAL: Read the "Basic Concept" provided for each level in the progression, and expand it into a UNIQUE, fully-fleshed out programming/logic problem in the `story` field for EACH level. Do NOT just repeat the basic concept.
 4. The `title` of each level should match its assigned Theme (e.g., "Bank Heist", "Space Station").
 5. CRITICAL: Generate a COMPLETELY UNIQUE `hint` for EVERY level. The hint MUST be different for Level 1, Level 2, etc., and must directly reference the specific theme/logic of that exact level! Do NOT repeat hints!