Update app.py

app.py

@@ -2,9 +2,9 @@
 #
 # Copyright (C) August 4, 2025  Carlos Rodrigues dos Santos
 #
-# Versão 9.0.0 (Live Dramatization UI)
-# Esta versão implementa a lógica de exibição do chat em tempo real,
-# criando o efeito de "digitação" e "pensamento" 100% na camada da UI.
+# Versão 9.1.0 (Reactivated Keyframe Generation UI)
+# Esta versão reintroduz a UI e a lógica para a Etapa 2 (Geração de Keyframes),
+# conectando o resultado da pré-produção (roteiro) com a engine de criação de imagens.
 
 import gradio as gr
 import yaml
@@ -13,7 +13,6 @@ import os
 import sys
 import shutil
 import time
-from typing import Dict, Any, Generator
 
 # --- 1. IMPORTAÇÃO DO FRAMEWORK E CONFIGURAÇÃO ---
 import aduc_framework
@@ -44,7 +43,6 @@ try:
     logger.info("Interface Gradio inicializada e conectada ao Aduc Framework.")
 except Exception as e:
     logger.critical(f"ERRO CRÍTICO durante a inicialização: {e}", exc_info=True)
-    # Bloco de erro para a UI se o framework não iniciar
     with gr.Blocks() as demo_error:
         gr.Markdown("# ERRO CRÍTICO NA INICIALIZAÇÃO")
         gr.Markdown("Não foi possível iniciar o Aduc Framework. Verifique os logs (`aduc_log.txt`) para mais detalhes.")
@@ -52,12 +50,12 @@ except Exception as e:
     demo_error.launch()
     exit()
 
-# --- 2. FUNÇÃO WRAPPER (UI <-> FRAMEWORK) ---
+# --- 2. FUNÇÕES WRAPPER (UI <-> FRAMEWORK) ---
 
 def run_pre_production_wrapper(prompt, num_scenes, ref_files, resolution_str, fast_mode, progress=gr.Progress(track_tqdm=True)):
     """
-    Wrapper que consome os yields do orquestrador e os "dramatiza" para a UI,
-    criando o efeito de chat em tempo real SEM alterar o backend.
+    Wrapper para a Etapa 1 (Pré-Produção). Consome os yields do orquestrador e
+    "dramatiza" o chat para a UI. Ao final, habilita a Etapa 2.
     """
     if not ref_files:
         raise gr.Error("Por favor, forneça pelo menos uma imagem de referência.")
@@ -70,102 +68,133 @@ def run_pre_production_wrapper(prompt, num_scenes, ref_files, resolution_str, fa
         resolution=target_resolution, duration_per_fragment=4.0, fast_mode=fast_mode
     )
 
-    # --- ESTADO DA UI ---
     chatbot_display_history = []
     fully_displayed_message_count = 0
+    final_dna = {}
 
-    # Consome os yields do orquestrador em tempo real
     for update in aduc.task_pre_production(params, progress_callback=progress):
-        
         backend_chat_dna = update.get("chat_dna", [])
         dna_snapshot = update.get("dna_snapshot", gr.skip())
+        if update.get("status") == "complete":
+            final_dna = update.get("dna", {})
 
-        # --- LÓGICA DE DRAMATIZAÇÃO ---
-        # Processa todas as novas mensagens que chegaram do backend
         while len(backend_chat_dna) > fully_displayed_message_count:
-            
             new_message_obj = backend_chat_dna[fully_displayed_message_count]
             role = f"**{new_message_obj.get('role', 'Sistema')}**"
             content_to_type = new_message_obj.get('content', '')
-
-            # Adiciona um placeholder para a nova mensagem na tela
             chatbot_display_history.append([role, ""])
-            
             full_typed_message = ""
             for char in content_to_type:
                 full_typed_message += char
-                chatbot_display_history[-1][1] = full_typed_message + "▌"  # Adiciona cursor
-                yield { 
-                    chat_history_chatbot: chatbot_display_history, 
-                    dna_display: dna_snapshot 
-                }
+                chatbot_display_history[-1][1] = full_typed_message + "▌"
+                yield { chat_history_chatbot: chatbot_display_history, dna_display: dna_snapshot }
                 time.sleep(0.02)
-
-            chatbot_display_history[-1][1] = full_typed_message # Remove cursor
-            yield { 
-                chat_history_chatbot: chatbot_display_history, 
-                dna_display: dna_snapshot 
-            }
-
+            chatbot_display_history[-1][1] = full_typed_message
+            yield { chat_history_chatbot: chatbot_display_history, dna_display: dna_snapshot }
             fully_displayed_message_count += 1
-            
-            # --- EFEITO DE "PENSANDO" ---
             if "Llama, preciso da sua ajuda" in content_to_type:
                 chatbot_display_history.append([f"**Llama (IA Criativa)**", "..."])
-                yield { 
-                    chat_history_chatbot: chatbot_display_history, 
-                    dna_display: dna_snapshot 
-                }
+                yield { chat_history_chatbot: chatbot_display_history, dna_display: dna_snapshot }
                 time.sleep(1.5)
-                chatbot_display_history.pop() # Remove o "..."
+                chatbot_display_history.pop()
+
+        yield { chat_history_chatbot: chatbot_display_history, dna_display: dna_snapshot }
 
-        # Garante que o DNA continue atualizando mesmo sem novas mensagens de chat
+    # Retorno final para habilitar a próxima etapa
+    yield {
+        chat_history_chatbot: chatbot_display_history,
+        dna_display: final_dna,
+        step2_accordion: gr.update(visible=True, open=True) # HABILITA A ETAPA 2
+    }
+
+
+def run_keyframe_generation_wrapper(current_state_dict, progress=gr.Progress(track_tqdm=True)):
+    """
+    Wrapper para a Etapa 2 (Geração de Keyframes).
+    Chama o orquestrador e atualiza a galeria em tempo real.
+    """
+    for update in aduc.task_generate_keyframes(current_state_dict, progress_callback=progress):
+        keyframe_paths = update.get("keyframe_paths", [])
+        
         yield {
-            chat_history_chatbot: chatbot_display_history,
-            dna_display: dna_snapshot,
+            keyframe_gallery: gr.update(value=keyframe_paths),
+            generation_state_holder: update.get("updated_state", gr.skip())
         }
+        
+        if update.get("status") == "keyframes_complete":
+            yield {
+                keyframe_gallery: gr.update(value=keyframe_paths),
+                generation_state_holder: update.get("updated_state"),
+                step3_accordion: gr.update(visible=True, open=True) # HABILITA A ETAPA 3
+            }
+            return
 
 # --- 3. DEFINIÇÃO DA UI ---
 with gr.Blocks(css="style.css") as demo:
     generation_state_holder = gr.State(value={})
 
     gr.Markdown("<h1>ADUC-SDR 🎬 - A Sala de Roteiristas de IA</h1>")
-    gr.Markdown("Observe em tempo real a equipe de IAs especialistas colaborar para criar seu filme.")
-
-    with gr.Row():
-        with gr.Column(scale=1):
-            with gr.Accordion("🤖 A Equipe de IA (Ao Vivo)", open=True):
-                chat_history_chatbot = gr.Chatbot(
-                    label="Conversa da Equipe",
-                    bubble_full_width=False,
-                    height=500,
-                )
-            
-        with gr.Column(scale=1):
-            with gr.Accordion("🧬 DNA da Produção (Ao Vivo)", open=True):
-                dna_display = gr.JSON(label="DNA em Construção", scale=2)
-            with gr.Accordion("🎞️ Resultado Final", open=True):
-                 final_video_output = gr.Video(label="Filme Concluído", interactive=False)
-
-    with gr.Accordion("📋 Briefing do Projeto (Sua Entrada)", open=True):
+    
+    # ETAPA 1
+    with gr.Accordion("Etapa 1: Planejamento do Roteiro", open=True):
         prompt_input = gr.Textbox(label="Ideia Geral do Filme", value="Um leão majestoso caminha pela savana e ruge para o sol poente.")
         with gr.Row():
             resolution_selector = gr.Radio(["512x512", "768x768"], value="512x512", label="Resolução Base")
             num_scenes_slider = gr.Slider(minimum=2, maximum=5, value=3, step=1, label="Número de Cenas")
         ref_image_input = gr.File(label="Imagens de Referência", file_count="multiple", file_types=["image"])
         fast_mode_checkbox = gr.Checkbox(label="Modo Rápido (Planejamento em etapa única)", value=False)
-        start_button = gr.Button("🎬 Enviar Briefing para a Equipe de IA", variant="primary")
+        start_planning_button = gr.Button("📝 Iniciar Planejamento (Etapa 1)", variant="primary")
 
-    with gr.Accordion("Etapa 2 & 3: Produção e Pós-Produção", open=False, visible=False) as production_accordion:
-        gr.Markdown("Após a pré-produção, os controles para gerar o vídeo final aparecerão aqui.")
+    with gr.Accordion("🧠 Diário do Planejador (Ao Vivo)", open=False) as planning_log_accordion:
+        with gr.Row():
+            chat_history_chatbot = gr.Chatbot(label="Conversa da Equipe", bubble_full_width=False, height=500, scale=1)
+            dna_display = gr.JSON(label="DNA da Produção (em construção)", scale=1)
+
+    # ETAPA 2 (RESTAURADA)
+    with gr.Accordion("Etapa 2: Geração de Keyframes (Cenas-Chave)", open=False, visible=False) as step2_accordion:
+        gr.Markdown("O roteiro foi aprovado. Agora, o Diretor de IA irá criar as cenas-chave (keyframes) para cada ato.")
+        start_keyframe_button = gr.Button("🖼️ Gerar Keyframes a partir do Roteiro (Etapa 2)", variant="primary")
+        keyframe_gallery = gr.Gallery(label="Keyframes Gerados (em tempo real)", visible=True, object_fit="contain", height="auto")
+
+    # ETAPA 3
+    with gr.Accordion("Etapa 3: Produção do Vídeo Original", open=False, visible=False) as step3_accordion:
+        gr.Markdown("Com os keyframes definidos, vamos gerar o vídeo principal.")
+        # ... (Sliders e botão de produção aqui, se necessário) ...
+        produce_original_button = gr.Button("🎬 Produzir Vídeo Original (Etapa 3)", variant="primary")
+
+    # SAÍDA E LOGS
+    with gr.Accordion("🎞️ Resultado Final", open=True):
+        final_video_output = gr.Video(label="Filme Concluído", interactive=False)
+    # ... (outros componentes de log/pós-produção podem ser adicionados aqui) ...
 
 # --- 4. CONEXÕES DE EVENTOS ---
-    start_button.click(
+
+    # Fluxo da Etapa 1
+    start_planning_button.click(
+        fn=lambda: gr.update(open=True), 
+        outputs=[planning_log_accordion]
+    ).then(
         fn=run_pre_production_wrapper, 
         inputs=[prompt_input, num_scenes_slider, ref_image_input, resolution_selector, fast_mode_checkbox], 
-        outputs=[chat_history_chatbot, dna_display] # Removidos os outputs que não são mais necessários aqui
+        outputs=[chat_history_chatbot, dna_display, step2_accordion]
     )
 
+    # Garante que o estado seja atualizado com o DNA final da Etapa 1
+    dna_display.change(
+        fn=lambda data: data, 
+        inputs=dna_display, 
+        outputs=generation_state_holder
+    )
+
+    # Fluxo da Etapa 2
+    start_keyframe_button.click(
+        fn=run_keyframe_generation_wrapper,
+        inputs=[generation_state_holder],
+        outputs=[keyframe_gallery, generation_state_holder, step3_accordion]
+    )
+
+    # (Conexão para Etapa 3 e pós-produção será adicionada aqui depois)
+
 # --- 5. INICIALIZAÇÃO DA APLICAÇÃO ---
 if __name__ == "__main__":
     if os.path.exists(WORKSPACE_DIR):