Update aduc_orchestrator.py
Browse files- aduc_orchestrator.py +21 -8
aduc_orchestrator.py
CHANGED
|
@@ -104,9 +104,12 @@ class AducOrchestrator:
|
|
| 104 |
logger.info("Maestro: Especialista de Imagem concluiu a geração dos keyframes.")
|
| 105 |
return final_keyframes
|
| 106 |
|
| 107 |
-
|
| 108 |
-
|
| 109 |
-
|
|
|
|
|
|
|
|
|
|
| 110 |
handler_strength,
|
| 111 |
destination_convergence_strength,
|
| 112 |
video_resolution, use_continuity_director,
|
|
@@ -115,19 +118,29 @@ class AducOrchestrator:
|
|
| 115 |
logger.info("AducOrchestrator: Delegando a produção do filme completo ao Deformes4DEngine.")
|
| 116 |
storyboard = self.director.get_state("storyboard", [])
|
| 117 |
|
| 118 |
-
#
|
|
|
|
|
|
|
| 119 |
for update in self.editor.generate_full_movie(
|
| 120 |
keyframes=keyframes,
|
| 121 |
global_prompt=global_prompt,
|
| 122 |
storyboard=storyboard,
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 126 |
handler_strength=handler_strength,
|
| 127 |
destination_convergence_strength=destination_convergence_strength,
|
| 128 |
video_resolution=video_resolution,
|
| 129 |
use_continuity_director=use_continuity_director,
|
| 130 |
-
progress=progress
|
|
|
|
|
|
|
|
|
|
|
|
|
| 131 |
):
|
| 132 |
if "fragment_path" in update and update["fragment_path"]:
|
| 133 |
yield {"fragment_path": update["fragment_path"]}
|
|
|
|
| 104 |
logger.info("Maestro: Especialista de Imagem concluiu a geração dos keyframes.")
|
| 105 |
return final_keyframes
|
| 106 |
|
| 107 |
+
def task_produce_final_movie_with_feedback(self, keyframes, global_prompt,
|
| 108 |
+
# Novos parâmetros de controle da UI
|
| 109 |
+
n_chunks_to_generate, video_end_chunk,
|
| 110 |
+
eco_start_chunk, handler_start_chunk,
|
| 111 |
+
handler_frame_target,
|
| 112 |
+
# Parâmetros restantes
|
| 113 |
handler_strength,
|
| 114 |
destination_convergence_strength,
|
| 115 |
video_resolution, use_continuity_director,
|
|
|
|
| 118 |
logger.info("AducOrchestrator: Delegando a produção do filme completo ao Deformes4DEngine.")
|
| 119 |
storyboard = self.director.get_state("storyboard", [])
|
| 120 |
|
| 121 |
+
# A chamada para o motor agora usa os novos parâmetros de controle de chunk.
|
| 122 |
+
# Os parâmetros antigos (seconds_per_fragment, trim_chunks, echo_chunks) são passados
|
| 123 |
+
# com valores fictícios, pois a função no motor agora os espera.
|
| 124 |
for update in self.editor.generate_full_movie(
|
| 125 |
keyframes=keyframes,
|
| 126 |
global_prompt=global_prompt,
|
| 127 |
storyboard=storyboard,
|
| 128 |
+
# Novos parâmetros de controle
|
| 129 |
+
n_chunks_to_generate=n_chunks_to_generate,
|
| 130 |
+
video_end_chunk=video_end_chunk,
|
| 131 |
+
eco_start_chunk=eco_start_chunk,
|
| 132 |
+
handler_start_chunk=handler_start_chunk,
|
| 133 |
+
handler_frame_target=handler_frame_target,
|
| 134 |
+
# Parâmetros restantes
|
| 135 |
handler_strength=handler_strength,
|
| 136 |
destination_convergence_strength=destination_convergence_strength,
|
| 137 |
video_resolution=video_resolution,
|
| 138 |
use_continuity_director=use_continuity_director,
|
| 139 |
+
progress=progress,
|
| 140 |
+
# Parâmetros antigos que a assinatura de generate_full_movie ainda espera (com valores fictícios)
|
| 141 |
+
seconds_per_fragment=0,
|
| 142 |
+
trim_chunks=0,
|
| 143 |
+
echo_chunks=0
|
| 144 |
):
|
| 145 |
if "fragment_path" in update and update["fragment_path"]:
|
| 146 |
yield {"fragment_path": update["fragment_path"]}
|