FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime

WORKDIR /app

ENV PYTHONUNBUFFERED=1 \
    PIP_DISABLE_PIP_VERSION_CHECK=1 \
    TOKENIZERS_PARALLELISM=false \
    POLYGUARD_OFFLINE_MODE=false \
    POLYGUARD_MODEL_ID=Qwen/Qwen2.5-0.5B-Instruct \
    POLYGUARD_AUTORUN=1

COPY . .

RUN python -m pip install --upgrade pip setuptools wheel \
    && python -m pip install --no-cache-dir -r requirements.txt \
    && python -m pip install --no-cache-dir --no-build-isolation -e .

EXPOSE 7860

CMD ["python", "-m", "app.hf_space.training_runner"]