Update aduc_framework/orchestrator.py

aduc_framework/orchestrator.py

@@ -2,11 +2,11 @@
 #
 # Copyright (C) August 4, 2025  Carlos Rodrigues dos Santos
 #
-# Versão 5.0.0 (Conductor for Autonomous Director)
+# Versão 6.0.0 (Streaming Conductor)
 #
-# Esta versão do orquestrador está alinhada com a arquitetura do Diretor Autônomo.
-# Ele delega a complexa tarefa de geração de keyframes ao Deformes3DEngine (`painter`),
-# que agora gerencia seu próprio ciclo de avaliação, correção e improviso.
+# Esta versão final do orquestrador está alinhada com a arquitetura do
+# Diretor Autônomo e suporta a emissão de atualizações em tempo real (yield)
+# para interfaces de usuário interativas.
 
 import logging
 from typing import List, Dict, Any, Tuple, Callable, Optional, Generator
@@ -66,34 +66,42 @@ class AducOrchestrator:
         logger.info(f"Imagem de referência processada e salva em: {processed_path}")
         return processed_path
 
-    # --- ETAPA 1: PRÉ-PRODUÇÃO ---
-    def task_pre_production(self, params: PreProductionParams, progress_callback: ProgressCallback = None) -> Tuple[List[str], List[str], GenerationState]:
-        """Orquestra a criação do storyboard e a geração autônoma de keyframes."""
+    # --- ETAPA 1: PRÉ-PRODUÇÃO (STREAMING) ---
+    def task_pre_production(self, params: PreProductionParams, progress_callback: ProgressCallback = None) -> Generator[Dict[str, Any], None, None]:
+        """
+        Orquestra a pré-produção, agora como um gerador que emite
+        atualizações de estado para a UI em tempo real.
+        """
         logger.info("Maestro: Iniciando tarefa de Pré-Produção.")
         self.director.update_parameters("pre_producao", params)
         
-        # Passo 1A: Gerar o roteiro inicial (storyboard)
         if progress_callback: progress_callback(0.1, "Gerando roteiro inicial...")
         storyboard_list = deformes2d_thinker_singleton.generate_storyboard(
             prompt=params.prompt, num_keyframes=params.num_keyframes, ref_image_paths=params.ref_paths
         )
         self.director.update_pre_production_state(params.prompt, params.ref_paths, storyboard_list)
         
-        # Passo 1B: Entregar produção ao Diretor Autônomo para gerar os keyframes
+        yield {
+            "storyboard": storyboard_list,
+            "updated_state": self.director.get_full_state().model_dump()
+        }
+        
         if progress_callback: progress_callback(0.2, "Entregando produção ao Diretor Autônomo...")
         
-        # CORREÇÃO APLICADA: Chamamos o método correto e renomeado `generate_keyframes`.
-        keyframes_detailed_data = self.painter.generate_keyframes(
+        final_keyframes_data = []
+        for keyframes_update in self.painter.generate_keyframes(
             generation_state=self.director.get_full_state_as_dict(), 
             progress_callback=progress_callback
-        )
-        
-        self.director.update_keyframes_state(keyframes_detailed_data)
+        ):
+            self.director.update_keyframes_state(keyframes_update)
+            final_keyframes_data = keyframes_update
+
+            yield {
+                "final_keyframes": [kf["caminho_pixel"] for kf in final_keyframes_data],
+                "updated_state": self.director.get_full_state().model_dump()
+            }
         
-        final_keyframe_paths = [kf["caminho_pixel"] for kf in keyframes_detailed_data]
-        final_state = self.director.get_full_state()
         logger.info("Maestro: Tarefa de Pré-Produção concluída.")
-        return storyboard_list, final_keyframe_paths, final_state
 
     # --- ETAPA 2: PRODUÇÃO ---
     def task_produce_original_movie(self, params: ProductionParams, progress_callback: ProgressCallback = None) -> Tuple[str, List[str], GenerationState]:
@@ -113,7 +121,7 @@ class AducOrchestrator:
         logger.info("Maestro: Tarefa de Produção do Filme Original concluída.")
         return final_video_path, latent_paths, final_state
 
-    # --- ETAPA 3: PÓS-PRODUÇÃO (Cadeia de Efeitos) ---
+    # --- ETAPA 3: PÓS-PRODUÇÃO (CADEIA DE EFEITOS) ---
 
     def task_run_latent_upscaler(self, latent_paths: List[str], chunk_size: int, progress_callback: ProgressCallback = None) -> Generator[Dict[str, Any], None, None]:
         """Aplica upscale 2x nos latentes e os decodifica para um novo vídeo."""
@@ -138,7 +146,8 @@ class AducOrchestrator:
             pixel_tensor = vae_manager_singleton.decode(upscaled_latent_chunk)
 
             current_clip_path = os.path.join(temp_dir, f"upscaled_clip_{i:04d}.mp4")
-            video_encode_tool_singleton.save_video_from_tensor(pixel_tensor, current_clip_path)
+            # Usando o save_video do Deformes4D para consistência
+            self.editor._save_video_from_tensor(pixel_tensor, current_clip_path, fps=24)
             final_upscaled_clip_paths.append(current_clip_path)
             
             del tensors_in_chunk, sub_group_latent, upscaled_latent_chunk, pixel_tensor