Update main.py

main.py

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