Update core/app.py

core/app.py

@@ -79,7 +79,13 @@ def __init__(self):
         self.config = get_config()  # Use singleton config
         self.device_manager = DeviceManager()
         self.memory_manager = MemoryManager(self.device_manager.get_optimal_device())
-        self.model_loader = ModelLoader(self.device_manager, self.memory_manager)
+        
+        # Initialize ModelLoader with correct parameters (cache_dir and device string)
+        self.model_loader = ModelLoader(
+            cache_dir=None,  # Will use default cache directory
+            device=self.device_manager.get_optimal_device()  # Pass device string, not manager object
+        )
+        
         self.audio_processor = AudioProcessor()
         self.progress_tracker = None
         
@@ -125,8 +131,8 @@ def load_models(self, progress_callback: Optional[Callable] = None) -> str:
                     progress_callback(0.0, f"Starting model loading on {self.device_manager.get_optimal_device()}")
                 
                 # Load models using the specialized loader
-                # Note: load_models returns a boolean, not the models themselves
-                success = self.model_loader.load_models()
+                # load_models() returns a boolean indicating success
+                success = self.model_loader.load_models(force_reload=False)
                 
                 if self.cancel_event.is_set():
                     return "Model loading cancelled"
@@ -168,13 +174,18 @@ def load_models(self, progress_callback: Optional[Callable] = None) -> str:
                 
                 # Add two-stage status to message
                 if self.two_stage_processor is not None:
-                    message += "\n✅ Two-stage processor ready"
+                    message += "\n✅ Two-stage processor ready with AI models"
                 else:
                     message += "\n⚠️ Two-stage processor not available"
                 
                 logger.info(message)
                 return message
                 
+            except AttributeError as e:
+                self.models_loaded = False
+                error_msg = f"Model loading failed - method not found: {str(e)}"
+                logger.error(error_msg)
+                return error_msg
             except ModelLoadingError as e:
                 self.models_loaded = False
                 error_msg = f"Model loading failed: {str(e)}"
@@ -359,8 +370,12 @@ def get_status(self) -> Dict[str, Any]:
         # Add model-specific status if available
         if self.model_loader:
             base_status['model_loader_available'] = True
-            base_status['sam2_loaded'] = self.model_loader.get_sam2() is not None
-            base_status['matanyone_loaded'] = self.model_loader.get_matanyone() is not None
+            try:
+                base_status['sam2_loaded'] = self.model_loader.get_sam2() is not None
+                base_status['matanyone_loaded'] = self.model_loader.get_matanyone() is not None
+            except AttributeError:
+                base_status['sam2_loaded'] = False
+                base_status['matanyone_loaded'] = False
         
         # Add processing status if available
         if self.core_processor: