Update aduc_orchestrator.py
Browse files- 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
|
| 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 (
|
| 116 |
"""
|
| 117 |
-
logger.info(f"Act 1, Scene 1: Script. Instructing Scriptwriter (
|
| 118 |
-
progress(0.2, desc="Consulting AI Scriptwriter (
|
| 119 |
|
| 120 |
-
storyboard =
|
| 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 (
|
| 132 |
from a "scene bank" to match the script (Photographer Mode).
|
| 133 |
"""
|
| 134 |
-
logger.info(f"Act 1, Scene 2 (Photographer Mode): Instructing Editor (
|
| 135 |
|
| 136 |
-
selected_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)
|