nerd-dictation

2025-12-23 11:26:39 +00:00
parent b9e8198607
commit 68faf941b7
2 changed files with 91 additions and 0 deletions
@@ -0,0 +1,90 @@
+{
+  lib,
+  config,
+  pkgs,
+  ...
+}:
+let
+  venvDir = "$HOME/.local/share/nerd-dictation/venv";
+
+  # Create an FHS environment where vosk can be pip-installed
+  nerd-dictation-env = pkgs.buildFHSEnv {
+    name = "nerd-dictation-fhs";
+    targetPkgs = pkgs: with pkgs; [
+      # Python and pip for vosk
+      python3
+      python3Packages.pip
+      python3Packages.virtualenv
+
+      # Audio recording (pipewire)
+      pipewire
+      pulseaudio # for parec
+
+      # Input simulation (wayland-compatible)
+      ydotool
+      wtype
+      xdotool
+
+      # Manylinux compatibility
+      pythonManylinuxPackages.manylinux2014Package
+    ];
+    runScript = "bash";
+  };
+
+  # Fetch nerd-dictation from GitHub
+  nerd-dictation-src = pkgs.fetchFromGitHub {
+    owner = "ideasman42";
+    repo = "nerd-dictation";
+    rev = "main";
+    hash = "sha256-xjaHrlJvk8bNvWp1VE4EAHi2VJlAutBxUgWB++3Qo+s=";
+  };
+
+  modelUrl = "https://alphacephei.com/vosk/models/vosk-model-en-us-0.22-lgraph.zip";
+  modelName = "vosk-model-en-us-0.22-lgraph";
+  modelDir = "$HOME/.config/nerd-dictation";
+
+  # Wrapper script that handles setup and runs nerd-dictation
+  nerd-dictation = pkgs.writeShellScriptBin "nerd-dictation" ''
+    set -e
+    VENV="${venvDir}"
+    MODEL_DIR="${modelDir}"
+
+    # Setup venv if needed
+    if [ ! -d "$VENV" ]; then
+      echo "[nerd-dictation] Setting up Python venv and installing vosk..."
+      ${nerd-dictation-env}/bin/nerd-dictation-fhs -c "
+        set -e
+        mkdir -p \$(dirname ${venvDir})
+        python3 -m venv ${venvDir}
+        source ${venvDir}/bin/activate
+        pip install --upgrade pip -q
+        pip install vosk -q
+      "
+      echo "[nerd-dictation] Venv ready."
+    fi
+
+    # Download model if needed
+    if [ ! -d "$MODEL_DIR/model" ]; then
+      echo "[nerd-dictation] Downloading VOSK model (${modelName})..."
+      mkdir -p "$MODEL_DIR"
+      cd "$MODEL_DIR"
+      ${pkgs.curl}/bin/curl -L --progress-bar -o "${modelName}.zip" "${modelUrl}"
+      echo "[nerd-dictation] Extracting model..."
+      ${pkgs.unzip}/bin/unzip -q "${modelName}.zip"
+      mv "${modelName}" model
+      rm "${modelName}.zip"
+      echo "[nerd-dictation] Model ready."
+    fi
+
+    exec ${nerd-dictation-env}/bin/nerd-dictation-fhs -c "source $VENV/bin/activate && python3 ${nerd-dictation-src}/nerd-dictation $*"
+  '';
+in
+{
+  home.packages = [
+    nerd-dictation
+    pkgs.ydotool
+  ];
+
+  # Note: You may need to enable ydotoold as a systemd service
+  # Either via NixOS configuration or by running: ydotoold &
+}
@@ -21,6 +21,7 @@
    ../common/programs/gpg.nix
    ../common/programs/ethui.nix
    ../common/programs/3d.nix
+    ../common/programs/nerd-dictation.nix
    ../common/features/xdg.nix
    ../common/features/bluetooth.nix
    ./monitors.nix