Update app.py

app.py

@@ -1,4 +1,3 @@
-print("eeeh")
 import asyncio
 import json
 from websockets.server import serve
@@ -27,6 +26,7 @@ print("started")
 print("-------")
 
 async def echo(websocket):
+    global userData
     async for message in websocket:
         data = json.loads(message)
         if not "message" in message:
@@ -36,7 +36,6 @@ async def echo(websocket):
         m = data["message"]
         token = data["token"]
         docs = retriever.get_relevant_documents(m)
-        userData[token]["docs"] = str(docs)
         response = conversational_rag_chain.invoke(
             {"input": m},
             config={
@@ -49,6 +48,7 @@ async def main():
     async with serve(echo, "0.0.0.0", 7860):
         await asyncio.Future()
 def g():
+    global retriever, conversational_rag_chain
     if not os.path.isdir('database'):
         os.system("unzip database.zip")
     
@@ -156,7 +156,6 @@ def g():
         history_messages_key="chat_history",
         output_messages_key="answer",
     )
-    globals().update(locals())
 
 def f():
     asyncio.run(main())