Update app.py

app.py

@@ -1,29 +1,35 @@
 import gradio as gr
 import subprocess
-import imageio
-import imageio_ffmpeg
+import cv2
 
-def convert_video(input_file, output_file, format='mp4', codec='libx264'):
+def convert_video(input_file, output_file, codec='mp4v'):
     try:
         # Define input and output files
         input_path = input_file
-        output_path = output_file + '.' + format
-
-        # Read video file
-        reader = imageio.get_reader(input_path)
-
-        # Configure FFmpeg parameters
-        ffmpeg_params = ['-vcodec', codec]
-
-        # Get writer for output video file with specified codec
-        writer = imageio.get_writer(output_path, format=format, ffmpeg_params=ffmpeg_params, plugin='imageio_ffmpeg')
-
-        # Iterate through video frames and write to output file
-        for frame in reader:
-            writer.append_data(frame)
-
-        # Close writer
-        writer.close()
+        output_path = output_file
+
+        # Open input video file
+        cap = cv2.VideoCapture(input_path)
+
+        # Get video codec and frame dimensions
+        fourcc = cv2.VideoWriter_fourcc(*codec)
+        width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+        height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+
+        # Create output video writer
+        out = cv2.VideoWriter(output_path, fourcc, 20.0, (width, height))
+
+        # Read and write video frames
+        while(cap.isOpened()):
+            ret, frame = cap.read()
+            if ret:
+                out.write(frame)
+            else:
+                break
+
+        # Release video objects
+        cap.release()
+        out.release()
 
         print(f"Video converted successfully: {output_path}")