Dockerfile

# Use official Python image
FROM python:3.11-slim

# Set environment variables
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
ENV OLLAMA_HOST="0.0.0.0"
ENV PYTHONPATH=/app

# Set working directory
WORKDIR /app

# Install minimal system dependencies
RUN apt-get update && apt-get install -y \
    curl \
    && rm -rf /var/lib/apt/lists/* \
    && apt-get clean

# Install Ollama (CPU version) with retry
RUN for i in {1..3}; do \
        if curl -fsSL https://ollama.com/install.sh | sh; then \
            break; \
        else \
            echo "Ollama installation failed, retrying in 5 seconds..."; \
            sleep 5; \
        fi; \
    done && \
    if ! command -v ollama >/dev/null 2>&1; then \
        echo "Ollama installation failed after retries"; \
        exit 1; \
    fi

# Copy requirements first for better caching
COPY requirements.txt .

# Install Python dependencies
RUN pip install --no-cache-dir --upgrade pip \
    && pip install --no-cache-dir -r requirements.txt \
    && pip cache purge

# Copy application files
COPY . .

# Create necessary directories
RUN mkdir -p /app/uploads /app/kg_storage /app/vector_stores /app/chroma_db

# Set proper permissions
RUN chmod -R 755 /app

# Expose port
EXPOSE 8004

# Add health check
HEALTHCHECK --interval=30s --timeout=30s --start-period=10s --retries=3 \
    CMD curl --fail http://localhost:8004/health || exit 1

# Create startup script
RUN echo '#!/bin/bash\n\
set -e\n\
echo "Starting Ollama server..."\n\
ollama serve &\n\
OLLAMA_PID=$!\n\
echo "Waiting for Ollama to be ready..."\n\
sleep 5\n\
echo "Starting FastAPI application..."\n\
uvicorn ontographrag.api.app:app --host 0.0.0.0 --port 8004 --workers 1\n\
' > /app/start.sh && chmod +x /app/start.sh

# Command to run the application
CMD ["/app/start.sh"]