Update app.py

app.py

@@ -2,7 +2,7 @@
 #
 # Copyright (C) August 4, 2025  Carlos Rodrigues dos Santos
 #
-# Version: 2.0.1
+# Version: 2.0.2
 #
 # Contact:
 # Carlos Rodrigues dos Santos
@@ -194,8 +194,8 @@ def run_upscaler_wrapper(latent_paths, chunk_size, progress=gr.Progress()):
     }
 
     final_path = None
-    for path in aduc.task_run_latent_upscaler(latent_paths, int(chunk_size), progress=progress):
-        final_path = path['final_path']
+    for update in aduc.task_run_latent_upscaler(latent_paths, int(chunk_size), progress=progress):
+        final_path = update['final_path']
 
     yield {
         upscaler_video_output: gr.update(value=final_path, label="✅ Latent Upscale Complete"),
@@ -215,8 +215,8 @@ def run_hd_wrapper(source_video, model_version, steps, global_prompt, progress=g
     }
     
     final_path = None
-    for path in aduc.task_run_hd_mastering(source_video, model_version, int(steps), global_prompt, progress=progress):
-        final_path = path['final_path']
+    for update in aduc.task_run_hd_mastering(source_video, model_version, int(steps), global_prompt, progress=progress):
+        final_path = update['final_path']
 
     yield {
         hd_video_output: gr.update(value=final_path, label="✅ HD Mastering Complete"),
@@ -238,8 +238,8 @@ def run_audio_wrapper(source_video, audio_prompt, global_prompt, progress=gr.Pro
     final_audio_prompt = audio_prompt if audio_prompt and audio_prompt.strip() else global_prompt
 
     final_path = None
-    for path in aduc.task_run_audio_generation(source_video, final_audio_prompt, progress=progress):
-        final_path = path['final_path']
+    for update in aduc.task_run_audio_generation(source_video, final_audio_prompt, progress=progress):
+        final_path = update['final_path']
 
     yield {
         audio_video_output: gr.update(value=final_path, label="✅ Audio Generation Complete"),
@@ -258,9 +258,11 @@ def update_ui_language(lang_emoji):
     lang_code = lang_code_map.get(lang_emoji, "en")
     lang_map = i18n.get(lang_code, i18n.get('en', {}))
     return {
+        # General
         title_md: gr.update(value=f"# {lang_map.get('app_title')}"),
         subtitle_md: gr.update(value=lang_map.get('app_subtitle')),
         lang_selector: gr.update(label=lang_map.get('lang_selector_label')),
+        # Step 1: Pre-Production
         step1_accordion: gr.update(label=lang_map.get('step1_accordion')),
         prompt_input: gr.update(label=lang_map.get('prompt_label')),
         ref_image_input: gr.update(label=lang_map.get('ref_images_label')),
@@ -271,6 +273,7 @@ def update_ui_language(lang_emoji):
         step1_mode_b_info_md: gr.update(value=f"*{lang_map.get('step1_mode_b_info')}*"),
         storyboard_output: gr.update(label=lang_map.get('storyboard_output_label')),
         keyframe_gallery: gr.update(label=lang_map.get('keyframes_gallery_label')),
+        # Step 3: Production
         step3_accordion: gr.update(label=lang_map.get('step3_accordion')),
         step3_description_md: gr.update(value=lang_map.get('step3_description')),
         produce_original_button: gr.update(value=lang_map.get('produce_original_button')),
@@ -283,6 +286,7 @@ def update_ui_language(lang_emoji):
         guidance_scale_slider: gr.update(label=lang_map.get('guidance_scale_label'), info=lang_map.get('guidance_scale_info')),
         stg_scale_slider: gr.update(label=lang_map.get('stg_scale_label'), info=lang_map.get('stg_scale_info')),
         inference_steps_slider: gr.update(label=lang_map.get('steps_label'), info=lang_map.get('steps_info')),
+        # Step 4: Post-Production
         step4_accordion: gr.update(label=lang_map.get('step4_accordion')),
         step4_description_md: gr.update(value=lang_map.get('step4_description')),
         sub_step_a_accordion: gr.update(label=lang_map.get('sub_step_a_upscaler')),
@@ -301,6 +305,7 @@ def update_ui_language(lang_emoji):
         audio_options_accordion: gr.update(label=lang_map.get('audio_options')),
         audio_prompt_input: gr.update(label=lang_map.get('audio_prompt_label'), info=lang_map.get('audio_prompt_info')),
         run_audio_button: gr.update(value=lang_map.get('run_audio_button')),
+        # Final Outputs & Logs
         final_video_output: gr.update(label=lang_map.get('final_video_label')),
         log_accordion: gr.update(label=lang_map.get('log_accordion_label')),
         log_display: gr.update(label=lang_map.get('log_display_label')),
@@ -311,21 +316,18 @@ def update_ui_language(lang_emoji):
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     default_lang = i18n.get('pt', {})
     
-    # State components
     original_latents_paths_state = gr.State(value=None)
     original_video_path_state = gr.State(value=None)
     upscaled_video_path_state = gr.State(value=None)
     hd_video_path_state = gr.State(value=None)
     current_source_video_state = gr.State(value=None)
 
-    # UI Header
     title_md = gr.Markdown(f"# {default_lang.get('app_title')}")
     subtitle_md = gr.Markdown(default_lang.get('app_subtitle'))
     with gr.Row():
         lang_selector = gr.Radio(["🇧🇷", "🇺🇸", "🇨🇳"], value="🇧🇷", label=default_lang.get('lang_selector_label'))
         resolution_selector = gr.Radio(["480x480", "720x720", "960x960"], value="480x480", label="Base Resolution")
 
-    # Step 1 & 2: Pre-Production
     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.")
         ref_image_input = gr.File(label=default_lang.get('ref_images_label'), file_count="multiple", file_types=["image"])
@@ -339,7 +341,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         storyboard_output = gr.JSON(label=default_lang.get('storyboard_output_label'))
         keyframe_gallery = gr.Gallery(label=default_lang.get('keyframes_gallery_label'), visible=True, object_fit="contain", height="auto", type="filepath")
 
-    # Step 3: Production
     with gr.Accordion(default_lang.get('step3_accordion'), open=False, visible=False) as step3_accordion:
         step3_description_md = gr.Markdown(default_lang.get('step3_description'))
         with gr.Accordion(default_lang.get('ltx_advanced_options'), open=False) as ltx_advanced_options_accordion:
@@ -356,7 +357,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         produce_original_button = gr.Button(default_lang.get('produce_original_button'), variant="primary")
         original_video_output = gr.Video(label="Original Master Video", visible=False, interactive=False)
 
-    # Step 4: Post-Production
     with gr.Accordion(default_lang.get('step4_accordion'), open=False, visible=False) as step4_accordion:
         step4_description_md = gr.Markdown(default_lang.get('step4_description'))
         with gr.Accordion(default_lang.get('sub_step_a_upscaler'), open=True) as sub_step_a_accordion:
@@ -379,7 +379,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             run_audio_button = gr.Button(default_lang.get('run_audio_button'), variant="secondary")
             audio_video_output = gr.Video(label="Video with Audio", visible=False, interactive=False)
 
-    # Final Output & Logs
     final_video_output = gr.Video(label=default_lang.get('final_video_label'), visible=False, interactive=False)
     with gr.Accordion(default_lang.get('log_accordion_label'), open=False) as log_accordion:
         log_display = gr.Textbox(label=default_lang.get('log_display_label'), lines=20, interactive=False, autoscroll=True)