# Use multi-stage builds to combine Python and Ollama
FROM python:3.9 AS python-base

WORKDIR /code

COPY ./requirements.txt /code/requirements.txt

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

# Set up a new user named "user" with user ID 1000
RUN useradd -m -u 1000 user

# Switch to the "user" user
USER user

# Set home to the user's home directory
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH

# Set the working directory to the user's home directory
WORKDIR $HOME/app

# Copy the current directory contents into the container at $HOME/app setting the owner to the user
COPY --chown=user . $HOME/app

# Use Ollama image
FROM ollama/ollama AS ollama-base

# Copy any necessary files
COPY . /root/.ollama

# Expose the port
EXPOSE 11434

# Combine the stages
FROM python-base

COPY --from=ollama-base /root/.ollama /root/.ollama

# Install Ollama
# RUN curl -fsSL https://ollama.com/install.sh | sh
# RUN ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama

# RUN ollama serve & sleep 5 && ollama pull MarcoAland/llama3.1-rag-indo

# RUN ollama pull MarcoAland/llama3.1-rag-indo

EXPOSE 7860
ENV GRADIO_SERVER_NAME="0.0.0.0"

CMD ["python", "app.py"]