#talk2knowledgegraphs
services:
  # CPU variant – start with: docker compose --profile cpu up
  ollama:
    image: ollama/ollama:latest
    container_name: ollama-server
    ports:
      - "11434:11434"
    volumes:
      - ollama_data:/root/.ollama
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:11434/api/version"]
      interval: 30s
      timeout: 10s
      retries: 3
    entrypoint: ["/bin/sh", "-c"]
    command: >
      "ollama serve &
      sleep 10 &&
      ollama pull nomic-embed-text &&
      tail -f /dev/null"
    networks:
      - app-network
    profiles: ["cpu"]

  # NVIDIA GPU variant – start with: docker compose --profile nvidia up
  ollama-nvidia:
    image: ollama/ollama:latest
    container_name: ollama-server-nvidia
    ports:
      - "11434:11434"
    volumes:
      - ollama_data:/root/.ollama
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:11434/api/version"]
      interval: 30s
      timeout: 10s
      retries: 3
    entrypoint: ["/bin/sh", "-c"]
    command: >
      "ollama serve &
      sleep 10 &&
      ollama pull nomic-embed-text &&
      tail -f /dev/null"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    networks:
      - app-network
    profiles: ["nvidia"]

  # AMD GPU variant – start with: docker compose --profile amd up
  ollama-amd:
    image: ollama/ollama:rocm
    container_name: ollama-server-amd
    ports:
      - "11434:11434"
    volumes:
      - ollama_data:/root/.ollama
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:11434/api/version"]
      interval: 30s
      timeout: 10s
      retries: 3
    entrypoint: ["/bin/sh", "-c"]
    command: >
      "ollama serve &
      sleep 10 &&
      ollama pull nomic-embed-text &&
      tail -f /dev/null"
    devices:
      - /dev/kfd
      - /dev/dri
    networks:
      - app-network
    profiles: ["amd"]

  talk2aiagents4pharma:
    image: virtualpatientengine/talk2knowledgegraphs:latest
    container_name: talk2knowledgegraphs
    ports:
      - "8501:8501"
    env_file:
      - .env
    restart: unless-stopped
    networks:
      - app-network

volumes:
  ollama_data:
    name: ollama_data
    # Uncomment the line below if you want to use an existing volume:
    # external: true

networks:
  app-network:
    name: app-network