Add doctr to Nix dependencies

Author: matthid

nix/dependencies.nix

@@ -25,6 +25,10 @@
 let
   python_packages = my_python.pkgs;
   unstable_python_packages = unstable_pkgs.my_python.pkgs;
+  python_doctr = pkgs.callPackage ./packages/doctr.nix {
+    pkgs = pkgs;
+    my_python = my_python;
+  };
 in {
   pkgs = pkgs;
   lib = lib;
@@ -34,6 +38,7 @@ in {
     python_packages.numpy
     python_packages.torch-bin
     unstable_python_packages.pytorch-lightning
+    python_doctr
     cudatoolkit
   ];
 }

nix/packages/doctr.nix

@@ -0,0 +1,38 @@
+{ pkgs ? import <nixpkgs> {}
+, lib ? pkgs.lib
+, my_python ? pkgs.python311
+, buildPythonPackage ? my_python.pkgs.buildPythonPackage
+, fetchFromGitHub ? pkgs.fetchFromGitHub }:
+
+buildPythonPackage rec {
+  pname = "doctr";
+  version = "0.8.1";  # Update this to the latest release version
+
+  src = fetchFromGitHub {
+    owner = "mindee";
+    repo = pname;
+    rev = "v${version}";
+    sha256 = "rlIGq5iHDAEWy1I0sAXVSN2/Jh2ub/xLCLCLLp7+9ik=";  # You can generate this with nix-prefetch-github
+  };
+
+  nativeBuildInputs = [ my_python.pkgs.setuptools ];
+  propagatedBuildInputs = [ my_python.pkgs.torch ];
+
+
+  postInstall = let
+    libPath = "lib/${my_python.libPrefix}/site-packages";
+  in
+    ''
+      mkdir -p $out/nix-support
+      echo "$out/${libPath}" > "$out/nix-support/propagated-build-inputs"
+    '';
+
+  doCheck = false;
+
+  meta = with lib; {
+    description = "Doctr Python package";
+    homepage = https://github.com/mindee/doctr;
+    license = licenses.asl20;  # Update as per project's licensing
+    maintainers = [ ];  # You need to add your name to the list of Nix maintainers
+  };
+}