install-geoai

name	install-geoai
description	Verify that the geoai Python package is installed and functional. If not, provide installation instructions. Optionally check extra dependencies for deep learning models.
argument-hint	[--check] [--extras]
allowed-tools	Bash

Arguments: $@

Step 1 -- Check if geoai is importable

python3 -c "import geoai; print(f'geoai v{geoai.__version__}')"

• Success -> report the version and continue to Step 2 if --check or --extras was passed, otherwise stop.
• ImportError -> go to Step 3.

Step 2 -- Check optional dependencies

If --check is present in $@, run a comprehensive dependency check:

python3 -c "
deps = [
    'geoai', 'geopandas', 'rasterio', 'rioxarray', 'shapely',
    'leafmap', 'numpy', 'pandas', 'matplotlib',
]
for dep in deps:
    try:
        mod = __import__(dep)
        ver = getattr(mod, '__version__', 'unknown')
        print(f'{dep}: {ver}')
    except ImportError:
        print(f'{dep}: NOT INSTALLED')
"

If --extras is present in $@, also check deep learning dependencies:

python3 -c "
import sys
dl_deps = ['torch', 'torchvision', 'transformers', 'timm', 'segmentation_models_pytorch']
for dep in dl_deps:
    try:
        mod = __import__(dep)
        ver = getattr(mod, '__version__', 'unknown')
        print(f'{dep}: {ver}')
    except ImportError:
        print(f'{dep}: NOT INSTALLED')

try:
    import torch
    if torch.cuda.is_available():
        print(f'CUDA: {torch.version.cuda} (device: {torch.cuda.get_device_name(0)})')
    else:
        print('CUDA: not available (CPU only)')
except ImportError:
    print('CUDA: torch not installed')
"

Report the results and note any missing packages.

Step 3 -- Installation instructions

If geoai is not installed, tell the user:

geoai is not installed. Install it with:

pip install geoai-py

For GPU-accelerated AI models (object detection, segmentation), also install PyTorch:

pip install torch torchvision

For the full set of optional dependencies:

pip install "geoai-py[extra]"

Stop after showing the instructions. Do not attempt to install automatically unless the user explicitly asks.