more duration fixes

app.py

@@ -202,43 +202,33 @@ def is_portrait(video_file):
 
 def calculate_time_required(max_duration_s, rc_bool):
 
-    frames_count = 30 * max_duration_s
-
-    chunks = frames_count // 77 + 1
-
-
-    if rc_bool:
-        pose2d_tracking_duration_s = 75
-        iteration_per_step_s = 12
-    else:
-        pose2d_tracking_duration_s = 65
-        iteration_per_step_s = 12
-        
-    time_required = pose2d_tracking_duration_s + iteration_per_step_s * 5 * chunks
-    print(f'for frames_count:{frames_count} doing {chunks} chunks the time_required is {time_required}')
-    return time_required
+    if max_duration_s == 2:
+        return 120
+    elif max_duration_s == 4:
+        return 180
+    elif max_duration_s == 6:
+        return 260
+    elif max_duration_s == 8:
+        return 330
+    elif max_duration_s == 10:
+        return 340
+    
+def get_display_time_required(max_duration_s, rc_bool):
+    # the 30 seconds extra is just for saftey in case of a unexpected slow down
+    return calculate_time_required(max_duration_s, rc_bool) - 30
 
 def update_time_required(max_duration_s, rc_str):
 
     rc_bool = rc_mapping[rc_str]
 
-    duration_s = calculate_time_required(max_duration_s, rc_bool)
+    duration_s = get_display_time_required(max_duration_s, rc_bool)
     duration_m = duration_s / 60
     
     return gr.update(value=f"⌚ Zero GPU Required: ~{duration_s}.0s ({duration_m:.1f} mins)")
 
-def log_slider_change():
-    print("slider value changed")
 
 def get_duration(input_video, max_duration_s, edited_frame, rc_bool, session_id, progress):
-    if max_duration_s == 2:
-        return 120
-    elif max_duration_s == 4:
-        return 180
-    elif max_duration_s == 6:
-        return 240
-    elif max_duration_s == 8:
-        return 300
+    return calculate_time_required(max_duration_s, rc_bool)
 
 @spaces.GPU(duration=get_duration)
 def _animate(input_video, max_duration_s, edited_frame, rc_bool, session_id = None, progress=gr.Progress(track_tqdm=True),):
@@ -269,14 +259,6 @@ def _animate(input_video, max_duration_s, edited_frame, rc_bool, session_id = No
     if rc_bool:
         tag_string = "replace_flag"
 
-    # sh("python ./wan/modules/animate/preprocess/preprocess_data.py "
-    #    "--ckpt_path ./Wan2.2-Animate-14B/process_checkpoint "
-    #    f"--video_path {input_video} "
-    #    f"--refer_path {edited_frame} "
-    #    f"--save_path {preprocess_dir} "
-    #    f"--resolution_area {w} {h} --{tag_string} "
-    #    )
-
     preprocess_model = load_preprocess_models()
 
     run_preprocess(preprocess_model, input_video, edited_frame, preprocess_dir, w, h, tag_string)
@@ -302,12 +284,6 @@ def _animate(input_video, max_duration_s, edited_frame, rc_bool, session_id = No
 
     return output_video_path
 
-def animate_example(input_video, max_duration_s, edited_frame, rc_str, session_id = None, progress=gr.Progress(track_tqdm=True),):
-
-    print("Example used")
-
-    return animate_scene(input_video, max_duration_s, edited_frame, rc_str, session_id, progress)
-
 def animate_scene(input_video, max_duration_s, edited_frame, rc_str, session_id = None, progress=gr.Progress(track_tqdm=True),):
 
     if not input_video:
@@ -449,6 +425,14 @@ css = """
     transform: translateX(100%);
     }
     """
+
+def log_change(log_source, session_id, meta_data = None):
+
+    if not meta_data:
+        print(f'{session_id} changed {log_source}')
+    else:
+        print(f'{session_id} changed {log_source} with {meta_data}')
+
 def start_session(request: gr.Request):
 
     return request.session_hash
@@ -495,7 +479,7 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
                 </div>
                 """)
                 input_video = gr.Video(label="Input Video", height=512)
-                max_duration_slider = gr.Slider(2, 4, 2, step=2, label="Max Duration", visible=False)
+                max_duration_slider = gr.Slider(2, 10, 2, step=2, label="Max Duration", visible=False)
 
                 gr.Examples(
                     examples=[
@@ -521,8 +505,10 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
                 </div>
                 """)
                 edited_frame = gr.Image(label="Ref Image", type="filepath", height=512)
+
+                default_replace_string = "Video ← Ref Image"
                 replace_character_string = gr.Radio(
-                    ["Video → Ref Image", "Video ← Ref Image"], value="Video ← Ref Image", show_label=False
+                    ["Video → Ref Image", "Video ← Ref Image"], value=default_replace_string, show_label=False
                 )
 
 
@@ -558,7 +544,12 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
                 """)
                 output_video = gr.Video(label="Edited Video", height=512)
 
-                time_required = gr.Text(value="⌚ Zero GPU Required: ~195.0s (3.2 mins)", show_label=False, visible=False)
+                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("Preprocessed Data", open=False, visible=True):
@@ -574,7 +565,6 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
                 gr.Examples(
                     examples=[
 
-
                         [
                             "./examples/okay.mp4", 
                             2,
@@ -613,14 +603,17 @@ with gr.Blocks(css=css, title="Wan 2.2 Animate --replace", theme=gr.themes.Ocean
                     ],
                     inputs=[input_video, max_duration_slider, edited_frame, replace_character_string],
                     outputs=[output_video, pose_video, bg_video, mask_video, face_video],
-                    fn=animate_example,
+                    fn=animate_scene,
                     cache_examples=True,
                 )
 
     action_button.click(fn=animate_scene, inputs=[input_video, max_duration_slider, edited_frame, replace_character_string, session_state], outputs=[output_video, pose_video, bg_video, mask_video, face_video])
-    max_duration_slider.change(log_slider_change)
     replace_character_string.change(update_time_required, inputs=[max_duration_slider, replace_character_string], outputs=[time_required])
-                   
+
+    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])
+    input_video.change(log_change, inputs=[gr.State("video"), session_state])
+    edited_frame.change(log_change, inputs=[gr.State("ref image"), session_state])        
+
 if __name__ == "__main__":
     demo.queue()
     demo.unload(cleanup)