Utkarsh Verma commited on
Commit
95ae9c6
·
1 Parent(s): 0637ee1

Changes in UI

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -47,7 +47,8 @@ def chat():
47
  bot_reply = data['generated_text']
48
  else:
49
  bot_reply = "\ud83e\udde0 Sorry, I didn't understand that."
50
-
 
51
  # Clean extra prompt echoes like "User:"
52
  cleaned_reply = bot_reply.replace("User:", "").replace("You:", "").replace("Bot:", "").strip()
53
  return jsonify({"reply": cleaned_reply})
 
47
  bot_reply = data['generated_text']
48
  else:
49
  bot_reply = "\ud83e\udde0 Sorry, I didn't understand that."
50
+ if user_message in bot_reply:
51
+ bot_reply = bot_reply.replace(user_message, "").strip()
52
  # Clean extra prompt echoes like "User:"
53
  cleaned_reply = bot_reply.replace("User:", "").replace("You:", "").replace("Bot:", "").strip()
54
  return jsonify({"reply": cleaned_reply})