euiia commited on
Commit
a7c0fd0
·
verified ·
1 Parent(s): eafff49

Update aduc_orchestrator.py

Browse files
Files changed (1) hide show
  1. aduc_orchestrator.py +8 -8
aduc_orchestrator.py CHANGED
@@ -19,7 +19,7 @@ import gradio as gr
19
  from PIL import Image, ImageOps
20
 
21
  from engineers.deformes4D import Deformes4DEngine
22
- from managers.gemini_manager import gemini_manager_singleton
23
  from engineers.deformes3D import deformes3d_engine_singleton
24
 
25
  # The logger is configured in app.py; here we just get the instance.
@@ -112,12 +112,12 @@ class AducOrchestrator:
112
  def task_generate_storyboard(self, prompt: str, num_keyframes: int, ref_image_paths: List[str],
113
  progress: gr.Progress) -> Tuple[List[str], str, Any]:
114
  """
115
- Delegates the task of creating the storyboard to the Scriptwriter (Gemini).
116
  """
117
- logger.info(f"Act 1, Scene 1: Script. Instructing Scriptwriter (Gemini) to create {num_keyframes} scenes from: '{prompt}'")
118
- progress(0.2, desc="Consulting AI Scriptwriter (Gemini)...")
119
 
120
- storyboard = gemini_manager_singleton.generate_storyboard(prompt, num_keyframes, ref_image_paths)
121
 
122
  logger.info(f"Scriptwriter returned the score: {storyboard}")
123
  self.director.update_state("storyboard", storyboard)
@@ -128,12 +128,12 @@ class AducOrchestrator:
128
  def task_select_keyframes(self, storyboard: List[str], base_ref_paths: List[str],
129
  pool_ref_paths: List[str]) -> List[str]:
130
  """
131
- Delegates to the Editor/Photographer (Gemini) the task of selecting the best images
132
  from a "scene bank" to match the script (Photographer Mode).
133
  """
134
- logger.info(f"Act 1, Scene 2 (Photographer Mode): Instructing Editor (Gemini) to select {len(storyboard)} keyframes.")
135
 
136
- selected_paths = gemini_manager_singleton.select_keyframes_from_pool(storyboard, base_ref_paths, pool_ref_paths)
137
 
138
  logger.info(f"Editor selected the following scenes: {[os.path.basename(p) for p in selected_paths]}")
139
  self.director.update_state("keyframes", selected_paths)
 
19
  from PIL import Image, ImageOps
20
 
21
  from engineers.deformes4D import Deformes4DEngine
22
+ from engineers.deformes2D_thinker import deformes2d_thinker_singleton
23
  from engineers.deformes3D import deformes3d_engine_singleton
24
 
25
  # The logger is configured in app.py; here we just get the instance.
 
112
  def task_generate_storyboard(self, prompt: str, num_keyframes: int, ref_image_paths: List[str],
113
  progress: gr.Progress) -> Tuple[List[str], str, Any]:
114
  """
115
+ Delegates the task of creating the storyboard to the Scriptwriter (deformes2D_thinker).
116
  """
117
+ logger.info(f"Act 1, Scene 1: Script. Instructing Scriptwriter (deformes2D_thinker) to create {num_keyframes} scenes from: '{prompt}'")
118
+ progress(0.2, desc="Consulting AI Scriptwriter (deformes2D_thinker)...")
119
 
120
+ storyboard = deformes2d_thinker_singleton.generate_storyboard(prompt, num_keyframes, ref_image_paths)
121
 
122
  logger.info(f"Scriptwriter returned the score: {storyboard}")
123
  self.director.update_state("storyboard", storyboard)
 
128
  def task_select_keyframes(self, storyboard: List[str], base_ref_paths: List[str],
129
  pool_ref_paths: List[str]) -> List[str]:
130
  """
131
+ Delegates to the Editor/Photographer (deformes2D_thinker) the task of selecting the best images
132
  from a "scene bank" to match the script (Photographer Mode).
133
  """
134
+ logger.info(f"Act 1, Scene 2 (Photographer Mode): Instructing Editor (deformes2D_thinker) to select {len(storyboard)} keyframes.")
135
 
136
+ selected_paths = deformes2d_thinker_singleton.select_keyframes_from_pool(storyboard, base_ref_paths, pool_ref_paths)
137
 
138
  logger.info(f"Editor selected the following scenes: {[os.path.basename(p) for p in selected_paths]}")
139
  self.director.update_state("keyframes", selected_paths)