Update backend/agents/hint_agent.py

backend/agents/hint_agent.py

@@ -19,46 +19,64 @@ class HintAgent:
             self.model = genai.GenerativeModel('gemini-1.5-flash')
 
     def generate_hint(self, 
+                     mode: str,
                      level_context: Dict[str, Any], 
-                     user_code: str, 
-                     error_message: Optional[str] = None) -> str:
+                     user_state: Dict[str, Any]) -> str:
         """
         Generates a hint based on the level and user's current attempt.
         """
         if not hasattr(self, 'model'):
              return "SYSTEM ERROR: API Key missing for Hint Agent."
 
-        prompt = f"""
-        You are a coding tutor. The student is stuck.
-        
-        LEVEL CONTEXT:
-        Title: {level_context.get('title')}
-        Goal: {level_context.get('goal_description') or level_context.get('problem')}
-        
-        USER'S CURRENT CODE:
-        {user_code}
-        
-        ERROR MESSAGE (if any):
-        {error_message}
-        
-        TASK:
-        Provide a helpful, encouraging hint. 
-        DO NOT give the full solution immediately unless they are very stuck.
-        If there is a syntax error, point it out gently.
-        Keep it under 2 sentences.
-        """
+        if mode == "maze":
+            prompt = f"""
+            You are a coding tutor helping a student navigate a maze in a STEM app. The student is asking for a hint.
+            
+            LEVEL CONTEXT:
+            Title: {level_context.get('title')}
+            Goal: {level_context.get('goal_description') or level_context.get('problem')}
+            
+            USER'S CURRENT STATE:
+            Current Position (Row, Col): {user_state.get('current_position')}
+            Target Objective they need to reach next: {user_state.get('next_objective')}
+            
+            TASK:
+            Provide a helpful, encouraging hint. 
+            Tell them what they should look for next or what they might have missed (e.g. "Make sure you grab the Key before heading to the Exit! Look around the top right area.").
+            Keep it under 2 sentences. DO NOT give them exact turn-by-turn answers.
+            """
+        else:
+            prompt = f"""
+            You are a coding tutor. The student is stuck on a Blockly coding challenge.
+            
+            LEVEL CONTEXT:
+            Title: {level_context.get('title')}
+            Goal: {level_context.get('goal_description') or level_context.get('problem')}
+            
+            USER'S CURRENT CODE:
+            {user_state.get('user_code')}
+            
+            ERROR MESSAGE (if any):
+            {user_state.get('error')}
+            
+            TASK:
+            Provide a helpful, encouraging hint. 
+            DO NOT give the full solution immediately unless they are very stuck.
+            If there is a syntax error or logical flaw, point it out gently.
+            Keep it under 2 sentences.
+            """
         
         try:
             response = self.model.generate_content(prompt)
             return response.text.strip()
         except Exception as e:
-            # Fallback Hint (Generic but helpful)
-            return "Try breaking the problem down into smaller steps. Check if your loops and variables are set correctly!"
+            # Fallback Hint
+            return "Try breaking the problem down into smaller steps. Check your logic again!"
 
 if __name__ == "__main__":
     from dotenv import load_dotenv
     load_dotenv()
     agent = HintAgent()
-    context = {"title": "Mars Rover", "goal_description": "Move forward 3 times"}
-    code = "move_forward\nturn_left"
-    print(agent.generate_hint(context, code))
+    context = {"title": "Mars Rover", "problem": "Move forward 3 times"}
+    state = {"user_code": "move_forward\nturn_left", "error": "Did not reach the goal"}
+    print(agent.generate_hint("blockly", context, state))