added

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
+    ];
+  };
+}