Update Dockerfile

Dockerfile

@@ -1,14 +1,11 @@
 # 1. Base Image
-FROM python:3.11-slim
-
-# Set the volume for Ollama data
-# This is where Ollama will store its models and data
-# VOLUME /root/.ollama
+FROM python:3.10-slim
 
 # 2. Set Environment Variables
 ENV PYTHONUNBUFFERED=1
 ENV GRADIO_SERVER_NAME="0.0.0.0"
 ENV OLLAMA_HOST="0.0.0.0:11434"
+ENV HOME=/root  # Ollama will try to create ~/.ollama, which becomes /root/.ollama
 
 # 3. Set Working Directory
 WORKDIR /app
@@ -23,29 +20,31 @@ RUN apt-get update && \
 # 5. Install Ollama
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# 6. Copy Application Requirements
+# 6. Create the .ollama directory for the root user and set permissions
+#    Ollama needs to create its 'id_ed25519' key and 'models' subdir here.
+#    We ensure the base .ollama directory exists and is writable by root.
+RUN mkdir -p /root/.ollama && \
+    chown -R root:root /root/.ollama && \
+    chmod -R 700 /root/.ollama # rwx for owner (root), no access for others
+
+# 7. Copy Application Requirements
 COPY requirements.txt .
 
-# 7. Install Python Dependencies
+# 8. Install Python Dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 8. Copy Your Application Code
+# 9. Copy Your Application Code AND startup script
 COPY app.py .
 COPY startup.sh .
-
-# >>> ADD THIS LINE TO MAKE THE SCRIPT EXECUTABLE <<<
 RUN chmod +x ./startup.sh
 
-# 9. Define Models to Pull (as an Argument with a default list)
-ARG OLLAMA_PULL_MODELS="qwen3:4b qwen3:1.7b qwen3:0.6b" # Default models if not overridden
-
-# Make the ARG available as an ENVironment variable for startup.sh
-ENV OLLAMA_PULL_MODELS=${OLLAMA_PULL_MODELS} 
+# 10. Define Models to Pull (as an Argument with a default list)
+ARG OLLAMA_PULL_MODELS="qwen2:7b mistral:7b" # Default models if not overridden
+ENV OLLAMA_PULL_MODELS=${OLLAMA_PULL_MODELS} # For startup.sh
 
-# 10. Expose Ports
+# 11. Expose Ports
 EXPOSE 11434
 EXPOSE 7860
 
-# 11. Entrypoint/Startup Script - NOW USING EXEC FORM FOR THE SCRIPT
-# CMD ["./startup.sh"] # <-- CHANGE TO THIS
+# 12. Entrypoint/Startup Script
 ENTRYPOINT ["./startup.sh"]