Update aduc_orchestrator.py

aduc_orchestrator.py

@@ -2,7 +2,7 @@
 #
 # Copyright (C) August 4, 2025  Carlos Rodrigues dos Santos
 #
-# Version: 2.0.0
+# Version: 2.0.2
 #
 # This file contains the core ADUC (Automated Discovery and Orchestration of Complex tasks)
 # orchestrator, known as the "Maestro" (Γ). Its responsibility is to manage the high-level
@@ -12,7 +12,6 @@
 # "Director" component. It does not perform AI inference itself but acts as the central conductor.
 
 import os
-import time
 import logging
 from typing import List, Dict, Any, Generator, Tuple
 
@@ -179,7 +178,6 @@ class AducOrchestrator:
         logger.info("Maestro: Delegating production of the original movie to Deformes4DEngine.")
         storyboard = self.director.get_state("storyboard", [])
 
-        # The engine will now handle the full generation and return the final paths
         result = self.editor.generate_original_movie(
             keyframes=keyframes,
             global_prompt=global_prompt,
@@ -190,7 +188,6 @@ class AducOrchestrator:
             destination_convergence_strength=destination_convergence_strength,
             video_resolution=video_resolution,
             use_continuity_director=use_continuity_director,
-            # New LTX pipeline parameters
             guidance_scale=guidance_scale,
             stg_scale=stg_scale,
             num_inference_steps=inference_steps,
@@ -208,7 +205,6 @@ class AducOrchestrator:
         """
         logger.info(f"Maestro: Delegating latent upscaling task for {len(latent_paths)} fragments.")
         
-        # This is a generator function to allow for UI updates
         for update in self.editor.upscale_latents_and_create_video(
             latent_paths=latent_paths,
             chunk_size=chunk_size,
@@ -216,28 +212,27 @@ class AducOrchestrator:
         ):
             if "final_path" in update and update["final_path"]:
                 self.director.update_state("final_video_path", update["final_path"])
-                yield update["final_path"]
+                yield update
                 break
         
         logger.info("Maestro: Latent upscaling complete.")
     
-    def task_run_hd_mastering(self, source_video_path: str, model_version: str, steps: int, progress: gr.Progress) -> Generator[Dict[str, Any], None, None]:
+    def task_run_hd_mastering(self, source_video_path: str, model_version: str, steps: int, prompt: str, progress: gr.Progress) -> Generator[Dict[str, Any], None, None]:
         """
         Orchestrates the HD mastering task.
         """
         logger.info(f"Maestro: Delegating HD mastering task using SeedVR {model_version}.")
         
-        # We'll need a new method in the editor to handle this
         for update in self.editor.master_video_hd(
             source_video_path=source_video_path,
             model_version=model_version,
             steps=steps,
-            prompt=self.director.get_state("storyboard", ["A cinematic high-quality movie."])[0],
+            prompt=prompt,
             progress=progress
         ):
             if "final_path" in update and update["final_path"]:
                 self.director.update_state("final_video_path", update["final_path"])
-                yield update["final_path"]
+                yield update
                 break
         
         logger.info("Maestro: HD mastering complete.")
@@ -248,7 +243,6 @@ class AducOrchestrator:
         """
         logger.info(f"Maestro: Delegating audio generation task.")
         
-        # Another new method in the editor
         for update in self.editor.generate_audio_for_final_video(
             source_video_path=source_video_path,
             audio_prompt=audio_prompt,
@@ -256,7 +250,7 @@ class AducOrchestrator:
         ):
              if "final_path" in update and update["final_path"]:
                 self.director.update_state("final_video_path", update["final_path"])
-                yield update["final_path"]
+                yield update
                 break
         
         logger.info("Maestro: Audio generation complete.")