depot/third_party/nixpkgs/pkgs/development/python-modules/speechrecognition/default.nix

80 lines
1.7 KiB
Nix

{
lib,
buildPythonPackage,
fetchFromGitHub,
flac,
openai,
openai-whisper,
pocketsphinx,
pyaudio,
pytestCheckHook,
pythonOlder,
requests,
setuptools,
soundfile,
typing-extensions,
}:
buildPythonPackage rec {
pname = "speechrecognition";
version = "3.11.0";
pyproject = true;
disabled = pythonOlder "3.9";
src = fetchFromGitHub {
owner = "Uberi";
repo = "speech_recognition";
rev = "refs/tags/${version}";
hash = "sha256-5DZ5QhaYpVtd+AX5OSYD3cM+37Ez0+EL5a+zJ+X/uNg=";
};
postPatch = ''
# Remove Bundled binaries
rm speech_recognition/flac-*
rm -r third-party
substituteInPlace speech_recognition/audio.py \
--replace-fail 'shutil_which("flac")' '"${lib.getExe flac}"'
'';
build-system = [ setuptools ];
dependencies = [
pyaudio
requests
typing-extensions
];
optional-dependencies = {
audio = [ pyaudio ];
whisper-api = [ openai ];
whisper-local = [
openai-whisper
soundfile
];
};
nativeCheckInputs = [
pytestCheckHook
pocketsphinx
] ++ lib.flatten (builtins.attrValues optional-dependencies);
pythonImportsCheck = [ "speech_recognition" ];
disabledTests = [
# Parsed string does not match expected
"test_sphinx_keywords"
];
meta = with lib; {
description = "Speech recognition module for Python, supporting several engines and APIs, online and offline";
homepage = "https://github.com/Uberi/speech_recognition";
changelog = "https://github.com/Uberi/speech_recognition/releases/tag/${version}";
license = with licenses; [
gpl2Only
bsd3
];
maintainers = with maintainers; [ fab ];
};
}