Update app.py

app.py

@@ -47,6 +47,8 @@ def calculate_resize_dimensions(width, height, max_width=1024):
     aspect_ratio = height / width
     new_width = max_width
     new_height = int(max_width * aspect_ratio)
+    # Make height even number for video encoding
+    new_height = new_height - (new_height % 2)
     return new_width, new_height
 
 def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True)):
@@ -98,28 +100,34 @@ def infer(image_path, prompt, orbit_type, progress=gr.Progress(track_tqdm=True))
    # Generate initial output video
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     temp_path = f"output_{timestamp}_temp.mp4"
-    output_path = f"output_{timestamp}.mp4"
+    final_path = f"output_{timestamp}.mp4"
     
-    # Export initial video
+    # First export the original video
     export_to_video(video.frames[0], temp_path, fps=8)
     
-    # Resize using moviepy with h264 codec
-    video_clip = VideoFileClip(temp_path)
-    resized_clip = video_clip.resize(width=target_width, height=target_height)
-    resized_clip.write_videofile(
-        output_path,
-        codec='libx264',
-        fps=8,
-        preset='medium',
-        ffmpeg_params=['-crf', '23']
-    )
-    
-    # Cleanup
-    video_clip.close()
-    resized_clip.close()
-    os.remove(temp_path)
+    # Then resize it with moviepy
+    try:
+        video_clip = VideoFileClip(temp_path)
+        resized_clip = video_clip.resize(width=target_width, height=target_height)
+        resized_clip.write_videofile(
+            final_path,
+            codec='libx264',
+            fps=8,
+            preset='medium',
+            ffmpeg_params=['-crf', '23'],
+            verbose=False,
+            logger=None
+        )
+    finally:
+        # Make sure we clean up the clips
+        if 'video_clip' in locals():
+            video_clip.close()
+        if 'resized_clip' in locals():
+            resized_clip.close()
+        if os.path.exists(temp_path):
+            os.remove(temp_path)
     
-    return output_path
+    return final_path
 
 # Set up Gradio UI
 with gr.Blocks(analytics_enabled=False) as demo: