Fix response cleaning to remove special tokens

app.py

@@ -97,8 +97,22 @@ def chat():
         )
         
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        # Extract only the new response
-        response = response.replace(combined_prompt, "").split(".")[0].strip()
+        
+        # Extract only the new AI response
+        # Split by <AI> and get the last response
+        if "<AI>" in response:
+            response = response.split("<AI>")[-1].strip()
+        
+        # Remove any <user> tags if they appear (model might generate them)
+        if "<user>" in response:
+            response = response.split("<user>")[0].strip()
+        
+        # Clean up any remaining special tokens
+        response = response.replace("<AI>", "").replace("<user>", "").strip()
+        
+        # If empty response, provide a default
+        if not response:
+            response = "I'm not sure how to respond to that."
         
         return jsonify({
             "response": response,