{
  pkgs,
  python ? pkgs.python312,
  ...
}:

let
  # Python environment with notebook tooling and core data-science libs
  pythonWithNotebook = python.withPackages (
    ps: with ps; [
      altair
      ipykernel
      jupyterlab
      matplotlib
      numpy
      pandas
      polars
      plotly
      scipy
      scikit-learn
      seaborn
    ]
  );
in
pkgs.mkShell {
  name = "jupyter-notebook";
  packages = [
    pythonWithNotebook
    pkgs.ruff
    pkgs.pyright
    pkgs.git
  ];
  shellHook = ''
    export PYTHONNOUSERSITE=1
    export JUPYTER_CONFIG_DIR="''${XDG_CONFIG_HOME:-$HOME/.config}/jupyter"
    mkdir -p "$JUPYTER_CONFIG_DIR"
    echo "Jupyter notebook shell ready"
    echo "Python: ${python.version}"
    echo "Launch notebook: jupyter lab"
    echo "Launch classic: jupyter notebook"
  '';
}