Add whisperx

hm/default.nix

@@ -8,7 +8,7 @@
     ./common.nix
     ./desktop
     ./dev
-    ./extra.nix
+    ./extra
     ./gaming
     ./git
     ./gpg

hm/extra/default.nix

@@ -21,6 +21,7 @@
 
       # Communication
       signal-desktop
+      (pkgs.callPackage ./whisperx.nix {}) # Transcribe voice messages
 
       # downloading
       # transmission TODO Collision if both transmissions are active?

hm/extra/whisperx.nix

@@ -0,0 +1,37 @@
+{ pkgs ? import <nixpkgs> { } }:
+pkgs.python3Packages.buildPythonPackage {
+  pname = "whisperx";
+  version = "2024-08-19";
+  # pypi doesn't have the requirements.txt file, and it's required
+  src = pkgs.fetchFromGitHub {
+    owner = "m-bain";
+    repo = "whisperX";
+    rev = "9e3a9e0e38fcec1304e1784381059a0e2c670be5"; # git doesn't have tags
+    hash = "sha256-IVtn9fe/yi4+fbH57s9LoiREnMZ2nhEObp1a4R/7gHg=";
+  };
+  pyproject = true;
+  dependencies = [
+    pkgs.python3Packages.torch
+    pkgs.python3Packages.torchaudio
+    (pkgs.python3Packages.faster-whisper.overrideAttrs (old: {
+      # 1.0.2 is actually breaking APIs (requires hotwords argument)
+      src = pkgs.fetchFromGitHub {
+        owner = "SYSTRAN";
+        repo = "faster-whisper";
+        rev = "v1.0.0";
+        hash = "sha256-0fE8X1d6CgDrrHtRudksN/tIGRtBKMvoNwkSVyFNda4=";
+      };
+    }))
+    pkgs.python3Packages.transformers
+    pkgs.python3Packages.pyannote-audio # Not in the requirements.txt for some reason
+    pkgs.python3Packages.pandas
+    pkgs.python3Packages.nltk
+  ];
+  build-system = [
+    pkgs.python3Packages.setuptools
+  ];
+  pythonImportsCheck = [
+    "whisperx"
+  ];
+}
+