FROM python:3.12-slim

WORKDIR /app

# Install Git, GCC, G++, and Make
RUN apt-get update && apt-get install -y \
    build-essential \
    git \
    && rm -rf /var/lib/apt/lists/*
    
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Download models at build time so the app starts instantly
RUN mkdir -p models && \
    python -c "from huggingface_hub import hf_hub_download; hf_hub_download(repo_id='tc043/step-zero-nemotron', filename='step-zero-nemotron-finetuned.gguf', local_dir='models')" && \
    python -c "from huggingface_hub import hf_hub_download; hf_hub_download(repo_id='mradermacher/MiniCPM3-4B-GGUF', filename='MiniCPM3-4B.Q4_K_M.gguf', local_dir='models')" && \
    mv models/MiniCPM3-4B.Q4_K_M.gguf models/minicpm-3-4b.gguf

COPY . .

ENV STEP_ZERO_MOCK=0
EXPOSE 7860

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