Update Dockerfile

Dockerfile

@@ -11,6 +11,10 @@ RUN curl -fsSL https://ollama.com/install.sh | sh
 # Set working directory
 WORKDIR /code
 
+# Set Ollama environment variables
+ENV OLLAMA_HOST=0.0.0.0:11434
+ENV OLLAMA_HOME=/code/.ollama
+
 # Copy requirements and install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
@@ -18,11 +22,20 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy application files
 COPY . .
 
-# Create directory for Ollama
-RUN mkdir -p /root/.ollama
+# Create Ollama directories with proper permissions
+RUN mkdir -p /code/.ollama && chmod 755 /code/.ollama
 
-# Expose port
+# Expose ports
 EXPOSE 7860
-
-# Start Ollama service and then the app
-CMD ollama serve & sleep 10 && ollama pull llm_hub/child_trauma_gemma && python app.py 
+EXPOSE 11434
+
+# Create startup script
+RUN echo '#!/bin/bash\n\
+export OLLAMA_HOME=/code/.ollama\n\
+export OLLAMA_HOST=0.0.0.0:11434\n\
+ollama serve &\n\
+sleep 15\n\
+ollama pull llm_hub/child_trauma_gemma\n\
+python app.py' > /code/start.sh && chmod +x /code/start.sh
+
+CMD ["/code/start.sh"]