docker-compose.yml

services:
  online-api:
    build:
      context: .
      dockerfile: Dockerfile.backend
    environment:
      DATA_DIR: /app/data/ml-1m
      ARTIFACT_DIR: /app/artifacts/latest
      FEEDBACK_BACKEND: postgres
      CORS_ORIGINS: http://localhost:8080
      REDIS_URL: redis://redis:6379/0
      POSTGRES_DSN: postgresql://recsys:recsys@postgres:5432/recsys
      MILVUS_HOST: milvus
      MILVUS_PORT: "19530"
      MINIO_ENDPOINT: http://minio:9000
      MINIO_ACCESS_KEY: minioadmin
      MINIO_SECRET_KEY: minioadmin
      MINIO_BUCKET: recsys-models
      USE_ONLINE_STORES: "true"
    volumes:
      - ./data:/app/data
      - ./artifacts:/app/artifacts
    depends_on:
      offline-bootstrap:
        condition: service_completed_successfully
      redis:
        condition: service_healthy
      postgres:
        condition: service_healthy
      milvus:
        condition: service_started
      minio:
        condition: service_started
    ports:
      - "8000:8000"
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8000/api/health')"]
      interval: 10s
      timeout: 3s
      retries: 5

  offline-runner:
    build:
      context: .
      dockerfile: Dockerfile.backend
    profiles: ["offline"]
    command:
      [
        "python",
        "scripts/train.py",
        "--data-dir",
        "/app/data/ml-1m",
        "--output",
        "/app/artifacts/latest"
      ]
    environment:
      DATA_DIR: /app/data/ml-1m
      ARTIFACT_DIR: /app/artifacts/latest
      REDIS_URL: redis://redis:6379/0
      POSTGRES_DSN: postgresql://recsys:recsys@postgres:5432/recsys
      MILVUS_HOST: milvus
      MILVUS_PORT: "19530"
      MINIO_ENDPOINT: http://minio:9000
      MINIO_ACCESS_KEY: minioadmin
      MINIO_SECRET_KEY: minioadmin
      MINIO_BUCKET: recsys-models
      PUBLISH_ONLINE_STORES: "true"
    volumes:
      - ./data:/app/data
      - ./artifacts:/app/artifacts
    depends_on:
      redis:
        condition: service_healthy
      postgres:
        condition: service_healthy
      milvus:
        condition: service_started
      minio:
        condition: service_started

  offline-bootstrap:
    build:
      context: .
      dockerfile: Dockerfile.backend
    command: ["python", "scripts/bootstrap_demo.py"]
    environment:
      DATA_DIR: /app/data/ml-1m
      ARTIFACT_DIR: /app/artifacts/latest
      REDIS_URL: redis://redis:6379/0
      POSTGRES_DSN: postgresql://recsys:recsys@postgres:5432/recsys
      MILVUS_HOST: milvus
      MILVUS_PORT: "19530"
      MINIO_ENDPOINT: http://minio:9000
      MINIO_ACCESS_KEY: minioadmin
      MINIO_SECRET_KEY: minioadmin
      MINIO_BUCKET: recsys-models
      PUBLISH_ONLINE_STORES: "true"
    volumes:
      - ./data:/app/data
      - ./artifacts:/app/artifacts
    depends_on:
      redis:
        condition: service_healthy
      postgres:
        condition: service_healthy
      milvus:
        condition: service_started
      minio:
        condition: service_started

  frontend:
    build:
      context: .
      dockerfile: Dockerfile.frontend
      args:
        VITE_API_URL: ""
    depends_on:
      online-api:
        condition: service_healthy
    ports:
      - "8080:80"
    healthcheck:
      test: ["CMD", "wget", "-qO-", "http://127.0.0.1/"]
      interval: 10s
      timeout: 3s
      retries: 5

  e2e-verify:
    image: python:3.11-slim
    profiles: ["verify"]
    working_dir: /app
    command:
      [
        "python",
        "scripts/verify_demo.py",
        "--frontend-url",
        "http://frontend",
        "--api-url",
        "http://online-api:8000",
        "--timeout",
        "180"
      ]
    volumes:
      - ./scripts:/app/scripts:ro
    depends_on:
      frontend:
        condition: service_healthy
      online-api:
        condition: service_healthy

  redis:
    image: redis:7.4-alpine
    ports:
      - "6379:6379"
    healthcheck:
      test: ["CMD", "redis-cli", "ping"]
      interval: 10s
      timeout: 3s
      retries: 5

  postgres:
    image: postgres:16-alpine
    environment:
      POSTGRES_DB: recsys
      POSTGRES_USER: recsys
      POSTGRES_PASSWORD: recsys
    ports:
      - "5432:5432"
    volumes:
      - postgres-data:/var/lib/postgresql/data
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U recsys -d recsys"]
      interval: 10s
      timeout: 3s
      retries: 5

  etcd:
    image: quay.io/coreos/etcd:v3.5.18
    environment:
      ETCD_AUTO_COMPACTION_MODE: revision
      ETCD_AUTO_COMPACTION_RETENTION: "1000"
      ETCD_QUOTA_BACKEND_BYTES: "4294967296"
      ETCD_SNAPSHOT_COUNT: "50000"
    command:
      [
        "etcd",
        "-advertise-client-urls=http://127.0.0.1:2379",
        "-listen-client-urls=http://0.0.0.0:2379",
        "--data-dir=/etcd"
      ]
    volumes:
      - etcd-data:/etcd

  minio:
    image: minio/minio:latest
    command: server /minio_data --console-address ":9001"
    environment:
      MINIO_ROOT_USER: minioadmin
      MINIO_ROOT_PASSWORD: minioadmin
    ports:
      - "9000:9000"
      - "9001:9001"
    volumes:
      - minio-data:/minio_data
  milvus:
    image: milvusdb/milvus:v2.6.4
    command: ["milvus", "run", "standalone"]
    environment:
      ETCD_ENDPOINTS: etcd:2379
      MINIO_ADDRESS: minio:9000
      MINIO_ACCESS_KEY_ID: minioadmin
      MINIO_SECRET_ACCESS_KEY: minioadmin
    ports:
      - "19530:19530"
      - "9091:9091"
    volumes:
      - milvus-data:/var/lib/milvus
    depends_on:
      - etcd
      - minio

volumes:
  postgres-data:
  etcd-data:
  minio-data:
  milvus-data: