duration added

app.py

@@ -871,12 +871,6 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
                 """)
                 output_video = gr.Video(label="Edited Video", height=512)
 
-                duration_s = get_display_time_required(2, default_replace_string)
-                duration_m = duration_s / 60
-                
-                time_required = f"⌚ Zero GPU Required: ~{duration_s}.0s ({duration_m:.1f} mins)"
-
-                time_required = gr.Text(value=time_required, show_label=False, visible=False)
                 action_button = gr.Button("Wan Animate 🦆", variant='primary', elem_classes="button-gradient")
 
                 with gr.Accordion("Generated Masks 🎭", open=False, visible=True):
@@ -952,6 +946,7 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
                 gr.Markdown("Choose a frame and adjust the relevant mask, by including and excluding points, to propagate across frames. If no mask is selected, one will be chosen automatically.")
                 gr.Markdown("")
             with gr.Row():
+                max_duration_slider_2 = gr.Slider(2, 10, 2, step=2, label="Max Duration", visible=True)
                 frame_index = gr.Slider(label="Frame Index", minimum=0, maximum=0, value=0, step=1,)
             with gr.Row():
                 first_frame = gr.Image(type="filepath", height=384, interactive=False)
@@ -963,6 +958,13 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
                     )
 
                 with gr.Column():
+                    duration_s = get_display_time_required(2, default_replace_string)
+                    duration_m = duration_s / 60
+                    
+                    time_required = f"⌚ Zero GPU Required: ~{duration_s}.0s ({duration_m:.1f} mins)"
+    
+                    time_required = gr.Text(value=time_required, show_label=False, visible=False)
+                    
                     confirm_sam2_points = gr.Button("Wan Animate 🦆", variant='primary', elem_classes="button-gradient")
                     cancel_button = gr.Button("Cancel")
 
@@ -970,8 +972,13 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
     replace_character_string.change(update_time_required, inputs=[max_duration_slider, replace_character_string], outputs=[time_required]) \
         .then(log_change, inputs=[gr.State("mode"), session_state, replace_character_string])
 
+    def sync_slider(val):
+        return val
+    
+    max_duration_slider_2.change(sync_slider, max_duration_slider_2, max_duration_slider)
     max_duration_slider.change(log_change, inputs=[gr.State("slider"), session_state, max_duration_slider]) \
-        .then(update_time_required, inputs=[max_duration_slider, replace_character_string], outputs=[time_required])
+        .then(update_time_required, inputs=[max_duration_slider, replace_character_string], outputs=[time_required]) \
+        .then(extract_frames, inputs=[input_video, max_duration_slider, session_state], outputs=[first_frame, frames, frame_index])
     
     input_video.change(log_change, inputs=[gr.State("video"), session_state]) \
         .then(set_reset, inputs=[prompt_state], outputs=[prompt_state, sam2_mode_select])