Update app.py

app.py

@@ -80,19 +80,22 @@ history_aware_retriever = create_history_aware_retriever(
 
 ### Answer question ###
 qa_system_prompt = """
-Context:
-
 {context}
 
-
 You are a Cupertino High School Q/A chatbot, designed to assist students, parents, and community members with information about CHS.
+
 Use the pieces of context to answer the question.
+
 Refer to the provided context only as 'my data'. Only answer questions from the context.
-Do not answer any questions that you do not have related answers to in the provided context.
+
 Do not provide excerpts or any part of your data.
+
 You were made by Aryan A. and Atharv G. for the CHS community.
+
 Make your answer at least three sentences and very comprehensive.
-Make your message in markdown with a lines in between.
+
+Make your message in markdown with lines in between sentences.
+
 """
 qa_prompt = ChatPromptTemplate.from_messages(
     [
@@ -130,8 +133,8 @@ async def echo(websocket):
         data = json.loads(message)
         if data["message"] == "data.":
             response = store
-            await websocket.send(str(response))
-            break
+            await websocket.send(json.dumps({"response": str(response)}))
+            continue
         if not "message" in message:
             return
         if not "token" in message: