Small fix

scripts/roop_logging.py

@@ -1,41 +1,41 @@
-import logging
+import logging
-import copy
+import copy
-import sys
+import sys
-
+
-from modules import shared
+from modules import shared
-
+
-
+
-class ColoredFormatter(logging.Formatter):
+class ColoredFormatter(logging.Formatter):
-    COLORS = {
+    COLORS = {
-        "DEBUG": "\033[0;36m",  # CYAN
+        "DEBUG": "\033[0;36m",  # CYAN
-        "INFO": "\033[0;32m",  # GREEN
+        "INFO": "\033[0;32m",  # GREEN
-        "WARNING": "\033[0;33m",  # YELLOW
+        "WARNING": "\033[0;33m",  # YELLOW
-        "ERROR": "\033[0;31m",  # RED
+        "ERROR": "\033[0;31m",  # RED
-        "CRITICAL": "\033[0;37;41m",  # WHITE ON RED
+        "CRITICAL": "\033[0;37;41m",  # WHITE ON RED
-        "RESET": "\033[0m",  # RESET COLOR
+        "RESET": "\033[0m",  # RESET COLOR
-    }
+    }
-
+
-    def format(self, record):
+    def format(self, record):
-        colored_record = copy.copy(record)
+        colored_record = copy.copy(record)
-        levelname = colored_record.levelname
+        levelname = colored_record.levelname
-        seq = self.COLORS.get(levelname, self.COLORS["RESET"])
+        seq = self.COLORS.get(levelname, self.COLORS["RESET"])
-        colored_record.levelname = f"{seq}{levelname}{self.COLORS['RESET']}"
+        colored_record.levelname = f"{seq}{levelname}{self.COLORS['RESET']}"
-        return super().format(colored_record)
+        return super().format(colored_record)
-
+
-
+
-# Create a new logger
+# Create a new logger
-logger = logging.getLogger("roop")
+logger = logging.getLogger("nsfw-roop")
-logger.propagate = False
+logger.propagate = False
-
+
-# Add handler if we don't have one.
+# Add handler if we don't have one.
-if not logger.handlers:
+if not logger.handlers:
-    handler = logging.StreamHandler(sys.stdout)
+    handler = logging.StreamHandler(sys.stdout)
-    handler.setFormatter(
+    handler.setFormatter(
-        ColoredFormatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+        ColoredFormatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
-    )
+    )
-    logger.addHandler(handler)
+    logger.addHandler(handler)
-
+
-# Configure logger
+# Configure logger
-loglevel_string = getattr(shared.cmd_opts, "controlnet_loglevel", "INFO")
+loglevel_string = getattr(shared.cmd_opts, "controlnet_loglevel", "INFO")
-loglevel = getattr(logging, loglevel_string.upper(), "info")
+loglevel = getattr(logging, loglevel_string.upper(), "info")
-logger.setLevel(loglevel)
+logger.setLevel(loglevel)