FROM python:3.11-slim


WORKDIR /app


COPY . .


RUN pip install --upgrade pip


# Install CPU-only torch/torchvision explicitly (avoids pulling ~2GB+ of unused NVIDIA/CUDA packages)
RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu


RUN pip install -r requirements.txt


# Bake both AI models into the image at build time (avoids cold-start timeout
# on the first real user request after a deploy)
RUN python prefetch_models.py


EXPOSE 7860


CMD ["uvicorn","main:app","--host","0.0.0.0","--port","7860"]