Update Dockerfile

Dockerfile

@@ -5,6 +5,9 @@ FROM ubuntu:22.04
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 ENV PYTHONPATH=/app
+ENV OLLAMA_HOST=0.0.0.0
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
+ENV HF_HOME=/app/.cache/huggingface
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -19,6 +22,16 @@ RUN apt-get update && apt-get install -y \
 # Create symbolic link for python3
 RUN ln -sf /usr/bin/python3.10 /usr/bin/python3
 
+# Create a non-root user
+RUN useradd -m -s /bin/bash appuser
+
+# Create necessary directories with proper permissions
+RUN mkdir -p /app/.cache/huggingface \
+    && mkdir -p /app/.ollama \
+    && chown -R appuser:appuser /app \
+    && chown -R appuser:appuser /app/.cache \
+    && chown -R appuser:appuser /app/.ollama
+
 # Install Ollama
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
@@ -34,6 +47,9 @@ RUN pip3 install --no-cache-dir -r requirements.txt
 # Copy the application code
 COPY . .
 
+# Set proper ownership of all files
+RUN chown -R appuser:appuser /app
+
 # Create a startup script
 RUN echo '#!/bin/bash\n\
 # Start Ollama in the background\n\
@@ -52,5 +68,8 @@ cd /app && uvicorn main:app --host 0.0.0.0 --port 8000\n\
 # Expose ports
 EXPOSE 8000 11434
 
+# Switch to non-root user
+USER appuser
+
 # Set the entrypoint
 ENTRYPOINT ["/app/start.sh"]