Introduce Socratic tutor turn counting logic to force wrap-up after 2 hints and add corresponding tests

backend/main.py

@@ -127,10 +127,17 @@ def run_flow_b(message: str, api_key: str, history: Optional[List[ChatMessage]]
         generation_config={"response_mime_type": "application/json"}
     )
     
-    custom_system = (
-        "Socratic tutor: guide with clear, substantial hints. Continue unless close. "
-        "If close: give solution to the initial question & ask: 'Do you want to learn something else?'"
-    )
+    num_user_turns = sum(1 for m in history if m.role == "user") if history else 0
+    if num_user_turns >= 2:
+        custom_system = (
+            "Socratic tutor: You have already given multiple hints. Do not ask any more Socratic questions. "
+            "Provide the final direct answer/solution to the initial question immediately now, and ask: 'Do you want to learn something else?'"
+        )
+    else:
+        custom_system = (
+            "Socratic tutor: guide with clear, substantial hints. Continue unless close. "
+            "If close: give solution to the initial question & ask: 'Do you want to learn something else?'"
+        )
     
     tone_instruction = (
         "JSON: {\"s\":\"sentiment\",\"r\":\"reply\"}\n"

backend/test_sentiment.py

@@ -5,7 +5,7 @@ from unittest.mock import MagicMock, patch
 # Add current directory to path for imports
 sys.path.append(os.path.dirname(os.path.abspath(__file__)))
 
-from main import run_flow_b
+from main import run_flow_b, ChatMessage
 
 def test_run_flow_b_parsing():
     print("Testing run_flow_b parsing and extraction logic...")
@@ -30,5 +30,34 @@ def test_run_flow_b_parsing():
         
         print("Mocked run_flow_b test passed!")
 
+def test_run_flow_b_force_wrap_up():
+    print("Testing force wrap-up system prompt switch...")
+    
+    mock_response = MagicMock()
+    mock_response.content = '{"s": "neutral", "r": "The answer is 4. Do you want to learn something else?"}'
+    
+    with patch("main.ChatGoogleGenerativeAI") as MockLLM:
+        mock_llm_instance = MagicMock()
+        mock_llm_instance.invoke.return_value = mock_response
+        MockLLM.return_value = mock_llm_instance
+        
+        history = [
+            ChatMessage(role="user", content="What is 2x = 4?"),
+            ChatMessage(role="assistant", content="What do you think?"),
+            ChatMessage(role="user", content="I don't know"),
+            ChatMessage(role="assistant", content="Try dividing both sides by 2.")
+        ]
+        
+        state, reply, context, est_in, est_out = run_flow_b(
+            message="Still confused.",
+            api_key="fake-api-key",
+            history=history
+        )
+        
+        # Verify that the force wrap-up system prompt was selected
+        assert "do not ask any more socratic questions" in context.lower()
+        print("Mocked force wrap-up test passed!")
+
 if __name__ == "__main__":
     test_run_flow_b_parsing()
+    test_run_flow_b_force_wrap_up()