From 241539bc94347029d0db0118c52f1af9e7236e8e Mon Sep 17 00:00:00 2001
From: mrhan1993 <50648276+mrhan1993@users.noreply.github.com>
Date: Wed, 24 Apr 2024 11:43:40 +0800
Subject: [PATCH] Colorlog and Packaging Dependencies Fail to Install Fixes
 #302

---
 fooocusapi/utils/logger.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/fooocusapi/utils/logger.py b/fooocusapi/utils/logger.py
index 5179cec..482ddea 100644
--- 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"