Code Change

app.py

@@ -29,7 +29,7 @@ def chat():
             "temperature": 0.5,   # 🔥 Controls randomness (lower = more deterministic)
             "top_p": 0.9,         # 🎯 Focus on high-probability words
             "max_new_tokens": 50,  # ⏳ Limits response length
-            "stop_sequences": ["User:"]  # ⛔ Stops response at natural points
+            "stop_sequences": ["User:", "You:", "user:"]  # ⛔ Stops response at natural points
         }
     }
         response = requests.post(API_URL, headers=headers, json=payload)
@@ -47,10 +47,23 @@ def chat():
         if "error" in data:
             return jsonify({"reply": f"Error: {data['error']}"})
 
-        raw_output = data[0]['generated_text'] if isinstance(data, list) else data.get('generated_text', "No response")
-        reply = raw_output.split("Bot:")[-1].strip()
+        # raw_output = data[0]['generated_text'] if isinstance(data, list) else data.get('generated_text', "No response")
+        # reply = raw_output.split("Bot:")[-1].strip()
 
         
+        # return jsonify({"reply": reply})
+
+        raw_output = data[0]['generated_text'] if isinstance(data, list) else data.get('generated_text', "No response")
+        # Clean up output
+        if "Bot:" in raw_output:
+            reply = raw_output.split("Bot:")[-1].strip()
+        else:
+            reply = raw_output.strip()
+
+        for trailing in ["User", "You", "User:", "You:"]:
+            if reply.endswith(trailing):
+                reply = reply.rsplit(trailing, 1)[0].strip()
+
         return jsonify({"reply": reply})
 
     except Exception as e: