.

accelerator.py

@@ -1,11 +1,18 @@
 from typing import Union
-from fastapi.websockets import WebSocket
+from fastapi.websockets import WebSocket, WebSocketState
 from websockets import ConnectionClosedError
 
 class Accelerator:
   ws: Union[WebSocket, None] = None
 
-  def connected(self): return self.ws != None
+  def connected(self):
+    if self.ws == None: return False
+    
+    if self.ws.client_state == WebSocketState.CONNECTED:
+      return True
+    else:
+      self.ws = None
+      return False
 
   async def connect(self, ws: WebSocket):
     await ws.accept()

main.py

@@ -1,5 +1,4 @@
 from asyncio import sleep
-from traceback import print_exc
 from typing import Union
 from fastapi import FastAPI
 from fastapi.encoders import jsonable_encoder