Update app.py

app.py

@@ -46,9 +46,11 @@ def chat(
     output_text = tokenizer.decode(output[0], skip_special_tokens=True)
     if "AI:" in output_text:
         response = output_text.split("AI:")[-1].strip()
+        if "User:" in response:
+            response = response.split("User:")[0].strip()
     else:
         response = output_text.strip()
-    return response
+
 
 with gr.Blocks() as demo:
     gr.Markdown("# MindVR Therapy Chatbot\n\nDùng thử UI hoặc gọi API!")