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