Update app.py

app.py

@@ -72,7 +72,7 @@ def preprocess_base_images_wrapper(uploaded_files):
     processed_paths = [aduc.process_image_for_story(f.name, 480, f"ref_processed_{i}.png") for i, f in enumerate(uploaded_files)]
     return gr.update(value=processed_paths)
 
-def run_mode_a_wrapper(prompt, num_keyframes, ref_files, resolution_str, duration_per_fragment, progress=gr.Progress()):
+def run_mode_a_wrapper(prompt, num_keyframes, ref_files, resolution_str, progress=gr.Progress()):
     if not ref_files: 
         raise gr.Error("Por favor, forneça pelo menos uma imagem de referência.")
     
@@ -116,9 +116,10 @@ def run_mode_b_wrapper(prompt, num_keyframes, ref_files, progress=gr.Progress())
     
     return gr.update(value=storyboard), gr.update(value=selected_keyframes), gr.update(visible=True, open=True)
 
-# --- ASSINATURA DA FUNÇÃO WRAPPER ATUALIZADA ---
-def run_video_production_wrapper(keyframes, prompt, duration, 
-                                 trim_chunks, echo_chunks,   # <-- PARÂMETROS ATUALIZADOS
+def run_video_production_wrapper(keyframes, prompt, 
+                                 n_chunks_to_generate, video_end_chunk,
+                                 eco_start_chunk, handler_start_chunk,
+                                 handler_frame_target,
                                  handler_strength, destination_convergence_strength,
                                  video_resolution, use_cont, use_cine,
                                  progress=gr.Progress()):
@@ -132,10 +133,11 @@ def run_video_production_wrapper(keyframes, prompt, duration,
     video_fragments_so_far = []
     final_movie_path = None
     
-    # --- CHAMADA PARA O ORQUESTRADOR ATUALIZADA ---
     for update in aduc.task_produce_final_movie_with_feedback(
-        keyframes, prompt, duration, 
-        int(trim_chunks), int(echo_chunks), # <-- Passando os novos valores (convertidos para int)
+        keyframes, prompt,
+        int(n_chunks_to_generate), int(video_end_chunk),
+        int(eco_start_chunk), int(handler_start_chunk),
+        int(handler_frame_target),
         handler_strength, destination_convergence_strength,
         resolution, use_cont, use_cine, progress
     ):
@@ -152,22 +154,14 @@ def run_video_production_wrapper(keyframes, prompt, duration,
     }
 
 def get_log_content():
-    """Função para ler e retornar o conteúdo do arquivo de log."""
     try:
         with open(LOG_FILE_PATH, "r", encoding="utf-8") as f:
             return f.read()
     except FileNotFoundError:
         return "Arquivo de log ainda não criado. Inicie uma geração."
 
-# --- FUNÇÃO DE TRADUÇÃO ATUALIZADA (NOVOS COMPONENTES) ---
 def update_ui_language(lang_code):
     lang_map = i18n.get(lang_code, i18n.get('en', {}))
-    # Adicionando traduções para os novos sliders (supondo que existam no i18n.json)
-    trim_chunks_label = lang_map.get('trim_chunks_label', 'Trim Chunks')
-    trim_chunks_info = lang_map.get('trim_chunks_info', 'How many chunks (8 frames each) to "trim" from the end before extracting the continuity guide (Echo).')
-    echo_chunks_label = lang_map.get('echo_chunks_label', 'Echo Chunks (Memory)')
-    echo_chunks_info = lang_map.get('echo_chunks_info', 'The size of the continuity guide (Echo) in chunks. This is the "memory" passed to the next scene.')
-    
     return {
         title_md: gr.update(value=f"# {lang_map.get('app_title')}"),
         subtitle_md: gr.update(value=lang_map.get('app_subtitle')),
@@ -186,9 +180,11 @@ def update_ui_language(lang_code):
         continuity_director_checkbox: gr.update(label=lang_map.get('continuity_director_label')),
         cinematographer_checkbox: gr.update(label=lang_map.get('cinematographer_label')),
         
-        # --- ATUALIZANDO OS NOVOS SLIDERS ---
-        trim_chunks_slider: gr.update(label=trim_chunks_label, info=trim_chunks_info),
-        echo_chunks_slider: gr.update(label=echo_chunks_label, info=echo_chunks_info),
+        n_chunks_to_generate_slider: gr.update(label=lang_map.get('n_chunks_generate_label'), info=lang_map.get('n_chunks_generate_info')),
+        video_end_chunk_slider: gr.update(label=lang_map.get('video_end_chunk_label'), info=lang_map.get('video_end_chunk_info')),
+        eco_start_chunk_slider: gr.update(label=lang_map.get('eco_start_chunk_label'), info=lang_map.get('eco_start_chunk_info')),
+        handler_start_chunk_slider: gr.update(label=lang_map.get('handler_start_chunk_label'), info=lang_map.get('handler_start_chunk_info')),
+        handler_frame_target_slider: gr.update(label=lang_map.get('handler_frame_target_label'), info=lang_map.get('handler_frame_target_info')),
         
         forca_guia_slider: gr.update(label=lang_map.get('forca_guia_label'), info=lang_map.get('forca_guia_info')),
         convergencia_destino_slider: gr.update(label=lang_map.get('convergencia_final_label'), info=lang_map.get('convergencia_final_info')),
@@ -215,8 +211,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     with gr.Accordion(default_lang.get('step1_accordion'), open=True) as step1_accordion:
         prompt_input = gr.Textbox(label=default_lang.get('prompt_label'), value="A majestic lion walks across the savanna, sits down, and then roars at the setting sun.")
         with gr.Row():
-            num_keyframes_slider = gr.Slider(minimum=3, maximum=100, value=3, step=1, label=default_lang.get('keyframes_label'), info="Mínimo de 3 para a lógica do cineasta.")
-            duration_per_fragment_slider = gr.Slider(label=default_lang.get('duration_label'), minimum=2.0, maximum=10.0, value=4.0, step=0.1)
+            num_keyframes_slider = gr.Slider(minimum=3, maximum=100, value=5, step=1, label=default_lang.get('keyframes_label'), info="Mínimo de 3 para a lógica do cineasta.")
+            duration_per_fragment_slider = gr.Slider(label=default_lang.get('duration_label'), minimum=2.0, maximum=10.0, value=4.0, step=0.1, visible=False) # Escondido, mas mantido por dependências
         ref_image_input = gr.File(label=default_lang.get('ref_images_label'), file_count="multiple", file_types=["image"])
         with gr.Row():
             storyboard_and_keyframes_button = gr.Button(default_lang.get('storyboard_and_keyframes_button'), variant="primary")
@@ -231,16 +227,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             continuity_director_checkbox = gr.Checkbox(label=default_lang.get('continuity_director_label'), value=True)
             cinematographer_checkbox = gr.Checkbox(label=default_lang.get('cinematographer_label'), value=True, visible=False)
         
-        # --- INÍCIO DA ATUALIZAÇÃO DA UI (CONTROLES DE CHUNK) ---
-        gr.Markdown("--- \n**Controles de Continuidade e Edição (Baseado em Chunks):**")
+        gr.Markdown("--- \n**Controles de Geração e Continuidade (Avançado):**")
+        with gr.Row():
+            n_chunks_to_generate_slider = gr.Slider(minimum=7, maximum=25, value=13, step=1, label="Chunks a Gerar", info="Total de chunks a gerar em cada passo. Mais chunks = mais 'cauda' para as guias.")
+            video_end_chunk_slider = gr.Slider(minimum=2, maximum=20, value=9, step=1, label="Fim do Clipe (Chunk)", info="O vídeo final usará os chunks de 0 até este valor. (Ex: 9 -> usa chunks 0-8).")
         with gr.Row():
-            trim_chunks_slider = gr.Slider(minimum=0, maximum=10, value=2, step=1, 
-                                             label="Chunks de Corte", 
-                                             info="Quantos chunks (de 8 frames) 'aparar' do final antes de extrair a guia de continuidade (Eco).")
-            echo_chunks_slider = gr.Slider(minimum=1, maximum=3, value=1, step=1, 
-                                             label="Chunks de Eco (Memória)", 
-                                             info="O tamanho da guia de continuidade (Eco) em chunks. Esta é a 'memória' que passa para a próxima cena.")
-        # --- FIM DA ATUALIZAÇÃO DA UI ---
+            eco_start_chunk_slider = gr.Slider(minimum=2, maximum=20, value=9, step=1, label="Início do Eco (Chunk)", info="Índice do chunk onde a guia de memória (eco) começa. (Ex: 9 -> usa chunks 9 e 10).")
+            handler_start_chunk_slider = gr.Slider(minimum=4, maximum=22, value=11, step=1, label="Início do Handler (Chunk)", info="Índice do chunk onde a guia de evolução (handler) começa. (Ex: 11 -> usa chunks 11 e 12).")
+        with gr.Row():
+             handler_frame_target_slider = gr.Slider(minimum=8, maximum=80, value=16, step=8, label="Alvo do Handler (Frame)", info="Frame exato onde a guia de evolução será aplicada na próxima geração.")
         
         gr.Markdown("**Controle de Influência:**")
         with gr.Row():
@@ -257,20 +252,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         update_log_button = gr.Button("Atualizar Log")
 
     # --- 5. CONEXÕES DA UI ---
-    # Removendo os componentes antigos e adicionando os novos para a tradução
-    all_ui_components_dict = update_ui_language('pt')
-    # Adicionando os novos componentes manualmente ao dicionário para garantir que sejam incluídos
-    all_ui_components_dict[trim_chunks_slider] = None
-    all_ui_components_dict[echo_chunks_slider] = None
-    all_ui_components = list(all_ui_components_dict.keys())
-    
+    all_ui_components = list(update_ui_language('pt').keys())
     lang_selector.change(fn=update_ui_language, inputs=lang_selector, outputs=all_ui_components)
     
     ref_image_input.upload(fn=preprocess_base_images_wrapper, inputs=ref_image_input, outputs=ref_image_input)
 
     storyboard_and_keyframes_button.click(
         fn=run_mode_a_wrapper, 
-        inputs=[prompt_input, num_keyframes_slider, ref_image_input, resolution_selector, duration_per_fragment_slider], 
+        inputs=[prompt_input, num_keyframes_slider, ref_image_input, resolution_selector], 
         outputs=[storyboard_output, keyframe_gallery, step3_accordion]
     )
     
@@ -280,13 +269,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         outputs=[storyboard_output, keyframe_gallery, step3_accordion]
     )
     
-    # --- CHAMADA DO BOTÃO ATUALIZADA ---
     produce_button.click(
         fn=run_video_production_wrapper,
         inputs=[
-            keyframe_gallery, prompt_input, duration_per_fragment_slider, 
-            trim_chunks_slider,           # <-- NOVO INPUT
-            echo_chunks_slider,           # <-- NOVO INPUT
+            keyframe_gallery, prompt_input, 
+            n_chunks_to_generate_slider,
+            video_end_chunk_slider,
+            eco_start_chunk_slider,
+            handler_start_chunk_slider,
+            handler_frame_target_slider,
             forca_guia_slider,
             convergencia_destino_slider,
             resolution_selector, continuity_director_checkbox, cinematographer_checkbox