# See https://hub.docker.com/r/nikolaik/python-nodejs
#     https://github.com/nikolaik/docker-python-nodejs
# Default user is 'pn' with uid 1000, gid 1000
FROM python:3.10

RUN apt-get update && apt-get install -y \
	git \
	cmake \
	&& rm -rf /var/lib/apt/lists/*

# User
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
	PATH=/home/user/.local/bin:$PATH
WORKDIR /home/user/app

# Install Python deps
RUN pip install --no-cache-dir pip && \
    pip install --no-cache-dir \
        torch \
        numpy \
        transformers

# Prepare bloomz.cpp
RUN git clone https://github.com/NouamaneTazi/bloomz.cpp@4fc96cbf2e2c257eaca1cd7b7ed8e31741e672ee

WORKDIR $HOME/app/bloomz.cpp

RUN make

# Add files
COPY . /home/user/app

ENV PYTHONPATH=$HOME/app \
	PYTHONUNBUFFERED=1 \
	GRADIO_ALLOW_FLAGGING=never \
	GRADIO_NUM_PORTS=1 \
	GRADIO_SERVER_NAME=0.0.0.0 \
	GRADIO_THEME=huggingface \
	SYSTEM=spaces

# Run script
CMD ["python", "app.py"]