Update app.py

app.py

@@ -25,32 +25,6 @@ print("-------")
 print("started")
 print("-------")
 
-async def echo(websocket):
-    global userData
-    async for message in websocket:
-        data = json.loads(message)
-        if not "message" in message:
-            return
-        if not "token" in message:
-            return
-        m = data["message"]
-        token = data["token"]
-        docs = retriever.get_relevant_documents(m)
-        response = conversational_rag_chain.invoke(
-            {"input": m},
-            config={
-                "configurable": {"session_id": token}
-            },
-        )["answer"]
-        await websocket.send(json.dumps({"response": response}))
-
-async def main():
-    async with serve(echo, "0.0.0.0", 7860):
-        await asyncio.Future()
-
-def f():
-    asyncio.run(main())
-Process(target=f).start()
 if not os.path.isdir('database'):
     os.system("unzip database.zip")
 
@@ -84,6 +58,33 @@ print("-------")
 print("Embeddings")
 print("-------")
 
+async def echo(websocket):
+    global retriever, conversational_rag_chain
+    async for message in websocket:
+        data = json.loads(message)
+        if not "message" in message:
+            return
+        if not "token" in message:
+            return
+        m = data["message"]
+        token = data["token"]
+        docs = retriever.get_relevant_documents(m)
+        response = conversational_rag_chain.invoke(
+            {"input": m},
+            config={
+                "configurable": {"session_id": token}
+            },
+        )["answer"]
+        await websocket.send(json.dumps({"response": response}))
+
+async def main():
+    async with serve(echo, "0.0.0.0", 7860):
+        await asyncio.Future()
+
+def f():
+    asyncio.run(main())
+Process(target=f).start()
+
 vectorstore = Chroma.from_documents(documents=splits, embedding=embedding)
 
 def format_docs(docs):