Code Change

app.py

@@ -47,11 +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")
+        #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 = raw_output.split("Bot:")[-1].strip()
         return jsonify({"reply": reply})
 
     except Exception as e: