Update docker file

Dockerfile

@@ -105,6 +105,14 @@
 # Use a base image with Python
 FROM python:3.10
 
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONPATH=/app
+ENV OLLAMA_HOST=0.0.0.0
+ENV TRANSFORMERS_CACHE=/home/appuser/.cache/huggingface
+ENV HF_HOME=/home/appuser/.cache/huggingface
+ENV API_BASE_URL=http://localhost:8000
+
 # Set the working directory
 WORKDIR /app
 
@@ -115,9 +123,11 @@ RUN useradd -m -u 1000 appuser
 RUN apt update && apt install -y curl && \
     curl -fsSL https://ollama.ai/install.sh | sh
 
-# Create .ollama directory and set permissions
+# Create necessary directories and set permissions
 RUN mkdir -p /home/appuser/.ollama && \
-    chown -R appuser:appuser /home/appuser/.ollama
+    mkdir -p /home/appuser/.cache/huggingface && \
+    chown -R appuser:appuser /home/appuser/.ollama && \
+    chown -R appuser:appuser /home/appuser/.cache
 
 # Install Python dependencies
 COPY requirements.txt requirements.txt
@@ -128,12 +138,19 @@ COPY . .
 
 # Create a more robust startup script
 RUN echo '#!/bin/bash\n\
+set -e\n\
+\n\
+echo "Starting Healthcare AI Assistant..."\n\
+\n\
+# Start Ollama server\n\
 echo "Starting Ollama server..."\n\
 ollama serve &\n\
+OLLAMA_PID=$!\n\
 \n\
 # Wait for Ollama server to be ready\n\
+echo "Waiting for Ollama server to be ready..."\n\
 until curl -s http://localhost:11434/api/tags >/dev/null; do\n\
-    echo "Waiting for Ollama server to be ready..."\n\
+    echo "Waiting for Ollama server..."\n\
     sleep 2\n\
 done\n\
 \n\
@@ -141,8 +158,8 @@ echo "Pulling Mistral model..."\n\
 ollama pull mistral\n\
 \n\
 echo "Starting FastAPI application..."\n\
-uvicorn main:app --host 0.0.0.0 --port 7860' > start.sh && \
-    chmod +x start.sh
+exec uvicorn main:app --host 0.0.0.0 --port 7860 --log-level info\n\
+' > start.sh && chmod +x start.sh
 
 # Set ownership of the application files
 RUN chown -R appuser:appuser /app
@@ -150,8 +167,8 @@ RUN chown -R appuser:appuser /app
 # Switch to non-root user
 USER appuser
 
-# Expose the port FastAPI will run on
-EXPOSE 7860
+# Expose both ports
+EXPOSE 7860 11434
 
 # Set the HOME environment variable
 ENV HOME=/home/appuser