Update app.py

app.py

@@ -107,7 +107,14 @@ if user_query is not None and user_query != "":
     response = get_response(user_query, st.session_state.chat_history)
 
     # Remove any unwanted prefixes from the response
-    response = response.replace("AI response:", "").replace("chat response:", "").replace("bot response:", "").strip()
+    response = (
+        response
+        .replace("AI response:", "")
+        .replace("chat response:", "")
+        .replace("bot response:", "")
+        .replace("Desired Assistant Response:", "")
+        .strip()
+    )
 
     with st.chat_message("AI"):
         st.write(response)