Rename model_loader.py to models/loaders/model_loader.py

model_loader.py → models/loaders/model_loader.py

@@ -22,10 +22,10 @@
 import gradio as gr
 from omegaconf import DictConfig, OmegaConf
 
-# Import modular components
-import exceptions
-import device_manager
-import memory_manager
+# Import modular components - Updated paths for BackgroundFX Pro structure
+from core.exceptions import ModelLoadingError  # Updated import
+from utils.hardware.device_manager import DeviceManager  # Updated import
+from utils.system.memory_manager import MemoryManager  # Updated import
 
 logger = logging.getLogger(__name__)
 
@@ -204,7 +204,7 @@ class ModelLoader:
     Handles automatic config detection, multiple fallback strategies, and memory management
     """
     
-    def __init__(self, device_mgr: device_manager.DeviceManager, memory_mgr: memory_manager.MemoryManager):
+    def __init__(self, device_mgr: DeviceManager, memory_mgr: MemoryManager):
         self.device_manager = device_mgr
         self.memory_manager = memory_mgr
         self.device = self.device_manager.get_optimal_device()
@@ -288,7 +288,7 @@ def load_all_models(self, progress_callback: Optional[callable] = None, cancel_e
             self.sam2_predictor = self._load_sam2_predictor(progress_callback)
             
             if self.sam2_predictor is None:
-                raise exceptions.ModelLoadingError("SAM2", "Failed to load SAM2 predictor")
+                raise ModelLoadingError("SAM2", "Failed to load SAM2 predictor")
             
             sam2_time = time.time() - start_time
             self.loading_stats['sam2_load_time'] = sam2_time
@@ -304,7 +304,7 @@ def load_all_models(self, progress_callback: Optional[callable] = None, cancel_e
             self.matanyone_model, self.matanyone_core = self._load_matanyone_model(progress_callback)
             
             if self.matanyone_model is None:
-                raise exceptions.ModelLoadingError("MatAnyone", "Failed to load MatAnyone model")
+                raise ModelLoadingError("MatAnyone", "Failed to load MatAnyone model")
             
             matanyone_time = time.time() - matanyone_start
             self.loading_stats['matanyone_load_time'] = matanyone_time