a51ae7a81a
* feat: add logging utility functions * feat: add logging setup and log level argument to CLI * feat: integrate logging across modules
58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
import importlib
|
|
|
|
|
|
def _lazy_import(name):
|
|
module = importlib.import_module(f"whisperx.{name}")
|
|
return module
|
|
|
|
|
|
def load_align_model(*args, **kwargs):
|
|
alignment = _lazy_import("alignment")
|
|
return alignment.load_align_model(*args, **kwargs)
|
|
|
|
|
|
def align(*args, **kwargs):
|
|
alignment = _lazy_import("alignment")
|
|
return alignment.align(*args, **kwargs)
|
|
|
|
|
|
def load_model(*args, **kwargs):
|
|
asr = _lazy_import("asr")
|
|
return asr.load_model(*args, **kwargs)
|
|
|
|
|
|
def load_audio(*args, **kwargs):
|
|
audio = _lazy_import("audio")
|
|
return audio.load_audio(*args, **kwargs)
|
|
|
|
|
|
def assign_word_speakers(*args, **kwargs):
|
|
diarize = _lazy_import("diarize")
|
|
return diarize.assign_word_speakers(*args, **kwargs)
|
|
|
|
|
|
def setup_logging(*args, **kwargs):
|
|
"""
|
|
Configure logging for WhisperX.
|
|
|
|
Args:
|
|
level: Logging level (debug, info, warning, error, critical). Default: warning
|
|
log_file: Optional path to log file. If None, logs only to console.
|
|
"""
|
|
logging_module = _lazy_import("log_utils")
|
|
return logging_module.setup_logging(*args, **kwargs)
|
|
|
|
|
|
def get_logger(*args, **kwargs):
|
|
"""
|
|
Get a logger instance for the given module.
|
|
|
|
Args:
|
|
name: Logger name (typically __name__ from calling module)
|
|
|
|
Returns:
|
|
Logger instance configured with WhisperX settings
|
|
"""
|
|
logging_module = _lazy_import("log_utils")
|
|
return logging_module.get_logger(*args, **kwargs)
|