Enforce strict Socratic tutor wrap-up and termination rules when the problem is solved or limits are reached

backend/main.py

@@ -131,10 +131,12 @@ def run_flow_b(message: str, api_key: str, history: Optional[List[ChatMessage]]
     custom_system = (
         f"Socratic tutor: guide with clear, substantial hints. "
         f"Current conversation: {num_user_turns} user turns so far. "
-        "Dynamically decide whether to nudge or give the final direct solution. "
-        "Provide the final direct answer/solution immediately if: the user is close, highly frustrated, "
-        "asks directly for the answer, or if hints exceed topic complexity (e.g., 2 hints for simple topics, 4 for complex topics). "
-        "Otherwise, continue guiding with a hint and ask exactly 1 Socratic question."
+        "CRITICAL RULES:\n"
+        "- If the user has answered correctly, solved the problem, or if hints exceed topic complexity "
+        "(e.g., 2 user turns for simple topics, 4 user turns for complex topics): DO NOT ask any more math/science/concept questions. "
+        "Immediately confirm their success (or provide the direct solution) and ask exactly: 'Do you want to learn something else?'\n"
+        "- If the user is close, highly frustrated, or asks directly: give the solution and ask exactly: 'Do you want to learn something else?'\n"
+        "- Otherwise, guide with a hint and ask exactly 1 Socratic question."
     )
     
     tone_instruction = (
@@ -144,7 +146,7 @@ def run_flow_b(message: str, api_key: str, history: Optional[List[ChatMessage]]
         "- Sympathize with s implicitly (tone/style); never name or mention the sentiment/emotion itself.\n"
         "- NEVER use 'if you' (use direct phrasing: 'think about', 'imagine').\n"
         "- Ask 1 question max.\n"
-        "Responses:\n"
+        "Responses (unless wrapping up / giving final answer):\n"
         "- frustration: acknowledge sentiment but not explicitly + simplify + question.\n"
         "- starting_to_get_bored: acknowledge sentiment but not explicitly + puzzle/analogy + question.\n"
         "- other: hint + question."

backend/test_sentiment.py

@@ -56,7 +56,7 @@ def test_run_flow_b_force_wrap_up():
         
         # Verify that turn counting and dynamic decision instructions are injected correctly
         assert "current conversation: 2 user turns so far" in context.lower()
-        assert "dynamically decide" in context.lower()
+        assert "critical rules" in context.lower()
         print("Mocked force wrap-up test passed!")
 
 if __name__ == "__main__":