Update app_lora.py

app_lora.py

@@ -105,9 +105,9 @@ def get_duration(input_image, prompt, height, width,
 
 @spaces.GPU(duration=get_duration)
 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, 
+                   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:
@@ -132,7 +132,32 @@ def generate_video(input_image, prompt, height, width,
 
     with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmpfile:
         video_path = tmpfile.name
-    export_to_video(output_frames_list, video_path, fps=FIXED_FPS)
+        
+        # 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
+            ]
+        )
+        
+        # Verify video can be read (basic sanity check)
+        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")
+        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)
+
     return video_path, current_seed
 
 with gr.Blocks() as demo: