FROM python:3.12-slim

WORKDIR /app

# Install git for pip installs from GitHub
RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*

# Copy project
COPY . .

# Install sql-env without deps
RUN pip install --no-deps .

# Install training deps (same as Colab setup cell)
RUN pip install \
    "trl>=0.29.0" \
    "accelerate>=0.34.0" \
    "openenv-core[core]>=0.2.1" \
    "pydantic>=2.0.0" \
    "jmespath" \
    "datasets>=3.0.0" \
    "huggingface_hub>=0.30.0" \
    "git+https://github.com/huggingface/transformers.git@main"

# Download Spider databases
RUN python scripts/download_spider_databases.py

CMD ["python", "scripts/test_training_local.py"]