Update app.py

app.py

@@ -39,10 +39,14 @@ FIXED_FPS = 24
 MIN_FRAMES_MODEL = 8
 MAX_FRAMES_MODEL = 81 
 
-def _calculate_new_dimensions_wan(pil_image: Image.Image, mod_val: int, calculation_max_area: float,
-                                 min_slider_h: int, max_slider_h: int,
-                                 min_slider_w: int, max_slider_w: int,
-                                 default_h: int, default_w: int) -> tuple[int, int]:
+default_prompt_i2v = "make this image come alive, cinematic motion, smooth animation"
+default_negative_prompt = "Bright tones, overexposed, static, blurred details, subtitles, style, works, paintings, images, static, overall gray, worst quality, low quality, JPEG compression residue, ugly, incomplete, extra fingers, poorly drawn hands, poorly drawn faces, deformed, disfigured, misshapen limbs, fused fingers, still picture, messy background, three legs, many people in the background, walking backwards, watermark, text, signature"
+
+
+def _calculate_new_dimensions_wan(pil_image, mod_val, calculation_max_area,
+                                 min_slider_h, max_slider_h,
+                                 min_slider_w, max_slider_w,
+                                 default_h, default_w):
     orig_w, orig_h = pil_image.size
     if orig_w <= 0 or orig_h <= 0:
         return default_h, default_w
@@ -60,7 +64,7 @@ def _calculate_new_dimensions_wan(pil_image: Image.Image, mod_val: int, calculat
     
     return new_h, new_w
 
-def handle_image_upload_for_dims_wan(uploaded_pil_image: Image.Image | None, current_h_val: int, current_w_val: int):
+def handle_image_upload_for_dims_wan(uploaded_pil_image, current_h_val, current_w_val):
     if uploaded_pil_image is None:
         return gr.update(value=DEFAULT_H_SLIDER_VALUE), gr.update(value=DEFAULT_W_SLIDER_VALUE)
     try:
@@ -75,10 +79,10 @@ def handle_image_upload_for_dims_wan(uploaded_pil_image: Image.Image | None, cur
         return gr.update(value=DEFAULT_H_SLIDER_VALUE), gr.update(value=DEFAULT_W_SLIDER_VALUE)
 
 @spaces.GPU
-def generate_video(input_image: Image.Image, prompt: str, negative_prompt: str,
-                   height: int, width: int, duration_seconds: float,
-                   guidance_scale: float, steps: int,
-                   seed: int, randomize_seed: bool, 
+def generate_video(input_image, prompt, height, width, 
+                   negative_prompt=default_negative_prompt, duration_seconds = 2,
+                   guidance_scale = 1, steps = 4,
+                   seed = 42, randomize_seed = False, 
                    progress=gr.Progress(track_tqdm=True)):
     
     if input_image is None:
@@ -106,9 +110,6 @@ def generate_video(input_image: Image.Image, prompt: str, negative_prompt: str,
     export_to_video(output_frames_list, video_path, fps=FIXED_FPS)
     return video_path
 
-default_prompt_i2v = "make this image come alive, cinematic motion, smooth animation"
-default_negative_prompt = "Bright tones, overexposed, static, blurred details, subtitles, style, works, paintings, images, static, overall gray, worst quality, low quality, JPEG compression residue, ugly, incomplete, extra fingers, poorly drawn hands, poorly drawn faces, deformed, disfigured, misshapen limbs, fused fingers, still picture, messy background, three legs, many people in the background, walking backwards, watermark, text, signature"
-
 with gr.Blocks() as demo:
     gr.Markdown("# Fast 4 steps Wan 2.1 I2V (14B) with CausVid LoRA")
     with gr.Row():
@@ -144,16 +145,16 @@ with gr.Blocks() as demo:
     )
     
     ui_inputs = [
-        input_image_component, prompt_input, negative_prompt_input,
-        height_input, width_input, duration_seconds_input,
+        input_image_component, prompt_input, height_input, width_input,
+        negative_prompt_input, duration_seconds_input,
         guidance_scale_input, steps_slider, seed_input, randomize_seed_checkbox
     ]
     generate_button.click(fn=generate_video, inputs=ui_inputs, outputs=video_output)
 
     gr.Examples(
         examples=[ 
-            ["peng.png", "a penguin playfully dancing in the snow, Antarctica", default_negative_prompt, 896, 512, 2.0, 1.0, 4, 42, False],
-            ["forg.jpg", "the frog jumps around", default_negative_prompt, 448, 832, 2.0, 1.0, 4, 123, False],
+            ["peng.png", "a penguin playfully dancing in the snow, Antarctica", 896, 512],
+            ["forg.jpg", "the frog jumps around", 448, 832],
         ],
         inputs=ui_inputs, outputs=video_output, fn=generate_video, cache_examples="lazy"
     )