Update Dockerfile

Dockerfile

@@ -1,22 +1,12 @@
-FROM python:3.9
-FROM ollama/ollama
+# Use multi-stage builds to combine Python and Ollama
+FROM python:3.9 AS python-base
 
 WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
-#####
-# Copy any necessary files
-COPY . /root/.ollama
 
-# Expose the port
-EXPOSE 11434
-#######
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-# Install Ollama
-# RUN curl -fsSL https://ollama.com/install.sh | sh
-# RUN ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
-
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 
@@ -33,8 +23,26 @@ 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
 
-CMD python app.py
+CMD ["python", "app.py"]