Update app.py

app.py

@@ -227,10 +227,13 @@ def get_response(message, history):
     response = conversation_rag_chain.invoke({
         "chat_history": history,
         "input": message,
-        "chat_message": message
+        "chat_message": message + " Assistant: "
     })
     #print("get_response " +response)
-    return response['answer']#[-1]['generation']['content']    
+    res = response['answer']
+    parts = res.split(" Assistant: ")
+    last_part = parts[-1]
+    return last_part#[-1]['generation']['content']    
 
     
 #####