Update engineers/deformes3D.py

engineers/deformes3D.py

@@ -29,7 +29,6 @@ logger = logging.getLogger(__name__)
 class Deformes3DEngine:
     """
     ADUC Specialist for static image (keyframe) generation.
-    Primary engine is FLUX, with an experimental LTX enrichment step.
     """
     def __init__(self, workspace_dir):
         self.workspace_dir = workspace_dir
@@ -38,7 +37,7 @@ class Deformes3DEngine:
 
     def _generate_single_keyframe(self, prompt: str, reference_images: list[Image.Image], output_filename: str, width: int, height: int, callback: callable = None) -> str:
         """
-        Low-level function that generates a single image using the FLUX helper.
+        Low-level function that generates a single image using the LTX helper.
         """
         logger.info(f"Generating keyframe '{output_filename}' with prompt: '{prompt}'")
         generated_image = self.image_generation_helper.generate_image(
@@ -52,9 +51,7 @@ class Deformes3DEngine:
 
     def generate_keyframes_from_storyboard(self, storyboard: list, initial_ref_path: str, global_prompt: str, keyframe_resolution: int, general_ref_paths: list, progress_callback_factory: callable = None):
         """
-        Orchestrates the generation of all keyframes. For each keyframe, first
-        generates a version with FLUX, and then an "enriched" version with LTX
-        for direct comparison.
+        Orchestrates the generation of all keyframes. 
         """
         current_base_image_path = initial_ref_path
         previous_prompt = "N/A (initial reference image)"
@@ -63,7 +60,7 @@ class Deformes3DEngine:
         target_resolution_tuple = (width, height)
         
         num_keyframes_to_generate = len(storyboard) - 1
-        logger.info(f"IMAGE SPECIALIST: Received order to generate {num_keyframes_to_generate} keyframes (FLUX + LTX versions).")
+        logger.info(f"IMAGE SPECIALIST: Received order to generate {num_keyframes_to_generate} keyframes (LTX versions).")
 
         for i in range(num_keyframes_to_generate):
             scene_index = i + 1
@@ -74,26 +71,26 @@ class Deformes3DEngine:
             logger.info(f"--> Generating Keyframe {scene_index}/{num_keyframes_to_generate}...")
 
             # --- STEP A: Generate with FLUX (Primary Method) ---
-            logger.info(f"    - Step A: Generating with FLUX...")
+            logger.info(f"    - Step A: Generating with keyframe...")
             
-            flux_prompt = deformes2d_thinker_singleton.get_anticipatory_keyframe_prompt(
+            img_prompt = deformes2d_thinker_singleton.get_anticipatory_keyframe_prompt(
                 global_prompt=global_prompt, scene_history=previous_prompt,
                 current_scene_desc=current_scene, future_scene_desc=future_scene,
                 last_image_path=current_base_image_path, fixed_ref_paths=general_ref_paths
             )
             
-            flux_ref_paths = list(set([current_base_image_path] + general_ref_paths))
-            flux_ref_images = [Image.open(p) for p in flux_ref_paths]
+            #flux_ref_paths = list(set([current_base_image_path] + general_ref_paths))
+            #flux_ref_images = [Image.open(p) for p in flux_ref_paths]
             
-            flux_keyframe_path = self._generate_single_keyframe(
-                prompt=flux_prompt, reference_images=flux_ref_images,
-                output_filename=f"keyframe_{scene_index}_flux.png", width=width, height=height,
-                callback=progress_callback_flux
-            )
-            final_keyframes_gallery.append(flux_keyframe_path)
+            #flux_keyframe_path = self._generate_single_keyframe(
+            #    prompt=img_prompt, reference_images=flux_ref_images,
+            #    output_filename=f"keyframe_{scene_index}_flux.png", width=width, height=height,
+            #    callback=progress_callback_flux
+            #)
+            #final_keyframes_gallery.append(flux_keyframe_path)
             
             # --- STEP B: LTX Enrichment Experiment ---
-            logger.info(f"    - Step B: Generating enrichment with LTX...")
+            #logger.info(f"    - Step B: Generating enrichment with LTX...")
 
             ltx_context_paths = []
             context_paths = []
@@ -102,7 +99,6 @@ class Deformes3DEngine:
             ltx_context_paths = list(reversed(context_paths))
             logger.info(f"    - LTX Context Order (Reversed): {[os.path.basename(p) for p in ltx_context_paths]}")
 
-            
             ltx_conditioning_items = []
             
             weight = 0.6
@@ -121,7 +117,7 @@ class Deformes3DEngine:
             generated_latents, _ = ltx_manager_singleton.generate_latent_fragment(
                 height=height, width=width,
                 conditioning_items_data=ltx_conditioning_items,
-                motion_prompt=flux_prompt,
+                motion_prompt=img_prompt,
                 video_total_frames=48,
                 video_fps=24,
                 **ltx_base_params
@@ -136,10 +132,10 @@ class Deformes3DEngine:
             final_keyframes_gallery.append(ltx_keyframe_path)
             
             # Use the FLUX keyframe as the base for the next iteration to maintain the primary narrative path
-            current_base_image_path = flux_keyframe_path 
-            previous_prompt = flux_prompt
+            current_base_image_path = ltx_keyframe_path #flux_keyframe_path 
+            previous_prompt = img_prompt
 
-        logger.info(f"IMAGE SPECIALIST: Generation of all keyframe versions (FLUX + LTX) complete.")
+        logger.info(f"IMAGE SPECIALIST: Generation of all keyframe versions (LTX) complete.")
         return final_keyframes_gallery
 
     # --- HELPER FUNCTIONS ---