Update app.py

app.py

@@ -50,14 +50,14 @@ def patched_get_type(schema):
     logger.error(f"Gradio patch failed: {e}")
 
 # Import modular components
-from app_config import ProcessingConfig
-from device_manager import DeviceManager
-from memory_manager import MemoryManager
-from model_loader import ModelLoader
-from video_processor import CoreVideoProcessor
-from audio_processor import AudioProcessor
-from progress_tracker import ProgressTracker
-from exceptions import VideoProcessingError, ModelLoadingError, DeviceError
+import app_config
+import device_manager
+import memory_manager
+import model_loader
+import video_processor
+import audio_processor
+import progress_tracker
+import exceptions
 
 # Import utilities (existing)
 from utilities import (
@@ -84,11 +84,11 @@ class VideoProcessor:
     
     def __init__(self):
         """Initialize the video processor with all required components"""
-        self.config = ProcessingConfig()
-        self.device_manager = DeviceManager()
-        self.memory_manager = MemoryManager(self.device_manager.get_optimal_device())
-        self.model_loader = ModelLoader(self.device_manager.get_optimal_device())
-        self.audio_processor = AudioProcessor()
+        self.config = app_config.ProcessingConfig()
+        self.device_manager = device_manager.DeviceManager()
+        self.memory_manager = memory_manager.MemoryManager(self.device_manager.get_optimal_device())
+        self.model_loader = model_loader.ModelLoader(self.device_manager.get_optimal_device())
+        self.audio_processor = audio_processor.AudioProcessor()
         
         # Initialize core processor (will be set up after models load)
         self.core_processor = None
@@ -123,7 +123,7 @@ def load_models(self, progress_callback: Optional[Callable] = None) -> str:
                     return "Model loading cancelled"
                 
                 # Initialize core processor with loaded models
-                self.core_processor = CoreVideoProcessor(
+                self.core_processor = video_processor.CoreVideoProcessor(
                     sam2_predictor=sam2_predictor,
                     matanyone_model=matanyone_model,
                     config=self.config,
@@ -143,7 +143,7 @@ def load_models(self, progress_callback: Optional[Callable] = None) -> str:
                 logger.info(message)
                 return message
                 
-            except ModelLoadingError as e:
+            except exceptions.ModelLoadingError as e:
                 self.models_loaded = False
                 error_msg = f"Model loading failed: {str(e)}"
                 logger.error(error_msg)
@@ -191,7 +191,7 @@ def process_video(
                     progress_callback, preview_mask, preview_greenscreen
                 )
                 
-        except VideoProcessingError as e:
+        except exceptions.VideoProcessingError as e:
             logger.error(f"Video processing failed: {e}")
             return None, f"Processing failed: {str(e)}"
         except Exception as e: