FROM python:3.11-slim

RUN apt-get update && apt-get install -y --no-install-recommends \
    build-essential git curl libgl1 \
 && rm -rf /var/lib/apt/lists/*

RUN pip install --upgrade pip \
 && pip install \
    fastapi==0.115.0 uvicorn[standard]==0.30.6 \
    pillow==10.4.0 pydantic==2.9.2 sqlalchemy==2.0.35 \
    python-multipart==0.0.9 \
 && pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

RUN pip install \
    hydra-core \
    iopath \
    matplotlib \
    opencv-python==4.8.0.76 \
    numpy==1.26.4 \
    transformers \
    sentence-transformers \
    pydantic[email]

WORKDIR /app

COPY . /app
EXPOSE 8000
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]