From 236eb82c3a91960ba5db7b82efbe0f6a9fd7cf24 Mon Sep 17 00:00:00 2001
From: Lucas Daniel Velazquez M <19197331+Luxter77@users.noreply.github.com>
Date: Thu, 16 Nov 2023 13:20:33 -0300
Subject: [PATCH 1/3] Adds tqdm handler to logging_config.py for progress bar
 integration

---
 modules/logging_config.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/modules/logging_config.py b/modules/logging_config.py
index 7db23d4b6..ce831b5c6 100644
--- a/modules/logging_config.py
+++ b/modules/logging_config.py
@@ -1,6 +1,19 @@
 import os
 import logging
 
+from tqdm.auto import tqdm
+
+class TqdmLoggingHandler(logging.Handler):
+    def __init__(self, level=logging.INFO):
+        super().__init__(level)
+
+    def emit(self, record):
+        try:
+            msg = self.format(record)
+            tqdm.write(msg)
+            self.flush()
+        except Exception:
+            self.handleError(record)
 
 def setup_logging(loglevel):
     if loglevel is None:
@@ -12,5 +25,6 @@ def setup_logging(loglevel):
             level=log_level,
             format='%(asctime)s %(levelname)s [%(name)s] %(message)s',
             datefmt='%Y-%m-%d %H:%M:%S',
+            handlers=[TqdmLoggingHandler()]
         )
 

From cdb60a690dcd35e865eb0caef6c6d8ff64e1b0d5 Mon Sep 17 00:00:00 2001
From: Lucas Daniel Velazquez M <19197331+Luxter77@users.noreply.github.com>
Date: Thu, 16 Nov 2023 16:43:59 -0300
Subject: [PATCH 2/3] Take into account tqdm not being installed before first
 boot for logging

---
 modules/logging_config.py | 37 ++++++++++++++++++++++++-------------
 1 file changed, 24 insertions(+), 13 deletions(-)

diff --git a/modules/logging_config.py b/modules/logging_config.py
index ce831b5c6..99ed2855b 100644
--- a/modules/logging_config.py
+++ b/modules/logging_config.py
@@ -1,30 +1,41 @@
 import os
 import logging
 
-from tqdm.auto import tqdm
+try:
+    from tqdm.auto import tqdm
 
-class TqdmLoggingHandler(logging.Handler):
-    def __init__(self, level=logging.INFO):
-        super().__init__(level)
+    class TqdmLoggingHandler(logging.Handler):
+        def __init__(self, level=logging.INFO):
+            super().__init__(level)
 
-    def emit(self, record):
-        try:
-            msg = self.format(record)
-            tqdm.write(msg)
-            self.flush()
-        except Exception:
-            self.handleError(record)
+        def emit(self, record):
+            try:
+                msg = self.format(record)
+                tqdm.write(msg)
+                self.flush()
+            except Exception:
+                self.handleError(record)
+
+    TQDM_IMPORTED = True
+except ImportError:
+    # tqdm does not exist before first launch
+    # I will import once the UI finishes seting up the enviroment and reloads.
+    TQDM_IMPORTED = False
 
 def setup_logging(loglevel):
     if loglevel is None:
         loglevel = os.environ.get("SD_WEBUI_LOG_LEVEL")
 
+    loghandlers = []
+
+    if TQDM_IMPORTED:
+        loghandlers.append(TqdmLoggingHandler())
+
     if loglevel:
         log_level = getattr(logging, loglevel.upper(), None) or logging.INFO
         logging.basicConfig(
             level=log_level,
             format='%(asctime)s %(levelname)s [%(name)s] %(message)s',
             datefmt='%Y-%m-%d %H:%M:%S',
-            handlers=[TqdmLoggingHandler()]
+            handlers=[]
         )
-

From 7021cdb1de12be3071ecb67aa8d2e34e7a0ec5ab Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Thu, 16 Nov 2023 17:53:57 -0300
Subject: [PATCH 3/3] actually adds handler to logging_config.py

---
 modules/logging_config.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/logging_config.py b/modules/logging_config.py
index 99ed2855b..792698756 100644
--- a/modules/logging_config.py
+++ b/modules/logging_config.py
@@ -37,5 +37,5 @@ def setup_logging(loglevel):
             level=log_level,
             format='%(asctime)s %(levelname)s [%(name)s] %(message)s',
             datefmt='%Y-%m-%d %H:%M:%S',
-            handlers=[]
+            handlers=loghandlers
         )