ARG ROCM_PYTORCH_IMAGE=rocm/pytorch:latest
FROM ${ROCM_PYTORCH_IMAGE}

WORKDIR /app

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

COPY pyproject.toml ./
ARG INSTALL_EXTRAS=.
RUN pip install --upgrade pip && pip install -e "${INSTALL_EXTRAS}"

COPY app ./app

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