refactor: extract Gradio logging setup into a separate module

app.py

@@ -36,22 +36,10 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 os.environ['HF_HUB_CACHE'] = './models'
 
-# --- Gradio Log Handler ---
-class GradioLogHandler(logging.Handler):
-    def __init__(self, log_queue):
-        super().__init__()
-        self.log_queue = log_queue
-        self.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
-
-    def emit(self, record):
-        self.log_queue.append(self.format(record))
 
-log_queue = collections.deque(maxlen=1000) # Store last 1000 log messages
-gradio_handler = GradioLogHandler(log_queue)
-
-# Set root logger level to DEBUG to capture all messages from agents
-logging.getLogger().setLevel(logging.INFO)
-logging.getLogger().addHandler(gradio_handler)
+# --- Gradio Log Handler ---
+from utils.logging_setup import setup_gradio_logging
+log_queue, gradio_handler = setup_gradio_logging()
 # --- End Gradio Log Handler ---
 
 LOCAL_LOG_DIR = "./hf_inference_logs"

utils/logging_setup.py

@@ -0,0 +1,18 @@
+import logging
+import collections
+
+class GradioLogHandler(logging.Handler):
+    def __init__(self, log_queue):
+        super().__init__()
+        self.log_queue = log_queue
+        self.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
+
+    def emit(self, record):
+        self.log_queue.append(self.format(record))
+
+def setup_gradio_logging():
+    log_queue = collections.deque(maxlen=1000)
+    gradio_handler = GradioLogHandler(log_queue)
+    logging.getLogger().setLevel(logging.INFO)
+    logging.getLogger().addHandler(gradio_handler)
+    return log_queue, gradio_handler