ffmpeg video creation code changed

Wav2Lip/video_generator.py

@@ -12,6 +12,7 @@ import platform
 import torch
 # import face_alignment
 # import streamlit as st
+import ffmpeg
 
 # checkpoint_path = 'wav2lip_model/wav2lip_gan.tflite'
 checkpoint_path = 'wav2lip_model/wav2lip_gan.pth'
@@ -269,6 +270,16 @@ def create_video(voice_audio, face):
             
     out.release()
 
-    command = 'ffmpeg -y -i {} -i {} -strict -2 -q:v 1 {}'.format(voice_audio, 'temp/result.avi', outfile)
+    # command = 'ffmpeg -y -i {} -i {} -strict -2 -q:v 1 {}'.format(voice_audio, 'temp/result.avi', outfile)
     # subprocess.call(command, shell=platform.system() != 'Windows')
-    subprocess.call(command, shell=False)
+    # subprocess.call(command, shell=False)
+
+    # Replace args.audio and args.outfile with the actual variables or arguments
+    audio_input = voice_audio
+    video_input = 'temp/result.avi'
+    output_file = outfile
+
+    # Construct the ffmpeg command using ffmpeg-python
+    audio_stream = ffmpeg.input(audio_input)
+    video_stream = ffmpeg.input(video_input)
+    ffmpeg.concat(video_stream, audio_stream, v=1, a=1).output(output_file, strict='-2', qscale=1).run(overwrite_output=True)