From 6703d2774b557e636956b97d53c23ff0b5c9b7b4 Mon Sep 17 00:00:00 2001
From: hidenori-endo <nintendwu@gmail.com>
Date: Fri, 10 Nov 2023 03:26:47 +0900
Subject: [PATCH] Drop ffmpeg-python dependency

---
 requirements.txt  |  1 -
 whisperx/audio.py | 31 ++++++++++++++++++++++---------
 2 files changed, 22 insertions(+), 10 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 0b8a0b1..bc0455a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,6 @@ torch>=2
 torchaudio>=2
 faster-whisper>=0.8
 transformers
-ffmpeg-python>=0.2
 pandas
 setuptools>=65
 nltk
diff --git a/whisperx/audio.py b/whisperx/audio.py
index 513ab7c..48fe3c1 100644
--- a/whisperx/audio.py
+++ b/whisperx/audio.py
@@ -1,8 +1,8 @@
 import os
+import subprocess
 from functools import lru_cache
 from typing import Optional, Union
 
-import ffmpeg
 import numpy as np
 import torch
 import torch.nn.functional as F
@@ -40,14 +40,27 @@ def load_audio(file: str, sr: int = SAMPLE_RATE):
     A NumPy array containing the audio waveform, in float32 dtype.
     """
     try:
-        # This launches a subprocess to decode audio while down-mixing and resampling as necessary.
-        # Requires the ffmpeg CLI and `ffmpeg-python` package to be installed.
-        out, _ = (
-            ffmpeg.input(file, threads=0)
-            .output("-", format="s16le", acodec="pcm_s16le", ac=1, ar=sr)
-            .run(cmd=["ffmpeg", "-nostdin"], capture_stdout=True, capture_stderr=True)
-        )
-    except ffmpeg.Error as e:
+        # Launches a subprocess to decode audio while down-mixing and resampling as necessary.
+        # Requires the ffmpeg CLI to be installed.
+        cmd = [
+            "ffmpeg",
+            "-nostdin",
+            "-threads",
+            "0",
+            "-i",
+            file,
+            "-f",
+            "s16le",
+            "-ac",
+            "1",
+            "-acodec",
+            "pcm_s16le",
+            "-ar",
+            str(sr),
+            "-",
+        ]
+        out = subprocess.run(cmd, capture_output=True, check=True).stdout
+    except subprocess.CalledProcessError as e:
         raise RuntimeError(f"Failed to load audio: {e.stderr.decode()}") from e
 
     return np.frombuffer(out, np.int16).flatten().astype(np.float32) / 32768.0