Update aduc_orchestrator.py
Browse files- aduc_orchestrator.py +3 -3
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
|
| 23 |
from engineers.deformes3D import Deformes3DEngine
|
| 24 |
|
| 25 |
# The logger is configured in app.py; here we just get the instance.
|
|
@@ -117,7 +117,7 @@ class AducOrchestrator:
|
|
| 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 =
|
| 121 |
|
| 122 |
logger.info(f"Scriptwriter returned the score: {storyboard}")
|
| 123 |
self.director.update_state("storyboard", storyboard)
|
|
@@ -133,7 +133,7 @@ class AducOrchestrator:
|
|
| 133 |
"""
|
| 134 |
logger.info(f"Act 1, Scene 2 (Photographer Mode): Instructing Editor (Gemini) to select {len(storyboard)} keyframes.")
|
| 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 managers.gemini_manager import gemini_manager_singleton
|
| 23 |
from engineers.deformes3D import Deformes3DEngine
|
| 24 |
|
| 25 |
# The logger is configured in app.py; here we just get the instance.
|
|
|
|
| 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)
|
|
|
|
| 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)
|