Update app.py

app.py

@@ -115,9 +115,15 @@ def process_video(radio, video, target_language, has_closeup_face):
     else:
         has_face = check_for_faces(video_path)
 
-    if has_face:
-        cmd = f"python Wav2Lip/inference.py --checkpoint_path 'Wav2Lip/checkpoints/wav2lip_gan.pth' --face {shlex.quote(video_path)} --audio '{run_uuid}_output_synth.wav' --pads {pad_top} {pad_bottom} {pad_left} {pad_right} --resize_factor {rescaleFactor} --nosmooth --outfile '{run_uuid}_output_video.mp4'"
-        subprocess.run(cmd, shell=True)
+    if has_closeup_face:
+        try:
+            cmd = f"python Wav2Lip/inference.py --checkpoint_path 'Wav2Lip/checkpoints/wav2lip_gan.pth' --face {shlex.quote(video_path)} --audio '{run_uuid}_output_synth.wav' --pads {pad_top} {pad_bottom} {pad_left} {pad_right} --resize_factor {rescaleFactor} --nosmooth --outfile '{run_uuid}_output_video.mp4'"
+            subprocess.run(cmd, shell=True, check=True)
+        except subprocess.CalledProcessError as e:
+            if "Face not detected! Ensure the video contains a face in all the frames." in str(e.stderr):
+                # Fallback to FFmpeg merge
+                cmd = f"ffmpeg -i {video_path} -i {run_uuid}_output_synth.wav -c:v copy -c:a aac -strict experimental -map 0:v:0 -map 1:a:0 {run_uuid}_output_video.mp4"
+                subprocess.run(cmd, shell=True)
     else:
         # Merge audio with the original video without running Wav2Lip
         cmd = f"ffmpeg -i {video_path} -i {run_uuid}_output_synth.wav -c:v copy -c:a aac -strict experimental -map 0:v:0 -map 1:a:0 {run_uuid}_output_video.mp4"
@@ -159,7 +165,7 @@ iface = gr.Interface(
         video,
         gr.Dropdown(choices=["English", "Spanish", "French", "German", "Italian", "Portuguese", "Polish", "Turkish", "Russian", "Dutch", "Czech", "Arabic", "Chinese (Simplified)"], label="Target Language for Dubbing", value="Spanish"),
         gr.Checkbox(
-                    label="Video has a close-up face",
+                    label="Video has a close-up face. Use Wav2lip.",
                     value=False,
                     info="Say if video have close-up face. For Wav2lip. Will not work if checked wrongly.")
     ],