Update main.py

main.py

@@ -1,3 +1,5 @@
+from threading import Thread
+
 from fastapi import FastAPI, WebSocket
 from fastapi.responses import HTMLResponse
 
@@ -60,15 +62,21 @@ async def answer(ws: WebSocket):
   await ws.accept()
 
   print("ws connected!")
-    
+
+  async def callback(output: str):
+    print(f"sent: '{output}'")
+    await ws.send_text(output)
+      
   input = await ws.receive_text()
-  output = answerer(input, 32)
+  answerer_thread = Thread(
+    target=answerer,
+    args=(input, 32),
+    kwargs={"callback": callback}
+  )
+  answerer_thread.start()
+  answerer_thread.join()
 
   print("output ready!")
-    
-  for el in output:
-    print(f"sent: '{el}'")
-    await ws.send_text(el) 
 
   await ws.close()