fix: add websocket in handlerToken

main.py

@@ -259,22 +259,18 @@ async def websocket_endpoint(websocket: WebSocket):
 
     await websocket.accept()
 
-    print(f'receive')
+    oldReceiveText = ''
 
     try:
-        prompt = await websocket.receive_text()
-        res = QA(prompt)
-
         while True:
+            prompt = await websocket.receive_text()
 
-            print(prompt)
-
-            tokenMessageLLM.after_update(lambda now, old: print(f"{old} updated to {now}."))
+            if(oldReceiveText != prompt):
+                tokenMessageLLM.after_update(lambda now, old: websocket.send_text(now))
+                oldReceiveText = prompt
+                res = QA(prompt)
 
-            print(f'tokenMessageLLM')
-
-
-            print(res)
+            print(prompt)
 
     except WebSocketDisconnect:
         print('disconnect')