From f4411250f311f1ba93b8ba57b4252e7c23f7d925 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Wed, 12 Mar 2025 07:13:40 -0400
Subject: [PATCH] Repeat frontend version warning at the end.

This way someone running ComfyUI with the command line is more likely to
actually see it.
---
 app/frontend_management.py |  3 ++-
 app/logger.py              | 14 ++++++++++++++
 main.py                    |  2 ++
 3 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/app/frontend_management.py b/app/frontend_management.py
index 4b7dfbb9..b4ba994d 100644
--- a/app/frontend_management.py
+++ b/app/frontend_management.py
@@ -17,6 +17,7 @@ import requests
 from typing_extensions import NotRequired
 
 from comfy.cli_args import DEFAULT_VERSION_STRING
+import app.logger
 
 # The path to the requirements.txt file
 req_path = Path(__file__).parents[1] / "requirements.txt"
@@ -42,7 +43,7 @@ def check_frontend_version():
         with open(req_path, "r", encoding="utf-8") as f:
             required_frontend = parse_version(f.readline().split("=")[-1])
         if frontend_version < required_frontend:
-            logging.warning("________________________________________________________________________\nWARNING WARNING WARNING WARNING WARNING\n\nInstalled frontend version {} is lower than the recommended version {}.\n\n{}\n________________________________________________________________________".format('.'.join(map(str, frontend_version)), '.'.join(map(str, required_frontend)), frontend_install_warning_message()))
+            app.logger.log_startup_warning("________________________________________________________________________\nWARNING WARNING WARNING WARNING WARNING\n\nInstalled frontend version {} is lower than the recommended version {}.\n\n{}\n________________________________________________________________________".format('.'.join(map(str, frontend_version)), '.'.join(map(str, required_frontend)), frontend_install_warning_message()))
         else:
             logging.info("ComfyUI frontend version: {}".format(frontend_version_str))
     except Exception as e:
diff --git a/app/logger.py b/app/logger.py
index 9e9f84cc..3d26d98f 100644
--- a/app/logger.py
+++ b/app/logger.py
@@ -82,3 +82,17 @@ def setup_logger(log_level: str = 'INFO', capacity: int = 300, use_stdout: bool
         logger.addHandler(stdout_handler)
 
     logger.addHandler(stream_handler)
+
+
+STARTUP_WARNINGS = []
+
+
+def log_startup_warning(msg):
+    logging.warning(msg)
+    STARTUP_WARNINGS.append(msg)
+
+
+def print_startup_warnings():
+    for s in STARTUP_WARNINGS:
+        logging.warning(s)
+    STARTUP_WARNINGS.clear()
diff --git a/main.py b/main.py
index c6f5c3c1..1b100fa8 100644
--- a/main.py
+++ b/main.py
@@ -139,6 +139,7 @@ from server import BinaryEventTypes
 import nodes
 import comfy.model_management
 import comfyui_version
+import app.logger
 
 
 def cuda_malloc_warning():
@@ -299,6 +300,7 @@ if __name__ == "__main__":
     event_loop, _, start_all_func = start_comfyui()
     try:
         x = start_all_func()
+        app.logger.print_startup_warnings()
         event_loop.run_until_complete(x)
     except KeyboardInterrupt:
         logging.info("\nStopped server")