MogensR's picture
Update utils/__init__.py
9ace6e6
raw
history blame
4.32 kB
"""
Utils Package for BackgroundFX Pro
=================================
This package provides utility modules for device management, logging,
configuration, and shared utilities.
Author: BackgroundFX Pro Team
License: MIT
"""
# -----------------------------------------------------------------------------
# Device management (moved under utils/hardware/)
# -----------------------------------------------------------------------------
from .hardware.device_manager import (
DeviceManager,
get_device_manager,
get_optimal_device,
fix_cuda_compatibility,
setup_optimal_threading,
get_system_diagnostics,
)
# -----------------------------------------------------------------------------
# Logging (still in utils/logger.py)
# -----------------------------------------------------------------------------
from .logger import (
BackgroundFXLogger,
setup_logging,
get_logger,
log_function_call,
log_processing_pipeline,
log_info,
log_error,
log_warning,
log_debug,
)
# -----------------------------------------------------------------------------
# Configuration (assuming utils/config.py exists)
# -----------------------------------------------------------------------------
from .config import (
ConfigManager,
ModelConfig,
QualityConfig,
ProcessingConfig,
VideoConfig,
get_config,
load_config,
get_model_config,
is_model_enabled,
get_quality_thresholds,
get_processing_config,
)
# -----------------------------------------------------------------------------
# Monitoring / Progress tracking (utils/monitoring/)
# -----------------------------------------------------------------------------
from .monitoring.progress_tracker import ProgressTracker
# -----------------------------------------------------------------------------
# Shared utilities (from your utilities file at utils/utils.py)
# -----------------------------------------------------------------------------
from .utils import (
FileManager,
VideoUtils,
ImageUtils,
ValidationUtils,
PerformanceUtils,
temporary_directory,
error_handler,
retry_on_failure,
batch_process,
format_duration,
get_system_info,
ConfigurationError,
ValidationError,
ProcessingError,
safe_division,
clamp,
interpolate,
moving_average,
)
# -----------------------------------------------------------------------------
# Package metadata
# -----------------------------------------------------------------------------
__version__ = "1.0.0"
__author__ = "BackgroundFX Pro Team"
__all__ = [
# Device management
"DeviceManager",
"get_device_manager",
"get_optimal_device",
"fix_cuda_compatibility",
"setup_optimal_threading",
"get_system_diagnostics",
# Logging
"BackgroundFXLogger",
"setup_logging",
"get_logger",
"log_function_call",
"log_processing_pipeline",
"log_info",
"log_error",
"log_warning",
"log_debug",
# Configuration
"ConfigManager",
"ModelConfig",
"QualityConfig",
"ProcessingConfig",
"VideoConfig",
"get_config",
"load_config",
"get_model_config",
"is_model_enabled",
"get_quality_thresholds",
"get_processing_config",
# Monitoring
"ProgressTracker",
# Utilities
"FileManager",
"VideoUtils",
"ImageUtils",
"ValidationUtils",
"PerformanceUtils",
"temporary_directory",
"error_handler",
"retry_on_failure",
"batch_process",
"format_duration",
"get_system_info",
"ConfigurationError",
"ValidationError",
"ProcessingError",
"safe_division",
"clamp",
"interpolate",
"moving_average",
]
# -----------------------------------------------------------------------------
# Auto‐initialize logging, device manager, and config on import
# -----------------------------------------------------------------------------
try:
_logger = setup_logging(logs_dir="LOGS")
_logger.info("✅ BackgroundFX Pro Utils package initialized")
_device_manager = get_device_manager()
_config = get_config(checkpoints_dir="checkpoints")
_logger.info("🚀 Utils package ready for BackgroundFX Pro")
except Exception as e:
print(f"⚠️ Utils package initialization warning: {e}")