Code Change

app.py

@@ -47,7 +47,11 @@ def chat():
         if "error" in data:
             return jsonify({"reply": f"Error: {data['error']}"})
 
-        reply = data[0]['generated_text'] if isinstance(data, list) else data.get('generated_text', "No response")
+        #reply = data[0]['generated_text'] if isinstance(data, list) else data.get('generated_text', "No response")
+
+        raw_output = data[0]['generated_text'] if isinstance(data, list) else data.get('generated_text', "No response")
+        # Extract only the part after "Bot:"
+        reply = raw_output.split("Bot:")[-1].strip()
         return jsonify({"reply": reply})
 
     except Exception as e: