ffprobe_audio

ffmpeg_util.py

@@ -21,6 +21,27 @@ class InputOutputVideoMetadata(BaseModel):
     output: VideoMetadata
 
 
+class AudioMetadata(BaseModel):
+    duration_sec: float = 0
+
+
+def ffprobe_audio(input_path: str) -> AudioMetadata:
+    cmd_args = [
+        "ffprobe",
+        "-v",
+        "error",
+        "-show_entries",
+        "format=duration",
+        "-of",
+        "default=noprint_wrappers=1:nokey=1",
+        input_path,
+    ]
+    print("\t$ " + " ".join(cmd_args))
+    return AudioMetadata(
+        duration_sec=float(subprocess.check_output(cmd_args, encoding="utf-8"))
+    )
+
+
 def ffprobe_video(input_path: str) -> VideoMetadata:
     cmd_args = [
         "ffprobe",

retro/sadtalker.py

@@ -19,6 +19,7 @@
 from celeryconfig import app, setup_queues
 from ffmpeg_util import (
     ffprobe_video,
+    ffprobe_audio,
     ffmpeg_read_input_frames,
     VideoMetadata,
     ffmpeg_get_writer_proc,
@@ -219,21 +220,7 @@ def sadtalker(
             print(subprocess.check_output(args, encoding="utf-8"))
             audio_path = wav_audio_path
         # make sure audio is not 0 seconds
-        audio_length = float(
-            subprocess.check_output(
-                [
-                    "ffprobe",
-                    "-v",
-                    "error",
-                    "-show_entries",
-                    "format=duration",
-                    "-of",
-                    "default=noprint_wrappers=1:nokey=1",
-                    audio_path,
-                ],
-                encoding="utf-8",
-            )
-        )
+        audio_length = ffprobe_audio(audio_path).duration_sec
         if audio_length <= 0.1:
             raise ValueError("Audio is too short")