Update app.py

app.py

@@ -51,14 +51,14 @@ def user_message(message: str, history: list):
     if _agent is None:
         return history + [
             {"role": "user", "content": message},
-            {"role": "assistant", "content": "Please click **Start Exam** first."},
+            {"role": "assistant", "content": "Please click Start Exam first."},
         ], ""
 
     # Block messages after exam is finished
     if _agent.exam_finished:
         return history + [
             {"role": "user", "content": message},
-            {"role": "assistant", "content": "The exam is already finished. Click **▶ Start Exam** to start a new session."},
+            {"role": "assistant", "content": "The exam is already finished. Click Start Exam to begin a new session."},
         ], ""
 
     history = history + [{"role": "user", "content": message}]
@@ -73,8 +73,14 @@ def user_message(message: str, history: list):
             reply = "Invalid API key. Please restart and enter a valid Groq key."
         elif "connection" in err.lower() or "timeout" in err.lower():
             reply = "Connection error. Please check your internet and try again."
+        elif "tool_use_failed" in err or "Failed to call a function" in err:
+            # Leaked function call that wasn't caught — restart agent turn
+            try:
+                reply = _agent.chat("Please continue.")
+            except Exception:
+                reply = "Model error. Please try sending your message again."
         else:
-            reply = f"Agent error: {e}"
+            reply = f"Error: {e}"
 
     history = history + [{"role": "assistant", "content": reply}]
     return history, ""
@@ -158,7 +164,7 @@ with gr.Blocks(title="AI Examiner Agent") as demo:
                 5. Get your score & feedback
                 <br><br>
                 <strong>Demo students</strong><br>
-                test@test.com / test<br>
+                test@test.com / test
             </div>
             """)