From 61f75643a44c6ab329f34d86871aeb114929f8c1 Mon Sep 17 00:00:00 2001
From: alsaiduq-lab <riiidge.racer@gmail.com>
Date: Sat, 22 Mar 2025 18:22:10 -0600
Subject: [PATCH] added

---
 modules/python.nix | 49 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)
 create mode 100644 modules/python.nix

diff --git a/modules/python.nix b/modules/python.nix
new file mode 100644
index 0000000..240b5ff
--- /dev/null
+++ b/modules/python.nix
@@ -0,0 +1,49 @@
+{
+  config,
+  pkgs,
+  lib,
+  ...
+}:
+let
+  customPkgs = import ../pkgs { inherit pkgs lib; };
+in {
+  options.custom.python = {
+    enable = lib.mkEnableOption "System Python Env";
+  };
+
+  config = lib.mkIf config.custom.python.enable {
+    environment.systemPackages = with pkgs; [
+      (python311.withPackages (ps: with ps; [
+        virtualenv
+        (torch.override { cudaSupport = true; })
+        torchvision
+        torchaudio
+        ipython
+        i3ipc
+        xlib
+        six
+        psutil
+        pynvml
+        pyqtgraph
+        pyqt6
+        numpy
+        pandas
+        matplotlib
+        scipy
+        requests
+        click
+        typer
+        rich
+        pyyaml
+        pytz
+        onnxruntime
+        pillow
+        timm
+      ] ++ [ customPkgs.python-rembg ]))
+      python3Packages.pip
+      uv
+      ruff
+      black
+    ];
+  };
+}