Spaces:

utkarsh1797
/

ai-chatbot

Running

Utkarsh Verma commited on Apr 12

Commit

326eced

1 Parent(s): afc89dd

Changes in UI

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,6 +51,7 @@ def chat():
             bot_reply = bot_reply.replace(user_message, "").strip()
         # Clean extra prompt echoes like "User:"
         cleaned_reply = bot_reply.replace("User:", "").replace("You:", "").replace("Bot:", "").strip()
         return jsonify({"reply": cleaned_reply})
     except Exception as e:

             bot_reply = bot_reply.replace(user_message, "").strip()
         # Clean extra prompt echoes like "User:"
         cleaned_reply = bot_reply.replace("User:", "").replace("You:", "").replace("Bot:", "").strip()
+        cleaned_reply = cleaned_reply.rstrip("User").rstrip("Bot").strip()
         return jsonify({"reply": cleaned_reply})
     except Exception as e: