Update app.py

app.py

@@ -191,8 +191,11 @@ def get_response(user_input):
         "chat_history": chat_history,
         "input": user_input
     })
-    
-    return response['answer']
+    #print("get_response " +response)
+    res = response['answer']
+    parts = res.split(" Assistant: ")
+    last_part = parts[-1]
+    return last_part
 
 
 def history_to_dialog_format(chat_history: list[str]):
@@ -226,7 +229,7 @@ def get_response(message, history):
     
     response = conversation_rag_chain.invoke({
         "chat_history": history,
-        "input": message,
+        "input": message  + " Assistant: ",
         "chat_message": message + " Assistant: "
     })
     #print("get_response " +response)