fix

app.py

@@ -37,7 +37,7 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Constants
-MAX_FRAMES = 80
+MAX_FRAMES = 81
 OUTPUT_FPS = 24
 RENDER_WIDTH = 512
 RENDER_HEIGHT = 384
@@ -287,6 +287,17 @@ def run_wan_ttm_generation(prompt, tweak_index, tstrong_index, first_frame_path,
 
     progress(0, desc="Loading Wan TTM Pipeline...")
 
+    import decord
+    vr = decord.VideoReader(motion_video_path)
+    actual_frame_count = len(vr)
+
+    target_num_frames = ((actual_frame_count - 1) // 4) * 4 + 1
+
+    if target_num_frames < 5:
+        return None, f"❌ Video too short. Only {actual_frame_count} frames tracked."
+
+    logger.info(f"Setting Wan num_frames to {target_num_frames} based on tracking output.")
+
     progress(0.2, desc="Preparing inputs...")
     image = load_image(first_frame_path)
 
@@ -307,7 +318,7 @@ def run_wan_ttm_generation(prompt, tweak_index, tstrong_index, first_frame_path,
         image.height, image.width, max_area, mod_value)
     image = image.resize((width, height))
 
-    progress(0.4, desc="Generating Video (this may take a few minutes)...")
+    progress(0.4, desc=f"Generating {target_num_frames} frames (this may take a few minutes)...")
     generator = torch.Generator(device="cuda").manual_seed(0)
 
     with torch.inference_mode():
@@ -317,7 +328,7 @@ def run_wan_ttm_generation(prompt, tweak_index, tstrong_index, first_frame_path,
             negative_prompt=negative_prompt,
             height=height,
             width=width,
-            num_frames=81,  # Wan default
+            num_frames=target_num_frames,
             guidance_scale=3.5,
             num_inference_steps=50,
             generator=generator,
@@ -331,7 +342,7 @@ def run_wan_ttm_generation(prompt, tweak_index, tstrong_index, first_frame_path,
         first_frame_path), "wan_ttm_output.mp4")
     export_to_video(result.frames[0], output_path, fps=16)
 
-    return output_path, "✅ TTM Video generated successfully with Wan 2.2!"
+    return output_path, f"✅ TTM Video ({target_num_frames} frames) generated successfully!"
 
 # --- MODIFIED PROCESS VIDEO TO RETURN FILE PATHS ---