{ pkgs ? import <nixpkgs> {} }:

pkgs.mkShell {
  name = "ga-dev-shell";

  buildInputs = with pkgs; [
    python3
    python3Packages.pip
    python3Packages.numpy
    python3Packages.matplotlib
    python3Packages.pygame

    # Optional: use pip to install clifford in virtualenv
  ];

  shellHook = ''
    echo "Python environment with pygame and numpy ready."

    # Optional: set up venv and install clifford via pip if needed
    if [ ! -d .venv ]; then
      python -m venv .venv
      source .venv/bin/activate
      pip install --upgrade pip
      pip install kingdon
      echo "Installed clifford in virtual environment."
    else
      source .venv/bin/activate
      echo "Activated existing virtual environment."
    fi
  '';
}