Update main.py

main.py

@@ -54,38 +54,6 @@ background_task_handle: Optional[asyncio.Task] = None
 # --- Create FastAPI app ---
 app = FastAPI(title="Pokemon Battle Livestream")
 
-# --- WebSocket connection manager ---
-class ConnectionManager:
-    def __init__(self):
-        self.active_connections: Set[WebSocket] = set()
-        self.current_html: str = create_idle_html("Initializing...", "Setting up Pokemon Battle Stream")
-
-    async def connect(self, websocket: WebSocket):
-        await websocket.accept()
-        self.active_connections.add(websocket)
-        # Send current state to newly connected client
-        await websocket.send_text(self.current_html)
-
-    def disconnect(self, websocket: WebSocket):
-        self.active_connections.remove(websocket)
-
-    async def update_all(self, html: str):
-        """Update the current HTML and broadcast to all clients"""
-        self.current_html = html
-        if self.active_connections:
-            # Only log if there are connections to update
-            print(f"Broadcasting update to {len(self.active_connections)} clients")
-        
-        # Make a copy of the connections set to avoid modification during iteration
-        connections_copy = self.active_connections.copy()
-        for connection in connections_copy:
-            try:
-                await connection.send_text(html)
-            except Exception as e:
-                print(f"Error sending to client: {e}")
-                # Don't remove here - will be handled by disconnect route
-
-manager = ConnectionManager()
 
 # --- Helper Functions ---
 def get_active_battle(agent: Player) -> Optional[Battle]:
@@ -515,6 +483,39 @@ def log_task_exception(task: asyncio.Task):
         print(f"Exception in background task {task.get_name()}: {e}")
         traceback.print_exc()
 
+
+# --- WebSocket connection manager ---
+class ConnectionManager:
+    def __init__(self):
+        self.active_connections: Set[WebSocket] = set()
+        self.current_html: str = create_idle_html("Initializing...", "Setting up Pokemon Battle Stream")
+
+    async def connect(self, websocket: WebSocket):
+        await websocket.accept()
+        self.active_connections.add(websocket)
+        # Send current state to newly connected client
+        await websocket.send_text(self.current_html)
+
+    def disconnect(self, websocket: WebSocket):
+        self.active_connections.remove(websocket)
+
+    async def update_all(self, html: str):
+        """Update the current HTML and broadcast to all clients"""
+        self.current_html = html
+        if self.active_connections:
+            # Only log if there are connections to update
+            print(f"Broadcasting update to {len(self.active_connections)} clients")
+        
+        # Make a copy of the connections set to avoid modification during iteration
+        connections_copy = self.active_connections.copy()
+        for connection in connections_copy:
+            try:
+                await connection.send_text(html)
+            except Exception as e:
+                print(f"Error sending to client: {e}")
+                # Don't remove here - will be handled by disconnect route
+
+manager = ConnectionManager()
 # --- API Routes ---
 @app.get("/", response_class=HTMLResponse)
 async def get_homepage():