Code Change

app.py

@@ -15,15 +15,16 @@ def home():
     return render_template('index.html')
 
 @app.route('/chat', methods=['POST'])
-def chat(user_input):
+def chat():
     user_message = request.json.get("message")
     
     if not user_message:
         return jsonify({"error": "Empty message received"})
 
     try:
+        formatted_prompt = f"User: {user_message}\nBot:"
         payload = {
-        "inputs": f"[INST] {user_input} [/INST]",
+        "inputs": formatted_prompt,
         "parameters": {
             "temperature": 0.5,   # 🔥 Controls randomness (lower = more deterministic)
             "top_p": 0.9,         # 🎯 Focus on high-probability words
@@ -46,7 +47,8 @@ def chat(user_input):
         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")
+        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})
 
     except Exception as e: