Code Change

app.py

@@ -46,12 +46,8 @@ def chat(user_message):
         
         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")
-        # Extract only the part after "Bot:"
-        reply = raw_output.split("Bot:")[-1].strip()
-
-        #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")
         return jsonify({"reply": reply})
 
     except Exception as e: