ARG GRADIO_PORT
ARG MODEL_NAME
ARG QUANTIZE

FROM ghcr.io/huggingface/text-generation-inference:latest

ENV TZ=Europe/Paris \
    PORT=3000

RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
    gnupg \
    curl \
    gettext && \
    rm -rf /var/lib/apt/lists/*
COPY entrypoint.sh entrypoint.sh

RUN mkdir -p /data/db
RUN chown -R 1000:1000 /data
RUN useradd -m -u 1000 user

# Switch to the "user" user
USER user

ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH

COPY app app
RUN pip install -r app/requirements.txt

ENV GRADIO_PORT=$GRADIO_PORT
ENV MODEL_NAME=$MODEL_NAME
ENV QUANTIZE=$QUANTIZE

ENTRYPOINT ["/bin/bash"]
CMD ["entrypoint.sh"]