Colorlog and Packaging Dependencies Fail to Install

Fixes #302
mrhan1993 · Apr 24, 2024 · 241539b · 241539b
1 parent 93edabd
commit 241539b
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/fooocusapi/utils/logger.py b/fooocusapi/utils/logger.py
@@ -12,7 +12,18 @@
 import os
 import sys
 
-from colorlog import ColoredFormatter
+try:
+    from colorlog import ColoredFormatter
+except ImportError:
+    from fooocusapi.utils.tools import run_pip
+    run_pip(
+        command="install colorlog",
+        desc="Install colorlog for logger.",
+        live=True
+    )
+finally:
+    from colorlog import ColoredFormatter
+
 
 own_path = os.path.dirname(os.path.abspath(__file__))
 log_dir = "logs"