Update app.py

app.py

@@ -86,7 +86,8 @@ qa_prompt = ChatPromptTemplate.from_messages(
     [
         ("system", qa_system_prompt),
         MessagesPlaceholder("chat_history"),
-        ("user", "{input} Assistant: "),
+        ("user", "{input}"),
+        ("assistant", "")
     ]
 )
 question_answer_chain = create_stuff_documents_chain(llm, qa_prompt)
@@ -131,7 +132,7 @@ async def echo(websocket):
         )["answer"]
         if response.startswith("? "):
             response.replace("? ", "")
-        if response.lower().startswith("assistant: ") or response.lower().startswith("ai: "):
+        if response.lower().replace("\n", "").startswith("assistant: ") or response.lower().replace("\n", "").startswith("ai: "):
             response.replace("assistant: ", "").replace("ai: ", "")
         await websocket.send(json.dumps({"response": response}))