Update utils/__init__.py

utils/__init__.py

@@ -1,160 +1,96 @@
 """
-Utils Package for BackgroundFX Pro
-=================================
-
-This package provides utility modules for device management, logging,
-configuration, and shared utilities.
-
-Author: BackgroundFX Pro Team
-License: MIT
+BackgroundFX Pro Utils Module
+Provides core utilities and computer vision processing functions
 """
 
-# -----------------------------------------------------------------------------
-# 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,
-)
+# Set OMP_NUM_THREADS at module import to prevent libgomp errors
+import os
+if 'OMP_NUM_THREADS' not in os.environ:
+    os.environ['OMP_NUM_THREADS'] = '4'
+    os.environ['MKL_NUM_THREADS'] = '4'
 
-# -----------------------------------------------------------------------------
-# 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,
-)
+import logging
 
-# -----------------------------------------------------------------------------
-# 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,
-)
+# Configure module logger
+logger = logging.getLogger(__name__)
 
-# -----------------------------------------------------------------------------
-# Monitoring / Progress tracking (utils/monitoring/)
-# -----------------------------------------------------------------------------
-from .monitoring.progress_tracker import ProgressTracker
+# ============================================================================
+# IMPORTS FROM UTILS.PY
+# ============================================================================
 
-# -----------------------------------------------------------------------------
-# Shared utilities (from your utilities file at utils/utils.py)
-# -----------------------------------------------------------------------------
 from .utils import (
+    # Classes
     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,
+    
+    # Functions
+    get_file_manager,
 )
 
-# -----------------------------------------------------------------------------
-# 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",
+# ============================================================================
+# IMPORTS FROM CV_PROCESSING.PY
+# ============================================================================
+
+from .cv_processing import (
+    # Configuration and constants
+    PROFESSIONAL_BACKGROUNDS,
+    USE_ENHANCED_SEGMENTATION,
+    USE_AUTO_TEMPORAL_CONSISTENCY,
+    USE_INTELLIGENT_PROMPTING,
+    USE_ITERATIVE_REFINEMENT,
+    
+    # Custom exceptions
+    SegmentationError,
+    MaskRefinementError,
+    BackgroundReplacementError,
+    
+    # Main functions
+    segment_person_hq,
+    segment_person_hq_original,
+    refine_mask_hq,
+    enhance_mask_opencv_advanced,
+    replace_background_hq,
+    create_professional_background,
+    validate_video_file,
+)
 
-    # Monitoring
-    "ProgressTracker",
+# ============================================================================
+# MODULE EXPORTS
+# ============================================================================
 
-    # 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",
+__all__ = [
+    # From utils.py
+    'FileManager',
+    'VideoUtils',
+    'ImageUtils',
+    'ValidationUtils',
+    'get_file_manager',
+    
+    # From cv_processing.py
+    'PROFESSIONAL_BACKGROUNDS',
+    'USE_ENHANCED_SEGMENTATION',
+    'USE_AUTO_TEMPORAL_CONSISTENCY',
+    'USE_INTELLIGENT_PROMPTING',
+    'USE_ITERATIVE_REFINEMENT',
+    'SegmentationError',
+    'MaskRefinementError',
+    'BackgroundReplacementError',
+    'segment_person_hq',
+    'segment_person_hq_original',
+    'refine_mask_hq',
+    'enhance_mask_opencv_advanced',
+    'replace_background_hq',
+    'create_professional_background',
+    'validate_video_file',
 ]
 
-# -----------------------------------------------------------------------------
-# Auto‐initialize logging, device manager, and config on import
-# -----------------------------------------------------------------------------
-try:
-    _logger = setup_logging(logs_dir="LOGS")
-    _logger.info("✅ BackgroundFX Pro Utils package initialized")
+# ============================================================================
+# MODULE METADATA
+# ============================================================================
 
-    _device_manager = get_device_manager()
-    _config = get_config(checkpoints_dir="checkpoints")
+__version__ = '2.0.0'
+__author__ = 'BackgroundFX Pro Team'
 
-    _logger.info("🚀 Utils package ready for BackgroundFX Pro")
-except Exception as e:
-    print(f"⚠️ Utils package initialization warning: {e}")
+logger.info(f"Utils module v{__version__} initialized successfully")