# build image
FROM debian:bullseye-slim AS builder
WORKDIR /build/
RUN apt-get update && apt-get install --no-install-recommends -y \
    make gcc g++ wget \
 && apt-get clean \
 && rm -rf /var/lib/apt/lists/*

# Install Whisper.cpp
ADD whisper.cpp/ /build/
RUN make

# main image
FROM python:3.10-slim-bullseye
WORKDIR /app/

# Install dependencies
RUN apt-get update && apt-get install -y \
    ffmpeg libolm-dev gcc make wget\
 && apt-get clean \
 && rm -rf /var/lib/apt/lists/*

ADD requirements.txt .

RUN pip install -r requirements.txt && \
  apt-get remove -y gcc make && \
  apt-get autoremove -y

COPY --from=builder /build/main /app/

VOLUME /data/

ADD ./*.py /app/

ARG PRELOAD_MODEL
ENV PRELOAD_MODEL ${PRELOAD_MODEL}
RUN if [ -n "$PRELOAD_MODEL" ]; then wget -nv "https://ggml.ggerganov.com/ggml-model-whisper-$PRELOAD_MODEL.bin"; fi

CMD ["python3", "-u", "main.py"]