euiia commited on
Commit
6714ca4
·
verified ·
1 Parent(s): 44746ab

Update aduc_orchestrator.py

Browse files
Files changed (1) hide show
  1. 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
- # --- ASSINATURA DA FUNÇÃO ATUALIZADA ---
108
- def task_produce_final_movie_with_feedback(self, keyframes, global_prompt, seconds_per_fragment,
109
- trim_chunks: int, echo_chunks: int, # <-- PARÂMETROS ATUALIZADOS
 
 
 
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
- # --- CHAMADA DA FUNÇÃO ATUALIZADA ---
 
 
119
  for update in self.editor.generate_full_movie(
120
  keyframes=keyframes,
121
  global_prompt=global_prompt,
122
  storyboard=storyboard,
123
- seconds_per_fragment=seconds_per_fragment,
124
- trim_chunks=trim_chunks, # <-- PARÂMETRO NOVO
125
- echo_chunks=echo_chunks, # <-- PARÂMETRO NOVO
 
 
 
 
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"]}