Update app_lora.py

app_lora.py

@@ -11,6 +11,10 @@ import numpy as np
 from PIL import Image
 import random
 
+import warnings
+warnings.filterwarnings("ignore", message=".*Attempting to use legacy OpenCV backend.*")
+warnings.filterwarnings("ignore", message=".*num_frames - 1.*")
+
 MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
 
 LORA_REPO_ID = "vrgamedevgirl84/Wan14BT2VFusioniX"
@@ -116,7 +120,11 @@ def generate_video(input_image, prompt, height, width,
     target_h = max(MOD_VALUE, (int(height) // MOD_VALUE) * MOD_VALUE)
     target_w = max(MOD_VALUE, (int(width) // MOD_VALUE) * MOD_VALUE)
     
-    num_frames = np.clip(int(round(duration_seconds * FIXED_FPS)), MIN_FRAMES_MODEL, MAX_FRAMES_MODEL)
+    # Ensure num_frames-1 is divisible by 4
+    raw_frames = int(round(duration_seconds * FIXED_FPS))
+    num_frames = ((raw_frames - 1) // 4) * 4 + 1
+    num_frames = np.clip(num_frames, MIN_FRAMES_MODEL, MAX_FRAMES_MODEL)
+    print(f"Using {num_frames} frames (requested {raw_frames})")
     
     current_seed = random.randint(0, MAX_SEED) if randomize_seed else int(seed)
 
@@ -133,30 +141,28 @@ def generate_video(input_image, prompt, height, width,
     with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmpfile:
         video_path = tmpfile.name
         
-        # Use imageio as the preferred backend with optimized settings
-        export_to_video(
-            output_frames_list,
-            video_path,
-            fps=FIXED_FPS,
-            codec="libx264",
-            output_params=[
-                "-pix_fmt", "yuv420p",  # Most compatible pixel format
-                "-movflags", "+faststart",  # Enable streaming
-                "-profile:v", "main",  # Broad compatibility profile
-                "-tune", "film"  # Optimize for high quality video
-            ]
-        )
+        # Simple export that works with current diffusers version
+        export_to_video(output_frames_list, video_path, fps=FIXED_FPS)
         
-        # Verify video can be read (basic sanity check)
+        # Optional: Use FFmpeg directly for better encoding if available
         try:
-            import imageio
-            with imageio.get_reader(video_path) as reader:
-                if len(reader) < 1:  # Check if video has frames
-                    raise ValueError("Empty video generated")
+            import subprocess
+            optimized_path = video_path + "_opt.mp4"
+            subprocess.run([
+                'ffmpeg',
+                '-y',  # Overwrite without asking
+                '-i', video_path,  # Input file
+                '-c:v', 'libx264',  # Codec
+                '-pix_fmt', 'yuv420p',  # Pixel format
+                '-profile:v', 'main',  # Compatibility profile
+                '-movflags', '+faststart',  # Streaming optimized
+                '-crf', '23',  # Quality level
+                '-preset', 'fast',  # Encoding speed
+                optimized_path
+            ], check=True)
+            video_path = optimized_path
         except Exception as e:
-            print(f"Video validation failed, regenerating: {str(e)}")
-            # Fallback to simple export if validation fails
-            export_to_video(output_frames_list, video_path, fps=FIXED_FPS)
+            print(f"FFmpeg optimization failed, using default export: {str(e)}")
 
     return video_path, current_seed