Update app.py

app.py

@@ -121,7 +121,7 @@ async def echo(websocket):
             return
         if not "token" in message:
             return
-        m = data["message"] + "\n\nAssistant: "
+        m = data["message"] + "\nAssistant: "
         token = data["token"]
         docs = retriever.get_relevant_documents(m)
         response = conversational_rag_chain.invoke(
@@ -130,8 +130,8 @@ async def echo(websocket):
                 "configurable": {"session_id": token}
             },
         )["answer"]
-        response.replace("Assistant:", "").replace("AI:", "")
-        response.replace("\n", "")
+        response = response.replace("Assistant:", "").replace("AI:", "")
+        response = response.replace("\n", "")
         response.strip()
         await websocket.send(json.dumps({"response": response}))