Update app.py

app.py

@@ -1,53 +1,17 @@
 import gradio as gr
 import subprocess
-import cv2
-import ffmpeg
+from moviepy.editor import VideoFileClip
 
-def convert_video(input_file, output_file, codec='mp4v'):
-    try:
-        # Define input and output files
-        input_path = input_file
-        output_path = output_file
-
-        # Read 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))
-        fps = cap.get(cv2.CAP_PROP_FPS)
-
-        # Create output video writer
-        out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
-
-        # Process 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()
-
-        # Process and write audio stream using ffmpeg
-        stream = ffmpeg.input(input_path)
-        stream = ffmpeg.output(stream, output_path, acodec='aac', vcodec=codec, strict='experimental', loglevel='error')
-        # Run FFmpeg command and capture stderr output
-        try:
-            ffmpeg.run(stream, capture_stderr=True)
-        except subprocess.CalledProcessError as e:
-            # Print stderr output if an error occurs
-            print(f"FFmpeg error: {e.stderr.decode()}")
-            raise e
+def convert_to_mp4_with_aac(input_path, output_path):
+    # Load the video
+    video = VideoFileClip(input_path)
+    
+    # Set the output format to mp4 with AAC codec
+    video.write_videofile(output_path, codec="libx264", audio_codec="aac")
 
-        print(f"Video converted successfully: {output_path}")
+    return output_path
 
-    except Exception as e:
-        print(f"Error converting video: {e}")
+    
 
 def execute_command(command: str) -> None:
     subprocess.run(command, check=True)
@@ -71,11 +35,13 @@ def infer():
     execute_command(command)
 
     # Convert video to compatible codecs
-    input_file_path = f"output_video/{output_name}.mp4"
-    output_file_path = f"{output_name}.mp4"
-    #convert_video(input_file_path, output_file_path)
+    input_file = f"output_video/{output_name}.mp4"
+    output_file = f"{output_name}.mp4"
+    
+    result = convert_to_mp4_with_aac(input_file, output_file)
+    
 
-    return f"output_video/{output_name}.mp4"
+    return result
 
 with gr.Blocks() as demo:
     with gr.Column():