Update aduc_framework/engineers/deformes2D_thinker.py

aduc_framework/engineers/deformes2D_thinker.py

@@ -2,11 +2,14 @@
 #
 # Copyright (C) 4 de Agosto de 2025  Carlos Rodrigues dos Santos
 #
-# Versão 3.0.0 (Autonomous Film Director)
+# Versão 4.0.0 (Dual-Role Autonomous Director & Cinematographer)
 #
-# Esta classe é o cérebro criativo do framework ADUC-SDR. Ela atua como um
-# Diretor de Cinema Autônomo, utilizando um LLM para gerenciar o roteiro,
-# a direção de arte e a continuidade da produção de keyframes em tempo real.
+# Esta classe é o cérebro criativo do framework. Ela utiliza um LLM para
+# desempenhar múltiplos papéis:
+# 1. Roteirista: Cria a lista inicial de atos.
+# 2. Diretor Autônomo: Gerencia a produção de keyframes, com poder de
+#    avaliar, corrigir e improvisar o roteiro.
+# 3. Cineasta: Cria os prompts de movimento que conectam os keyframes.
 
 import logging
 from pathlib import Path
@@ -20,9 +23,10 @@ logger = logging.getLogger(__name__)
 
 class Deformes2DThinker:
     """
-    O especialista cognitivo que atua como Diretor de Cinema Autônomo,
-    responsável por todas as decisões criativas na geração de keyframes.
+    O especialista cognitivo que atua como a inteligência criativa central,
+    desempenhando os papéis de Roteirista, Diretor Autônomo e Cineasta.
     """
+
     def _read_prompt_template(self, filename: str) -> str:
         """Lê um arquivo de template de prompt do diretório 'prompts'."""
         try:
@@ -32,19 +36,18 @@ class Deformes2DThinker:
         except FileNotFoundError:
             raise gr.Error(f"Arquivo de template de prompt não encontrado: {filename}")
 
+    # --- PAPEL 1: DIRETOR AUTÔNOMO (Para Deformes3D) ---
     def get_directorial_decision(self, context: Dict[str, Any]) -> Dict[str, Any]:
         """
-        Função principal do Diretor. Analisa o estado atual da produção, avalia
-        o trabalho anterior e retorna a próxima ação (avançar, corrigir, improvisar).
+        Função principal do Diretor. Analisa o estado da produção e retorna a
+        próxima ação (avançar, corrigir, improvisar) para a criação de KEYFRAMES.
         """
         try:
             template = self._read_prompt_template("autonomous_director_prompt.txt")
             
-            # Serializa o contexto complexo em strings legíveis para o LLM
             keyframes_str = "\n".join([f"- ID {kf['id']}: {kf['prompt_keyframe']}" for kf in context.get('keyframes_gerados', [])])
             script_str = "\n".join([f"- Ato {i+1}: {ato}" for i, ato in enumerate(context.get('roteiro_completo', []))])
 
-            # Preenche o template com todas as informações contextuais
             formatted_prompt = template.format(
                 prompt_geral=context.get('prompt_geral', ''),
                 midias_usuario=", ".join(context.get('midias_usuario', [])),
@@ -53,7 +56,6 @@ class Deformes2DThinker:
                 keyframes_gerados=keyframes_str if keyframes_str else "Nenhuma cena filmada ainda."
             )
             
-            # Prepara o payload para o Gemini. A imagem do último keyframe é crucial para a avaliação.
             prompt_parts = [formatted_prompt]
             keyframes_gerados = context.get('keyframes_gerados', [])
             if keyframes_gerados:
@@ -62,16 +64,13 @@ class Deformes2DThinker:
                 try:
                     prompt_parts.append(Image.open(last_keyframe_path))
                 except FileNotFoundError:
-                    logger.warning(f"Não foi possível encontrar a imagem do último keyframe em: {last_keyframe_path}")
+                    logger.warning(f"Não foi possível encontrar a imagem do último keyframe: {last_keyframe_path}")
 
-            # Chama o Gemini e espera uma decisão estruturada em JSON
             decision_json = gemini_manager_singleton.get_json_object(prompt_parts)
             return decision_json
 
         except Exception as e:
-            logger.error(f"O Diretor Autônomo (Gemini) falhou: {e}. Acionando fallback para 'avançar'.", exc_info=True)
-            # Fallback robusto: se o Gemini falhar, tentamos avançar com uma decisão genérica
-            # para não interromper a produção.
+            logger.error(f"O Diretor Autônomo (Gemini) falhou: {e}. Acionando fallback.", exc_info=True)
             return self._get_fallback_decision(context)
 
     def _get_fallback_decision(self, context: Dict[str, Any]) -> Dict[str, Any]:
@@ -89,6 +88,35 @@ class Deformes2DThinker:
             "is_cut": False
         }
 
+    # --- PAPEL 2: CINEASTA (Para Deformes4D) ---
+    def get_motion_decision(self, start_kf: Dict, end_kf: Dict, motion_history: str) -> str:
+        """
+        Atua como Cineasta para decidir o melhor `motion_prompt` que conecta
+        dois keyframes existentes, criando o MOVIMENTO.
+        """
+        try:
+            template = self._read_prompt_template("motion_director_prompt.txt")
+
+            formatted_prompt = template.format(
+                cena_atual_desc=start_kf.get('prompt_keyframe', 'início da cena'),
+                cena_futura_desc=end_kf.get('prompt_keyframe', 'fim da cena'),
+                historico_movimento=motion_history
+            )
+
+            prompt_parts = [
+                formatted_prompt,
+                "IMAGEM ATUAL:", Image.open(start_kf['caminho_pixel']),
+                "IMAGEM FUTURA:", Image.open(end_kf['caminho_pixel'])
+            ]
+
+            motion_prompt = gemini_manager_singleton.get_raw_text(prompt_parts)
+            return motion_prompt.strip().replace("`", "").replace("\"", "")
+
+        except Exception as e:
+            logger.error(f"O Cineasta (Gemini) falhou ao criar motion_prompt: {e}. Usando fallback.", exc_info=True)
+            return f"Uma transição cinematográfica suave de '{start_kf.get('prompt_keyframe', 'cena anterior')}' para '{end_kf.get('prompt_keyframe', 'próxima cena')}'."
+
+    # --- PAPEL 3: ROTEIRISTA (Para Orquestrador) ---
     def generate_storyboard(self, prompt: str, num_keyframes: int, ref_image_paths: List[str]) -> List[str]:
         """
         Atua como Roteirista para criar o roteiro inicial (lista de atos)
@@ -105,8 +133,9 @@ class Deformes2DThinker:
             storyboard = storyboard_data.get("scene_storyboard", [])
             if not storyboard or len(storyboard) != num_keyframes: 
                 logger.warning(f"Número de cenas gerado diferente do solicitado. Pedido: {num_keyframes}, Gerado: {len(storyboard)}")
-                # Truncate or pad to match the requested number
-                storyboard = (storyboard + [storyboard[-1]] * num_keyframes)[:num_keyframes]
+                # Garante que a lista tenha o tamanho exato solicitado
+                default_scene = storyboard[-1] if storyboard else "Cena de continuação."
+                storyboard = (storyboard + [default_scene] * num_keyframes)[:num_keyframes]
 
             return storyboard
         except Exception as e: