Update Dockerfile

Dockerfile

@@ -20,71 +20,63 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     && rm -rf /var/lib/apt/lists/*
 
 # ----------------------------------------------------------
-# 3. Create a Non-Root User
+# 3. Install Ollama as Root
 # ----------------------------------------------------------
-RUN useradd -m appuser
-
-# ----------------------------------------------------------
-# 4. Set Environment Variables for Ollama
-# ----------------------------------------------------------
-ENV HOME=/home/appuser
-ENV OLLAMA_HOME=/home/appuser/.ollama
+# Set OLLAMA_HOME to a writable directory
+ENV OLLAMA_HOME=/ollama-data
 
-# ----------------------------------------------------------
-# 5. Set Permissions for Ollama Directory
-# ----------------------------------------------------------
-RUN mkdir -p $OLLAMA_HOME && chmod 700 $OLLAMA_HOME
+# Create the Ollama data directory with appropriate permissions
+RUN mkdir -p $OLLAMA_HOME && chmod 755 $OLLAMA_HOME
 
-# ----------------------------------------------------------
-# 6. Switch to Non-Root User to Install Ollama
-# ----------------------------------------------------------
-USER appuser
+# Install Ollama
+RUN curl -fsSL https://ollama.com/install.sh | bash
 
 # ----------------------------------------------------------
-# 7. Install Ollama
+# 4. Create a Non-Root User
 # ----------------------------------------------------------
-RUN curl -fsSL https://ollama.com/install.sh | bash
+RUN useradd -m appuser
 
 # ----------------------------------------------------------
-# 8. Switch Back to Root to Install Python Dependencies
+# 5. Set Permissions for Ollama Directory
 # ----------------------------------------------------------
-USER root
+# Ensure appuser has ownership of the Ollama directory
+RUN chown -R appuser:appuser $OLLAMA_HOME
 
 # ----------------------------------------------------------
-# 9. Set Working Directory for the Application
+# 6. Set Working Directory for the Application
 # ----------------------------------------------------------
 WORKDIR /app
 
 # ----------------------------------------------------------
-# 10. Copy and Install Python Requirements
+# 7. Copy and Install Python Requirements
 # ----------------------------------------------------------
 COPY requirements.txt .
 RUN pip3 install --no-cache-dir -r requirements.txt
 
 # ----------------------------------------------------------
-# 11. Copy Application Files
+# 8. Copy Application Files
 # ----------------------------------------------------------
 COPY app.py /app/app.py
 COPY entrypoint.sh /entrypoint.sh
 RUN chmod +x /entrypoint.sh
 
 # ----------------------------------------------------------
-# 12. Set Environment Variables and Expose Port
+# 9. Set Environment Variables and Expose Port
 # ----------------------------------------------------------
 ENV OLLAMA_API_KEY=change_me
 EXPOSE 7860
 
 # ----------------------------------------------------------
-# 13. Adjust Ownership of Application Directory
+# 10. Adjust Ownership of Application Directory
 # ----------------------------------------------------------
 RUN chown -R appuser:appuser /app
 
 # ----------------------------------------------------------
-# 14. Switch to Non-Root User for Running the Application
+# 11. Switch to Non-Root User for Running the Application
 # ----------------------------------------------------------
 USER appuser
 
 # ----------------------------------------------------------
-# 15. Define Entrypoint
+# 12. Define Entrypoint
 # ----------------------------------------------------------
 CMD ["/entrypoint.sh"]