python3Packages.whisperx: init at 3.3.1

pkgs/development/python-modules/whisperx/default.nix

@@ -0,0 +1,66 @@
+{
+  lib,
+  buildPythonPackage,
+  fetchFromGitHub,
+
+  setuptools,
+
+  ctranslate2,
+  faster-whisper,
+  ffmpeg,
+  nltk,
+  pandas,
+  pyannote-audio,
+  torch,
+  torchaudio,
+  transformers,
+}:
+let
+  pname = "whisperX";
+  version = "3.3.1";
+in
+buildPythonPackage {
+  inherit pname version;
+  pyproject = true;
+
+  src = fetchFromGitHub {
+    owner = "m-bain";
+    repo = pname;
+    tag = "v${version}";
+    hash = "sha256-Wb9jKTs0rTdgjvcnKl1P8Uh6b5jzlkKgQBcJYA+7+W4=";
+  };
+
+  build-system = [ setuptools ];
+
+  dependencies = [
+    ctranslate2
+    faster-whisper
+    nltk
+    pandas
+    pyannote-audio
+    torch
+    torchaudio
+    transformers
+  ];
+
+  pythonRelaxDeps = [
+    "ctranslate2"
+    "faster-whisper"
+  ];
+
+  pythonImportsCheck = [ "whisperx" ];
+
+  makeWrapperArgs = [ "--prefix PATH : ${lib.makeBinPath [ ffmpeg ]}" ];
+
+  # No tests in repository
+  doCheck = false;
+
+  meta = {
+    mainProgram = "whisperx";
+    description = "Automatic Speech Recognition with Word-level Timestamps (& Diarization) ";
+
+    platforms = lib.platforms.all;
+    license = lib.licenses.bsd2;
+    maintainers = with lib.maintainers; [ bengsparks ];
+  };
+}

pkgs/top-level/python-packages.nix

@@ -18324,6 +18324,8 @@ self: super: with self; {
 
   whispers = callPackage ../development/python-modules/whispers { };
 
+  whisperx = callPackage ../development/python-modules/whisperx { };
+
   whitenoise = callPackage ../development/python-modules/whitenoise { };
 
   whodap = callPackage ../development/python-modules/whodap { };