FROM python:3.9

WORKDIR /code

# RUN git clone --branch gradio --single-branch https://version.aalto.fi/gitlab/schulte1/llama-annotate.git

# WORKDIR /code/llama-annotate

RUN apt-get update
RUN apt-get install -y cmake ccache wget
RUN apt-get clean && rm -rf /var/lib/apt/lists/*

RUN wget --no-verbose https://huggingface.co/TheBloke/Llama-2-7B-GGUF/resolve/main/llama-2-7b.Q2_K.gguf
RUN wget --no-verbose https://huggingface.co/TheBloke/Llama-2-7B-GGUF/resolve/main/llama-2-7b.Q4_K_M.gguf

COPY llama-annotate/ .

# RUN mkdir -p models
RUN mv *.gguf models

RUN cmake -S . -B build
RUN cmake --build build

# COPY ./requirements.txt /code/requirements.txt
# RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

RUN pip install --no-cache-dir gradio numpy

EXPOSE 7860
ENV GRADIO_SERVER_NAME="0.0.0.0"

CMD ["python", "gui/gui.py"]