Yuvraj SeegolamandClaude Sonnet 4.5
9ec12e4c17
- Implement ColoredLogger with lightning emoji for enhanced visual feedback - Add color-coded log levels (INFO=Green, WARNING=Yellow, ERROR=Red) - Refactor model configurations to config/models.json for easier customization - Integrate huggingface_hub for reliable downloads with progress tracking - Add tqdm progress bar styled to match logger format for batch processing - Improve ONNX model version detection for DA3, DAD, and custom models - Reorganize codebase: move nodes to nodes/ directory, utilities to separate files - Update dependencies: add huggingface_hub to requirements.txt - Enhance error handling and logging throughout engine builder and depth processing Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
32 lines
881 B
Python
32 lines
881 B
Python
class ColoredLogger:
|
|
COLORS = {
|
|
'DEBUG': '\033[36m', # Cyan
|
|
'INFO': '\033[32m', # Green
|
|
'WARNING': '\033[33m', # Yellow
|
|
'ERROR': '\033[31m', # Red
|
|
'CRITICAL': '\033[35m', # Magenta
|
|
'RESET': '\033[0m' # Reset
|
|
}
|
|
|
|
def __init__(self, name):
|
|
self.name = name
|
|
|
|
def _log(self, level, message):
|
|
color = self.COLORS.get(level, self.COLORS['RESET'])
|
|
reset = self.COLORS['RESET']
|
|
print(f"{color}[{self.name}|{level}]{reset} - {message}")
|
|
|
|
def debug(self, message):
|
|
self._log('DEBUG', message)
|
|
|
|
def info(self, message):
|
|
self._log('INFO', message)
|
|
|
|
def warning(self, message):
|
|
self._log('WARNING', message)
|
|
|
|
def error(self, message):
|
|
self._log('ERROR', message)
|
|
|
|
def critical(self, message):
|
|
self._log('CRITICAL', message) |