Make debug messages conditional on --debug flag to reduce noise

vidai/backend.py

@@ -56,16 +56,21 @@ def handle_web_message(message: Message, client_sock=None) -> 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}")
+        from .config import get_debug
+        if get_debug():
+            print(f"DEBUG: Backend forwarding analyze_request {message.msg_id} to worker {worker_key}")
         if worker_key in worker_sockets:
             # Forward to local worker
             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()}")
+            if get_debug():
+                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}")
+            if get_debug():
+                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")
+            if get_debug():
+                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

@@ -27,6 +27,7 @@ import os
 from typing import Dict, Any, Optional
 from dataclasses import dataclass
 from .compat import get_socket_path, is_unix_sockets_supported
+from .config import get_debug
 
 
 @dataclass
@@ -70,7 +71,8 @@ class SocketCommunicator:
                 'data': message.data
             }).encode('utf-8')
             full_data = data + b'\n'
-            print(f"DEBUG: SocketCommunicator sending: {full_data}")
+            if get_debug():
+                print(f"DEBUG: SocketCommunicator sending: {full_data}")
             self.sock.sendall(full_data)
 
     def receive_message(self) -> Optional[Message]:
@@ -153,7 +155,8 @@ class SocketServer:
                 messages = decoded.split('\n')
                 for msg_str in messages:
                     if msg_str.strip():
-                        print(f"DEBUG: SocketServer processing message: {repr(msg_str)}")
+                        if get_debug():
+                            print(f"DEBUG: SocketServer processing message: {repr(msg_str)}")
                         try:
                             msg_data = json.loads(msg_str)
                             message = Message(
@@ -161,7 +164,8 @@ class SocketServer:
                                 msg_id=msg_data['msg_id'],
                                 data=msg_data['data']
                             )
-                            print(f"DEBUG: SocketServer parsed message: {message}")
+                            if get_debug():
+                                print(f"DEBUG: SocketServer parsed message: {message}")
                             response = self.message_handler(message, client_sock)
                             if response:
                                 resp_data = json.dumps({
@@ -171,7 +175,8 @@ class SocketServer:
                                 }).encode('utf-8')
                                 client_sock.sendall(resp_data + b'\n')
                         except json.JSONDecodeError as e:
-                            print(f"DEBUG: SocketServer JSON decode error: {e}")
+                            if get_debug():
+                                print(f"DEBUG: SocketServer JSON decode error: {e}")
                             pass
         except:
             pass