Update app.py

app.py

@@ -80,8 +80,8 @@ history_aware_retriever = create_history_aware_retriever(
 qa_system_prompt = """You are an assistant for question-answering tasks.
 Use the following pieces of retrieved context to answer the question.
 If you don't know the answer, just say that you don't know.
-Use three sentences minimum and include a lot of information that the user might need.
-Make it readable with lots of newlines in between sentences.
+
+Make it and use markdown with spaces in between sentences.
 
 {context}"""
 qa_prompt = ChatPromptTemplate.from_messages(
@@ -125,17 +125,20 @@ async def echo(websocket):
         m = data["message"] + "\nAssistant: "
         token = data["token"]
         docs = retriever.get_relevant_documents(m)
-        response = conversational_rag_chain.invoke(
+        rawresponse = conversational_rag_chain.invoke(
             {"input": m},
             config={
                 "configurable": {"session_id": token}
             },
-        )["answer"]
+        )
+        response = rawresponse["answer"]
         response = response.replace("Assistant: ", "").replace("AI: ", "")
         response.strip()
+        response = response.split("Human:")[0]
         while response.startswith("\n"):
             response = response[1:]
         await websocket.send(json.dumps({"response": response}))
+        print(json.dumps(rawresponse))
 
 async def main():
     async with serve(echo, "0.0.0.0", 7860):