Update app.py

app.py

@@ -3,7 +3,6 @@ import cv2
 import gradio as gr
 import AnimeGANv3_src
 import numpy as np
-from moviepy.editor import VideoFileClip, ImageSequenceClip
 
 class AnimeGANv3:
     def __init__(self):
@@ -14,7 +13,7 @@ class AnimeGANv3:
         """Process a single frame with AnimeGANv3."""
         frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
         output = AnimeGANv3_src.Convert(frame_rgb, style_code, det_face)
-        return output
+        return output[:, :, ::-1]  # Convert back to BGR for OpenCV
 
     def inference(self, video_path, style, if_face=None):
         print(video_path, style, if_face)
@@ -33,30 +32,34 @@ class AnimeGANv3:
             style_code = style_codes.get(style, "U")
             det_face = if_face == "Yes"
 
-            # Open the video
+            # Open the input video
             cap = cv2.VideoCapture(video_path)
             if not cap.isOpened():
                 raise Exception("Could not open video file")
 
             # Get video properties
             fps = cap.get(cv2.CAP_PROP_FPS)
+            width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+            height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
             frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-            print(f"Processing {frame_count} frames at {fps} FPS")
+            print(f"Processing {frame_count} frames at {fps} FPS, {width}x{height}")
 
-            # Extract and process frames
-            frames = []
+            # Set up video writer
+            save_path = "output/out.mp4"
+            fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Use 'mp4v' codec for MP4
+            out = cv2.VideoWriter(save_path, fourcc, fps, (width, height))
+
+            # Process frames
             while True:
                 ret, frame = cap.read()
                 if not ret:
                     break
                 processed_frame = self.process_frame(frame, style_code, det_face)
-                frames.append(processed_frame)
-            cap.release()
+                out.write(processed_frame)
 
-            # Convert frames to video
-            save_path = "output/out.mp4"
-            clip = ImageSequenceClip(frames, fps=fps)
-            clip.write_videofile(save_path, codec="libx264", audio=False)
+            # Release resources
+            cap.release()
+            out.release()
 
             return save_path
         except Exception as error: