Update app.py

app.py

@@ -1,26 +1,28 @@
 import gradio as gr
 import subprocess
 import cv2
+import ffmpeg
 
-def convert_video(input_file, output_file, codec='mp4v'):
+def convert_video(input_file, output_file, codec='libx264'):
     try:
         # Define input and output files
         input_path = input_file
         output_path = output_file
 
-        # Open input video 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, 20.0, (width, height))
+        out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
 
-        # Read and write video frames
-        while(cap.isOpened()):
+        # Process and write video frames
+        while cap.isOpened():
             ret, frame = cap.read()
             if ret:
                 out.write(frame)
@@ -31,12 +33,16 @@ def convert_video(input_file, output_file, codec='mp4v'):
         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')
+        ffmpeg.run(stream)
+
         print(f"Video converted successfully: {output_path}")
 
     except Exception as e:
         print(f"Error converting video: {e}")
 
-
 def execute_command(command: str) -> None:
     subprocess.run(command, check=True)