Add debug logging to backend message forwarding and socket receive to trace communication issues

vidai/backend.py

@@ -56,13 +56,16 @@ def handle_web_message(message: Message) -> Message:
         from .config import get_analysis_backend
         backend = get_analysis_backend()
         worker_key = f'analysis_{backend}'
+        print(f"DEBUG: Backend forwarding analyze_request {message.msg_id} to worker {worker_key}")
         if worker_key in worker_sockets:
             # Forward to local worker
-            worker_sockets[worker_key].sendall(
-                f'{{"msg_type": "{message.msg_type}", "msg_id": "{message.msg_id}", "data": {message.data}}}\n'.encode('utf-8')
-            )
+            msg_json = f'{{"msg_type": "{message.msg_type}", "msg_id": "{message.msg_id}", "data": {message.data}}}\n'
+            print(f"DEBUG: Sending to worker: {msg_json.strip()}")
+            worker_sockets[worker_key].sendall(msg_json.encode('utf-8'))
+            print(f"DEBUG: Sent to worker {worker_key}")
             return None  # No immediate response
         else:
+            print(f"DEBUG: Worker {worker_key} not found in worker_sockets")
             return Message('error', message.msg_id, {'error': f'Worker {worker_key} not available'})
     elif message.msg_type == 'train_request':
         backend = get_training_backend()

vidai/comm.py

@@ -76,14 +76,20 @@ class SocketCommunicator:
         if self.sock:
             try:
                 data = self.sock.recv(4096)
+                print(f"DEBUG: SocketCommunicator received raw data: {data}")
                 if data:
-                    msg_data = json.loads(data.decode('utf-8').strip())
+                    decoded = data.decode('utf-8').strip()
+                    print(f"DEBUG: Decoded data: {repr(decoded)}")
+                    msg_data = json.loads(decoded)
                     return Message(
                         msg_type=msg_data['msg_type'],
                         msg_id=msg_data['msg_id'],
                         data=msg_data['data']
                     )
-            except:
+                else:
+                    print(f"DEBUG: No data received")
+            except Exception as e:
+                print(f"DEBUG: Exception in receive_message: {e}")
                 pass
         return None