Update app.py

app.py

@@ -125,6 +125,10 @@ async def echo(websocket):
     global retriever, conversational_rag_chain
     async for message in websocket:
         data = json.loads(message)
+        if data["message"]:
+            response = store
+            await websocket.send(json.dumps({"response": response}))
+            break
         if not "message" in message:
             return
         if not "token" in message: