166 lines
4.2 KiB
Nix
166 lines
4.2 KiB
Nix
{
|
|
description = "Local speech-to-text input method for Fcitx5";
|
|
|
|
inputs = {
|
|
nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
|
|
crane.url = "github:ipetkov/crane";
|
|
flake-utils.url = "github:numtide/flake-utils";
|
|
};
|
|
|
|
outputs =
|
|
{
|
|
self,
|
|
nixpkgs,
|
|
crane,
|
|
flake-utils,
|
|
...
|
|
}:
|
|
let
|
|
# Systems we support
|
|
supportedSystems = [
|
|
"x86_64-linux"
|
|
"aarch64-linux"
|
|
];
|
|
in
|
|
flake-utils.lib.eachSystem supportedSystems (
|
|
system:
|
|
let
|
|
pkgs = nixpkgs.legacyPackages.${system};
|
|
craneLib = crane.mkLib pkgs;
|
|
|
|
# Rust STT streaming CLI
|
|
stt-stream = craneLib.buildPackage {
|
|
pname = "stt-stream";
|
|
version = "0.1.0";
|
|
src = craneLib.cleanCargoSource ./stt-stream;
|
|
|
|
nativeBuildInputs = with pkgs; [
|
|
pkg-config
|
|
cmake # for whisper-rs
|
|
clang
|
|
llvmPackages.libclang
|
|
];
|
|
|
|
buildInputs = with pkgs; [
|
|
alsa-lib
|
|
openssl
|
|
# whisper.cpp dependencies
|
|
openblas
|
|
];
|
|
|
|
# For bindgen to find libclang
|
|
LIBCLANG_PATH = "${pkgs.llvmPackages.libclang.lib}/lib";
|
|
|
|
# Enable CUDA if available (user can override)
|
|
WHISPER_CUBLAS = "OFF";
|
|
};
|
|
|
|
# Fcitx5 C++ shim addon
|
|
fcitx5-stt = pkgs.stdenv.mkDerivation {
|
|
pname = "fcitx5-stt";
|
|
version = "0.1.0";
|
|
src = ./fcitx5-stt;
|
|
|
|
nativeBuildInputs = with pkgs; [
|
|
cmake
|
|
extra-cmake-modules
|
|
pkg-config
|
|
];
|
|
|
|
buildInputs = with pkgs; [
|
|
fcitx5
|
|
];
|
|
|
|
cmakeFlags = [
|
|
"-DSTT_STREAM_PATH=${stt-stream}/bin/stt-stream"
|
|
];
|
|
|
|
# Install to fcitx5 addon paths
|
|
postInstall = ''
|
|
mkdir -p $out/share/fcitx5/addon
|
|
mkdir -p $out/share/fcitx5/inputmethod
|
|
mkdir -p $out/lib/fcitx5
|
|
'';
|
|
};
|
|
in
|
|
{
|
|
packages = {
|
|
inherit stt-stream fcitx5-stt;
|
|
default = fcitx5-stt;
|
|
};
|
|
|
|
# Expose as runnable apps
|
|
apps = {
|
|
stt-stream = {
|
|
type = "app";
|
|
program = "${stt-stream}/bin/stt-stream";
|
|
};
|
|
default = {
|
|
type = "app";
|
|
program = "${stt-stream}/bin/stt-stream";
|
|
};
|
|
};
|
|
|
|
devShells.default = pkgs.mkShell {
|
|
inputsFrom = [ stt-stream ];
|
|
packages = with pkgs; [
|
|
rust-analyzer
|
|
rustfmt
|
|
clippy
|
|
fcitx5
|
|
];
|
|
};
|
|
}
|
|
)
|
|
// {
|
|
# NixOS module for integration
|
|
nixosModules.default =
|
|
{
|
|
config,
|
|
lib,
|
|
pkgs,
|
|
...
|
|
}:
|
|
let
|
|
cfg = config.ringofstorms.sttIme;
|
|
sttPkgs = self.packages.${pkgs.stdenv.hostPlatform.system};
|
|
in
|
|
{
|
|
options.ringofstorms.sttIme = {
|
|
enable = lib.mkEnableOption "Speech-to-text input method for Fcitx5";
|
|
|
|
model = lib.mkOption {
|
|
type = lib.types.str;
|
|
default = "base.en";
|
|
description = "Whisper model to use (tiny, base, small, medium, large)";
|
|
};
|
|
|
|
useGpu = lib.mkOption {
|
|
type = lib.types.bool;
|
|
default = false;
|
|
description = "Whether to use GPU acceleration (CUDA)";
|
|
};
|
|
};
|
|
|
|
config = lib.mkIf cfg.enable {
|
|
# Ensure fcitx5 addon is available
|
|
i18n.inputMethod.fcitx5.addons = [ sttPkgs.fcitx5-stt ];
|
|
|
|
# Add STT to the Fcitx5 input method group
|
|
# This assumes de_plasma sets up Groups/0 with keyboard-us (0) and mozc (1)
|
|
i18n.inputMethod.fcitx5.settings.inputMethod = {
|
|
"Groups/0/Items/2".Name = "stt";
|
|
};
|
|
|
|
# Make stt-stream available system-wide
|
|
environment.systemPackages = [ sttPkgs.stt-stream ];
|
|
|
|
# Set default model via environment
|
|
environment.sessionVariables = {
|
|
STT_STREAM_MODEL = cfg.model;
|
|
STT_STREAM_USE_GPU = if cfg.useGpu then "1" else "0";
|
|
};
|
|
};
|
|
};
|
|
};
|
|
}
|