Update app.py

app.py

@@ -84,6 +84,17 @@ def _calculate_new_dimensions_wan(pil_image, mod_val, calculation_max_area,
     return new_h, new_w
 
 def handle_image_upload_for_dims_wan(uploaded_pil_image, current_h_val, current_w_val):
+    """
+    Handle image upload and calculate appropriate dimensions for video generation.
+    
+    Args:
+        uploaded_pil_image: The uploaded image (PIL Image or numpy array)
+        current_h_val: Current height slider value
+        current_w_val: Current width slider value
+        
+    Returns:
+        Tuple of gr.update objects for height and width sliders
+    """
     if uploaded_pil_image is None:
         return gr.update(value=DEFAULT_H_SLIDER_VALUE), gr.update(value=DEFAULT_W_SLIDER_VALUE)
     try:
@@ -114,14 +125,7 @@ def get_duration(image,
                  seed,
                  progress):
     """Calculate dynamic GPU duration based on parameters."""
-    if duration_seconds >= 3:
-        return 220
-    elif sampling_steps > 35 and duration_seconds >= 2:
-        return 180
-    elif sampling_steps < 35 or duration_seconds < 2:
-        return 105
-    else:
-        return 90
+    return sampling_steps * 15
 
 # --- 2. Gradio Inference Function ---
 @spaces.GPU(duration=get_duration)
@@ -137,7 +141,24 @@ def generate_video(
     seed=42,
     progress=gr.Progress(track_tqdm=True)
 ):
-    """The main function to generate video, called by the Gradio interface."""
+    """
+    Generate a video from text prompt and optional image using the Wan 2.2 TI2V model.
+    
+    Args:
+        image: Optional input image (numpy array) for image-to-video generation
+        prompt: Text prompt describing the desired video
+        height: Target video height in pixels
+        width: Target video width in pixels
+        duration_seconds: Desired video duration in seconds
+        sampling_steps: Number of denoising steps for video generation
+        guide_scale: Guidance scale for classifier-free guidance
+        shift: Sample shift parameter for the model
+        seed: Random seed for reproducibility (-1 for random)
+        progress: Gradio progress tracker
+        
+    Returns:
+        Path to the generated video file
+    """
     if seed == -1:
         seed = random.randint(0, sys.maxsize)
 
@@ -250,4 +271,4 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(), delete_cache=(60, 900)) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(mcp_server=True)