depot/third_party/nixpkgs/pkgs/development/python-modules/videocr/default.nix
Default email 504525a148 Project import generated by Copybara.
GitOrigin-RevId: bd645e8668ec6612439a9ee7e71f7eac4099d4f6
2024-01-02 12:29:13 +01:00

48 lines
1 KiB
Nix

{ lib
, buildPythonPackage
, fetchPypi
, levenshtein
, pytesseract
, opencv4
, fuzzywuzzy
}:
buildPythonPackage rec {
pname = "videocr";
version = "0.1.6";
format = "setuptools";
src = fetchPypi {
inherit pname version;
sha256 = "1clifwczvhvbaw2spgxkkyqsbqh21vyfw3rh094pxfmq89ylyj63";
};
propagatedBuildInputs = [
levenshtein
pytesseract
opencv4
fuzzywuzzy
];
postPatch = ''
substituteInPlace setup.py \
--replace "python-Levenshtein" "Levenshtein" \
--replace "opencv-python" "opencv"
substituteInPlace videocr/constants.py \
--replace "master" "main"
substituteInPlace videocr/video.py \
--replace '--tessdata-dir "{}"' '--tessdata-dir="{}"'
'';
# Project has no tests
doCheck = false;
pythonImportsCheck = [ "videocr" ];
meta = with lib; {
description = "Extract hardcoded subtitles from videos using machine learning";
homepage = "https://github.com/apm1467/videocr";
license = licenses.mit;
maintainers = with maintainers; [ ozkutuk ];
};
}