Update aduc_framework/managers/seedvr_manager.py

aduc_framework/managers/seedvr_manager.py

@@ -2,7 +2,7 @@
 #
 # Copyright (C) 2025 Carlos Rodrigues dos Santos
 #
-# Version: 10.0.0 (Definitive Monkey Patch)
+# Version: 10.0.0 (Definitive Monkey Patch / Single Instance)
 #
 # Esta é a arquitetura final e mais robusta. O paralelismo problemático
 # é desativado programaticamente via "monkey patching" no decorador `master_only`.
@@ -48,43 +48,42 @@ class SeedVrManager:
 
     @staticmethod
     def _check_and_run_global_setup():
+        """Executa o setup de arquivos e aplica o patch para desativar o paralelismo."""
         setup_flag = DEPS_DIR / "seedvr.setup.complete"
         if str(APP_ROOT) not in sys.path: sys.path.insert(0, str(APP_ROOT))
-        if setup_flag.exists(): return True
-        
-        logger.info("--- Iniciando Setup Global do SeedVR (primeira execução) ---")
-        if not SEEDVR_SPACE_DIR.exists():
-            DEPS_DIR.mkdir(exist_ok=True, parents=True)
-            subprocess.run(["git", "clone", "--depth", "1", SEEDVR_SPACE_URL, str(SEEDVR_SPACE_DIR)], check=True)
         
-        required_dirs = ["projects", "common", "models", "configs_3b", "configs_7b", "data"]
-        for dirname in required_dirs:
-            source, target = SEEDVR_SPACE_DIR / dirname, APP_ROOT / dirname
-            if not target.exists(): shutil.copytree(source, target)
-        
-        # <<< --- MONKEY PATCH PARA DESATIVAR O PARALELISMO --- >>>
+        # O patch deve ser aplicado toda vez que o código roda.
         try:
             from common import decorators
             import functools
 
-            # Define um decorador que não faz nada e apenas retorna a função original
             def _passthrough_decorator(func):
                 @functools.wraps(func)
                 def wrapped(*args, **kwargs):
                     return func(*args, **kwargs)
                 return wrapped
             
-            # Substitui o decorador problemático pelo nosso decorador inofensivo
             decorators.master_only = _passthrough_decorator
             logger.info("Monkey patch aplicado com sucesso em 'common.decorators.master_only' para desativar o paralelismo.")
         except Exception as e:
             logger.error(f"Falha ao aplicar o monkey patch: {e}", exc_info=True)
-            # Continua mesmo se falhar, pode funcionar em alguns casos.
+
+        if setup_flag.exists(): return True
+        
+        logger.info("--- Iniciando Setup Global do SeedVR (primeira execução) ---")
+        if not SEEDVR_SPACE_DIR.exists():
+            DEPS_DIR.mkdir(exist_ok=True, parents=True)
+            subprocess.run(["git", "clone", "--depth", "1", SEEDVR_SPACE_URL, str(SEEDVR_SPACE_DIR)], check=True)
+        
+        required_dirs = ["projects", "common", "models", "configs_3b", "configs_7b", "data"]
+        for dirname in required_dirs:
+            source, target = SEEDVR_SPACE_DIR / dirname, APP_ROOT / dirname
+            if not target.exists(): shutil.copytree(source, target)
         
         try:
             import apex
         except ImportError:
-            apex_url = 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/apex-0.1-cp310-cp310-linux_x_86_64.whl'
+            apex_url = 'https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/apex-0.1-cp310-cp310-linux_x86_64.whl'
             apex_wheel_path = _load_file_from_url(url=apex_url, model_dir=str(DEPS_DIR))
             subprocess.run(f"pip install {apex_wheel_path}", check=True, shell=True)
 
@@ -121,7 +120,6 @@ class SeedVrManager:
         self.runner = VideoDiffusionInfer(config)
         OmegaConf.set_readonly(self.runner.config, False)
         
-        # Agora o código decorado dentro desta função usará nosso patch inofensivo
         self.runner.configure_dit_model(device=self.local_device_name, checkpoint=str(checkpoint_path))
         self.runner.configure_vae_model()
         
@@ -216,15 +214,16 @@ try:
     
     if seedvr_gpus_required > 0:
         seedvr_device_ids = hardware_manager.allocate_gpus('SeedVR', seedvr_gpus_required)
-        if 'cpu' not in seedvr_device_ids:
+        if seedvr_device_ids and 'cpu' not in seedvr_device_ids:
             device_to_use = seedvr_device_ids[0]
             seedvr_manager_singleton = SeedVrManager(device_id=device_to_use)
             logger.info(f"Especialista de Masterização HD (SeedVR Single Instance) pronto para usar a GPU {device_to_use}.")
         else:
             seedvr_manager_singleton = SeedVrPlaceholder()
+            logger.warning("SeedVR não foi inicializado porque nenhuma GPU pôde ser alocada.")
     else:
         seedvr_manager_singleton = SeedVrPlaceholder()
-        logger.warning("SeedVR Manager não foi inicializado (gpus_required: 0).")
+        logger.warning("SeedVR Manager não foi inicializado (gpus_required: 0 na config).")
 except Exception as e:
     logger.critical(f"Falha CRÍTICA ao inicializar o SeedVrManager: {e}", exc_info=True)
     seedvr_manager_singleton = SeedVrPlaceholder()