Fix SocketServer message_handler call to pass client_sock for worker registration

vidai/backend.py

@@ -50,7 +50,7 @@ def get_result(msg_id: str) -> dict:
         return {'error': str(e)}
 
 
-def handle_web_message(message: Message) -> Message:
+def handle_web_message(message: Message, client_sock=None) -> Message:
     """Handle messages from web interface."""
     if message.msg_type == 'analyze_request':
         from .config import get_analysis_backend

vidai/comm.py

@@ -162,7 +162,7 @@ class SocketServer:
                                 data=msg_data['data']
                             )
                             print(f"DEBUG: SocketServer parsed message: {message}")
-                            response = self.message_handler(message)
+                            response = self.message_handler(message, client_sock)
                             if response:
                                 resp_data = json.dumps({
                                     'msg_type': response.msg_type,