Update Dockerfile

Dockerfile

@@ -1,11 +1,12 @@
 FROM ollama/ollama:latest
 
-# Install Python3 and pip
+# Install Python3, pip, and bash
 RUN apt-get update && apt-get install -y \
     python3 \
     python3-pip \
     curl \
     bash \
+    supervisor \
     && rm -rf /var/lib/apt/lists/*
 
 # Set working directory
@@ -14,14 +15,15 @@ WORKDIR /app
 # Copy requirements
 COPY requirements.txt .
 
-# Install Python dependencies (with --break-system-packages for Docker)
+# Install Python dependencies
 RUN pip3 install --break-system-packages --no-cache-dir -r requirements.txt
 
 # Copy application
 COPY app.py .
 
-# Create startup script
-RUN echo '#!/bin/bash\nset -e\necho "Starting Ollama..."\nollama serve &\nOLLAMA_PID=$!\nsleep 5\necho "Pulling model..."\nollama pull mistral\necho "Starting FastAPI..."\npython3 app.py' > /app/start.sh && chmod +x /app/start.sh
+# Create supervisor config to manage both services
+RUN mkdir -p /etc/supervisor/conf.d && \
+    echo '[supervisord]\nnodaemon=true\n\n[program:ollama]\ncommand=/bin/ollama serve\nstartsecs=10\nautorestart=true\nstdout_logfile=/dev/stdout\nstdout_logfile_maxbytes=0\n\n[program:fastapi]\ncommand=/usr/bin/python3 /app/app.py\nstartsecs=5\nautorestart=true\nstdout_logfile=/dev/stdout\nstdout_logfile_maxbytes=0' > /etc/supervisor/conf.d/services.conf
 
 # Expose ports
 EXPOSE 7860 11434
@@ -30,5 +32,5 @@ EXPOSE 7860 11434
 ENV OLLAMA_HOST=0.0.0.0:11434
 ENV PORT=7860
 
-# Start using the script
-CMD ["/app/start.sh"]
+# Start supervisor (manages both ollama and fastapi)
+CMD ["/usr/bin/supervisord", "-c", "/etc/supervisor/supervisord.conf"]