Update utils/__init__.py

utils/__init__.py

@@ -9,17 +9,21 @@
 License: MIT
 """
 
-# Import device management modules
-from .device import (
+# -----------------------------------------------------------------------------
+# 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
+    get_system_diagnostics,
 )
 
-# Import logging modules
+# -----------------------------------------------------------------------------
+# Logging (still in utils/logger.py)
+# -----------------------------------------------------------------------------
 from .logger import (
     BackgroundFXLogger,
     setup_logging,
@@ -29,10 +33,12 @@
     log_info,
     log_error,
     log_warning,
-    log_debug
+    log_debug,
 )
 
-# Import configuration modules
+# -----------------------------------------------------------------------------
+# Configuration (assuming utils/config.py exists)
+# -----------------------------------------------------------------------------
 from .config import (
     ConfigManager,
     ModelConfig,
@@ -44,13 +50,17 @@
     get_model_config,
     is_model_enabled,
     get_quality_thresholds,
-    get_processing_config
+    get_processing_config,
 )
 
-# Import ProgressTracker from monitoring
+# -----------------------------------------------------------------------------
+# Monitoring / Progress tracking (utils/monitoring/)
+# -----------------------------------------------------------------------------
 from .monitoring.progress_tracker import ProgressTracker
 
-# Import shared utility classes and functions
+# -----------------------------------------------------------------------------
+# Shared utilities (from your utilities file at utils/utils.py)
+# -----------------------------------------------------------------------------
 from .utils import (
     FileManager,
     VideoUtils,
@@ -69,83 +79,82 @@
     safe_division,
     clamp,
     interpolate,
-    moving_average
+    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',
-    
+    "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',
-    
+    "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',
-    
+    "ConfigManager",
+    "ModelConfig",
+    "QualityConfig",
+    "ProcessingConfig",
+    "VideoConfig",
+    "get_config",
+    "load_config",
+    "get_model_config",
+    "is_model_enabled",
+    "get_quality_thresholds",
+    "get_processing_config",
+
     # Monitoring
-    'ProgressTracker',
-    
+    "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'
+    "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",
 ]
 
-# Initialize logging and device management on import
+# -----------------------------------------------------------------------------
+# Auto‐initialize logging, device manager, and config on import
+# -----------------------------------------------------------------------------
 try:
-    # Setup logging with LOGS directory
     _logger = setup_logging(logs_dir="LOGS")
     _logger.info("✅ BackgroundFX Pro Utils package initialized")
-    
-    # Initialize device manager 
+
     _device_manager = get_device_manager()
-    
-    # Load configuration
     _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}")