Create Dockerfile

Dockerfile

@@ -0,0 +1,56 @@
+# Use a base image with Python
+FROM python:3.10
+
+# Set the working directory
+WORKDIR /app
+
+# Create a non-root user
+RUN useradd -m -u 1000 appuser
+
+# Install system dependencies
+RUN apt update && apt install -y curl && \
+    curl -fsSL https://ollama.ai/install.sh | sh
+
+# Create .ollama directory and set permissions
+RUN mkdir -p /home/appuser/.ollama && \
+    chown -R appuser:appuser /home/appuser/.ollama
+
+# Install Python dependencies
+COPY requirements.txt requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application files
+COPY . .
+
+# Create a more robust startup script
+RUN echo '#!/bin/bash\n\
+echo "Starting Ollama server..."\n\
+ollama serve &\n\
+\n\
+# Wait 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\
+    sleep 2\n\
+done\n\
+\n\
+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
+
+# Set ownership of the application files
+RUN chown -R appuser:appuser /app
+
+# Switch to non-root user
+USER appuser
+
+# Expose the port FastAPI will run on
+EXPOSE 7860
+
+# Set the HOME environment variable
+ENV HOME=/home/appuser
+
+# Run the startup script
+CMD ["./start.sh"]