# SAGE-PRO Deployment Image for AMD MI300X (ROCm)
FROM python:3.11-slim

WORKDIR /app

# System deps
RUN apt-get update && apt-get install -y \
    git curl build-essential \
    && rm -rf /var/lib/apt/lists/*

# Python deps
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Copy SAGE-PRO source
COPY app.py .

# Environment
ENV SAGE_MODE=pro
ENV ROCM_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
ENV HSA_OVERRIDE_GFX_VERSION=9.4.2

EXPOSE 7860

CMD ["python", "app.py"]