Update aduc_framework/orchestrator.py

aduc_framework/orchestrator.py

@@ -2,14 +2,11 @@
 #
 # Copyright (C) August 4, 2025  Carlos Rodrigues dos Santos
 #
-# Versão 4.0.0 (Multi-Pool Orchestration)
+# Version: 4.1.0 (Multi-Pool with Forced SeedVR 3B)
 #
 # Esta versão do orquestrador opera com uma arquitetura de múltiplos pools de
-# especialistas. A responsabilidade de gerenciar a VRAM e o ciclo de vida dos
-# modelos foi delegada aos seus respectivos managers (LtxPoolManager,
-# SeedVrPoolManager, MMAudioPoolManager), cada um operando em um conjunto
-# dedicado de GPUs. O orquestrador agora foca exclusivamente na lógica de
-# alto nível do fluxo de trabalho.
+# especialistas e simplifica a tarefa de masterização HD, fixando o uso do
+# modelo SeedVR 3B e removendo a necessidade de selecionar a versão do modelo.
 
 import logging
 from typing import List, Dict, Any, Tuple, Callable, Optional, Generator
@@ -81,7 +78,6 @@ class AducOrchestrator:
         self.director.update_pre_production_state(params.prompt, params.ref_paths, storyboard_list)
         
         if progress_callback: progress_callback(0.2, "Iniciando geração de keyframes...")
-        # O 'painter' (Deformes3D) internamente usará o LTX Pool Manager para gerar as imagens
         keyframes_detailed_data = self.painter.generate_keyframes_from_storyboard(generation_state=self.director.get_full_state_as_dict(), progress_callback=progress_callback)
         self.director.update_keyframes_state(keyframes_detailed_data)
         
@@ -96,7 +92,6 @@ class AducOrchestrator:
         logger.info("Maestro: Iniciando tarefa de Produção do Filme Original.")
         self.director.update_parameters("producao", params)
         
-        # O 'editor' (Deformes4D) usa intensivamente o LTX Pool Manager
         result_data = self.editor.generate_original_movie(full_generation_state=self.director.get_full_state_as_dict(), progress_callback=progress_callback)
         self.director.update_video_state(result_data["video_data"])
         
@@ -146,22 +141,18 @@ class AducOrchestrator:
         logger.info(f"Upscaling de latentes completo! Vídeo final em: {final_video_path}")
         yield {"final_path": final_video_path}
 
-    def task_run_hd_mastering(self, source_video_path: str, model_version: str, steps: int, prompt: str, progress_callback: ProgressCallback = None) -> Generator[Dict[str, Any], None, None]:
-        """Aplica masterização em HD usando o pool de GPUs do SeedVR."""
+    def task_run_hd_mastering(self, source_video_path: str, steps: int, prompt: str, progress_callback: ProgressCallback = None) -> Generator[Dict[str, Any], None, None]:
+        """Aplica masterização em HD usando o pool de GPUs do SeedVR com o modelo 3B."""
         if not self.director.workspace_dir: raise RuntimeError("Orchestrator não inicializado.")
-        logger.info(f"--- ORQUESTRADOR: Tarefa de Masterização HD com SeedVR {model_version} ---")
-        
-        # Com pools dedicados, não há mais necessidade de descarregar modelos LTX.
-        # A chamada é direta e o SeedVrPoolManager gerencia seus próprios recursos.
+        logger.info(f"--- ORQUESTRADOR: Tarefa de Masterização HD com SeedVR 3B ---")
         
         run_timestamp = int(time.time())
-        output_path = os.path.join(self.director.workspace_dir, f"hd_mastered_movie_{model_version}_{run_timestamp}.mp4")
+        output_path = os.path.join(self.director.workspace_dir, f"hd_mastered_movie_3B_{run_timestamp}.mp4")
         
         final_path = seedvr_manager_singleton.process_video(
             input_video_path=source_video_path,
             output_video_path=output_path,
             prompt=prompt,
-            model_version=model_version,
             steps=steps
         )
         logger.info(f"Masterização HD completa! Vídeo final em: {final_path}")
@@ -191,7 +182,6 @@ class AducOrchestrator:
 
         if progress_callback: progress_callback(0.5, "Gerando trilha de áudio...")
         
-        # A chamada é direta. O MMAudioPoolManager gerencia suas GPUs dedicadas.
         final_path = mmaudio_manager_singleton.generate_audio_for_video(
             video_path=source_video_path,
             prompt=audio_prompt,