Update app.py

app.py

@@ -1726,12 +1726,12 @@ Suggested Resolution Approach:
 
         # Schedule an immediate status broadcast if anything was removed
         if (removed_from_list or removed_from_collab):
-             if self.main_loop.is_running():
+            if self.main_loop.is_running():
                  # Use call_soon_threadsafe to schedule the async coroutine in the main loop
-                 self.main_loop.call_soon_threadsafe(asyncio.create_task, self.broadcast_collaboration_status_once())
-                 logger.debug(f"Scheduled immediate status broadcast after removing client {client_desc}.")
-             else:
-                  logger.warning("Main loop not running, cannot schedule immediate status broadcast after client removal.")
+                self.main_loop.call_soon_threadsafe(asyncio.create_task, self.broadcast_collaboration_status_once())
+                logger.debug(f"Scheduled immediate status broadcast after removing client {client_desc}.")
+            else:
+                logger.warning("Main loop not running, cannot schedule immediate status broadcast after client removal.")
 
 
     async def broadcast_collaboration_status_once(self):
@@ -1742,30 +1742,30 @@ Suggested Resolution Approach:
          tasks = []
          # No need for disconnected_clients list here, remove_ws_client handles it
 
-         for client in active_clients_snapshot:
-             # Check if client is closed using the standard websockets property
-             if client.closed:
-                 continue
-             try:
-                 tasks.append(client.send(status_payload))
-             except (ConnectionClosed, ConnectionAbortedError, ConnectionResetError, WebSocketException) as e:
-                 logger.warning(f"Client {getattr(client, 'client_id', client.remote_address)} seems disconnected during single broadcast: {type(e).__name__}. Attempting removal.")
-                 if self.main_loop.is_running():
-                     self.main_loop.call_soon_threadsafe(self.remove_ws_client, client)
-                 else:
-                      logger.warning("Main loop not running, cannot schedule client removal.")
-                 continue
-             except Exception as e:
-                 logger.error(f"Unexpected error preparing single broadcast to client {getattr(client, 'client_id', client.remote_address)}: {e}. Attempting removal.")
-                 if self.main_loop.is_running():
-                     self.main_loop.call_soon_threadsafe(self.remove_ws_client, client)
-                 else:
-                      logger.warning("Main loop not running, cannot schedule client removal.")
-                 continue
+        for client in active_clients_snapshot:
+            # Check if client is closed using the standard websockets property
+            if client.closed:
+                continue
+            try:
+                tasks.append(client.send(status_payload))
+            except (ConnectionClosed, ConnectionAbortedError, ConnectionResetError, WebSocketException) as e:
+                logger.warning(f"Client {getattr(client, 'client_id', client.remote_address)} seems disconnected during single broadcast: {type(e).__name__}. Attempting removal.")
+                if self.main_loop.is_running():
+                    self.main_loop.call_soon_threadsafe(self.remove_ws_client, client)
+                else:
+                     logger.warning("Main loop not running, cannot schedule client removal.")
+                continue
+            except Exception as e:
+                logger.error(f"Unexpected error preparing single broadcast to client {getattr(client, 'client_id', client.remote_address)}: {e}. Attempting removal.")
+                if self.main_loop.is_running():
+                    self.main_loop.call_soon_threadsafe(self.remove_ws_client, client)
+                else:
+                    logger.warning("Main loop not running, cannot schedule client removal.")
+                continue
 
-         if tasks:
-             logger.debug(f"Broadcasting single status update to {len(tasks)} clients.")
-             # No need to await tasks here if remove_ws_client is scheduled via call_soon_threadsafe
+        if tasks:
+            logger.debug(f"Broadcasting single status update to {len(tasks)} clients.")
+            # No need to await tasks here if remove_ws_client is scheduled via call_soon_threadsafe
 
 
     def _identify_stale_issues(self):
@@ -3422,13 +3422,12 @@ def create_ui(manager: IssueManager) -> gr.Blocks:
 
                 }})(); // End IIFE
             </script>
-            """
+            }"""
         # The _js parameter injects the JavaScript code into the Gradio page
         demo_app.load(_js=web_socket_js(WS_PORT, GRADIO_PORT), fn=None, inputs=None, outputs=None)
 
     return demo_app
 
-
 # ========== WebSocket Server Logic ==========
 async def handle_ws_connection(websocket: WebSocketServerProtocol, path: str, manager: IssueManager):
     """Handles incoming WebSocket connections and messages for collaboration."""
@@ -3437,7 +3436,7 @@ async def handle_ws_connection(websocket: WebSocketServerProtocol, path: str, ma
     setattr(websocket, 'client_id', client_id)
     remote_addr = websocket.remote_address
     logger.info(f"WebSocket client connected: {remote_addr} assigned ID {client_id}")
-
+    
     # Add the new client to the list of active clients
     manager.ws_clients.append(websocket)
     logger.info(f"Client list size: {len(manager.ws_clients)}")