Update backend/main.py

backend/main.py

@@ -4,7 +4,8 @@ import os
 from dotenv import load_dotenv
 
 # Import our Agents
-from agents.scenario_agent import ScenarioAgent
+from agents.puzzle_agent import PuzzleAgent
+from agents.adventure_agent import AdventureAgent
 from agents.adaptive_agent import AdaptiveAgent
 from agents.hint_agent import HintAgent
 
@@ -15,7 +16,8 @@ app = Flask(__name__)
 CORS(app)  # Enable CORS for all routes
 
 # Initialize Agents
-scenario_agent = ScenarioAgent()
+puzzle_agent = PuzzleAgent()
+adventure_agent = AdventureAgent()
 adaptive_agent = AdaptiveAgent()
 hint_agent = HintAgent()
 
@@ -47,7 +49,11 @@ def generate_level():
     topic = data.get('topic') 
     
     try:
-        level_data = scenario_agent.generate_level(mode, difficulty, topic)
+        if mode == 'maze':
+            level_data = puzzle_agent.generate_level(difficulty, topic)
+        else:
+            level_data = adventure_agent.generate_level(mode, difficulty, topic)
+            
         if not level_data:
              raise Exception("Empty level data returned")
         return jsonify(level_data)
@@ -81,7 +87,11 @@ def level_feedback():
         
     try:
         # Teach the agent!
-        scenario_agent.learn_from_feedback(level_data, int(rating), developer_feedback)
+        if level_data.get('type') == 'maze':
+            puzzle_agent.learn_from_feedback(level_data, int(rating), developer_feedback)
+        else:
+            adventure_agent.learn_from_feedback(level_data, int(rating), developer_feedback)
+            
         return jsonify({"status": "learned", "message": "Thanks for the feedback!"})
     except Exception as e:
         print(f"Feedback Error: {e}")