fix: add callback

main.py

@@ -246,10 +246,11 @@ async def websocket_endpoint(websocket: WebSocket,  client_id: int):
 
             i  = 0
             for item in pubsub.listen():
-                message = item["data"].decode('utf-8')
-                if(item["type"] == "message") and (message == "end"): pubsub.unsubscribe({client_id})
-                if(i == 10): pubsub.unsubscribe({client_id})
-                await websocket.send_text(f'{message}')
+                if item["type"] == "message":
+                    message = item["data"].decode('utf-8')
+                    if message == "end": pubsub.unsubscribe({client_id})
+                    if(i == 10): pubsub.unsubscribe({client_id})
+                    await websocket.send_text(f'{message}')
                 i+=1