Create deformes3D_thinker.py

engineers/deformes3D_thinker.py

@@ -0,0 +1,77 @@
+# engineers/deformes3D_thinker.py
+#
+# Copyright (C) 2025 Carlos Rodrigues dos Santos
+#
+# Version: 1.0.0
+#
+# This file defines the Deformes3DThinker, the tactical cinematic director
+# of the ADUC framework. Its sole responsibility is to analyze the immediate
+# temporal context (past, present, future keyframes) to generate the optimal
+# motion prompt for the video generation engine.
+
+import logging
+from pathlib import Path
+from PIL import Image
+import gradio as gr
+
+from managers.gemini_manager import gemini_manager_singleton
+
+logger = logging.getLogger(__name__)
+
+class Deformes3DThinker:
+    """
+    The tactical specialist that handles cinematic decision-making.
+    """
+    def _read_prompt_template(self, filename: str) -> str:
+        """Reads a prompt template file from the 'prompts' directory."""
+        try:
+            prompts_dir = Path(__file__).resolve().parent.parent / "prompts"
+            with open(prompts_dir / filename, "r", encoding="utf-8") as f:
+                return f.read()
+        except FileNotFoundError:
+            raise gr.Error(f"Prompt template file not found: prompts/{filename}")
+
+    def get_cinematic_decision(self, global_prompt: str, story_history: str, 
+                               past_keyframe_path: str, present_keyframe_path: str, future_keyframe_path: str,
+                               past_scene_desc: str, present_scene_desc: str, future_scene_desc: str) -> dict:
+        """
+        Acts as a Film Director to make editing decisions and generate motion prompts
+        by analyzing the past, present, and future visual and narrative context.
+        """
+        try:
+            template = self._read_prompt_template("cinematic_director_prompt.txt")
+            prompt_text = template.format(
+                global_prompt=global_prompt, 
+                story_history=story_history,
+                past_scene_desc=past_scene_desc,
+                present_scene_desc=present_scene_desc,
+                future_scene_desc=future_scene_desc
+            )
+            
+            prompt_parts = [
+                prompt_text,
+                "[PAST_IMAGE]:", Image.open(past_keyframe_path),
+                "[PRESENT_IMAGE]:", Image.open(present_keyframe_path),
+                "[FUTURE_IMAGE]:", Image.open(future_keyframe_path)
+            ]
+            
+            decision_data = gemini_manager_singleton.get_json_object(prompt_parts)
+
+            if "transition_type" not in decision_data or "motion_prompt" not in decision_data:
+                raise ValueError("AI response (Cinematographer) is malformed. Missing 'transition_type' or 'motion_prompt'.")
+            
+            # --- LOGGING ADICIONADO ---
+            logger.info(f"Deformes3DThinker Decision -> Transition: '{decision_data['transition_type']}', Motion Prompt: '{decision_data['motion_prompt']}'")
+            
+            return decision_data
+        except Exception as e:
+            logger.error(f"The Film Director (Deformes3D Thinker) failed: {e}. Using fallback.", exc_info=True)
+            fallback_prompt = f"A smooth, continuous cinematic transition from '{present_scene_desc}' to '{future_scene_desc}'."
+            logger.info(f"Deformes3DThinker Fallback -> Transition: 'continuous', Motion Prompt: '{fallback_prompt}'")
+            return {
+                "transition_type": "continuous",
+                "motion_prompt": fallback_prompt
+            }
+
+# --- Singleton Instance ---
+deformes3d_thinker_singleton = Deformes3DThinker()