feat(server): make ComfyUI log level configurable

This commit introduces the `comfyui-log-level` and `comfyui-inference-log-level`
arguments to the server, allowing users to adjust the log level and reduce
excessive logging.

Author: rickstaa

server/app.py

@@ -287,7 +287,10 @@ async def on_startup(app: web.Application):
         patch_loop_datagram(app["media_ports"])
 
     app["pipeline"] = Pipeline(
-        cwd=app["workspace"], disable_cuda_malloc=True, gpu_only=True
+        cwd=app["workspace"],
+        disable_cuda_malloc=True,
+        gpu_only=True,
+        comfyui_log_level=app.get("comfyui_log_level", None),
     )
     app["pcs"] = set()
     app["video_tracks"] = {}
@@ -328,6 +331,12 @@ async def on_shutdown(app: web.Application):
         action="store_true",
         help="Include stream ID as a label in Prometheus metrics.",
     )
+    parser.add_argument(
+        "--comfyui-log-level",
+        default=None,
+        choices=logging._nameToLevel.keys(),
+        help="Override the ComfyUI client log level",
+    )
     args = parser.parse_args()
 
     logging.basicConfig(
@@ -377,4 +386,8 @@ def force_print(*args, **kwargs):
         print(*args, **kwargs, flush=True)
         sys.stdout.flush()
 
+    # Allow people to override the ComfyUI client log level.
+    if args.comfyui_log_level:
+        app["comfyui_log_level"] = args.comfyui_log_level
+
     web.run_app(app, host=args.host, port=int(args.port), print=force_print)

server/pipeline.py

@@ -10,8 +10,20 @@
 
 
 class Pipeline:
-    def __init__(self, **kwargs):
-        self.client = ComfyStreamClient(**kwargs, max_workers=5) # TODO: hardcoded max workers, should it be configurable?
+    def __init__(self, comfyui_log_level: int = None, **kwargs):
+        """Initialize the pipeline with the given configuration.
+
+        Args:
+            comfyui_log_level: Log level for the ComfyUI client. Defaults to None, using ComfyUI's global log level.
+                to None, which means using the global ComfyUI log level.
+            **kwargs: Additional arguments to pass to the ComfyStreamClient
+        """
+        if comfyui_log_level is not None:
+            kwargs["logging_level"] = comfyui_log_level
+
+        self.client = ComfyStreamClient(
+            max_workers=5, **kwargs
+        )  # TODO: hardcoded max workers, should it be configurable?
 
         self.video_incoming_frames = asyncio.Queue()
         self.audio_incoming_frames = asyncio.Queue()
@@ -109,4 +121,4 @@ async def get_nodes_info(self) -> Dict[str, Any]:
         return nodes_info
 
     async def cleanup(self):
-        await self.client.cleanup()
+        await self.client.cleanup()