services:
  gateway:
    build:
      context: .
      dockerfile: gateway/Dockerfile
    ports:
      - "8003:8000"
    environment:
      - CLIP_URL=http://clip:8000
      - BLIP_URL=http://blip:8000
      - YOLO_URL=http://yolo:8000
      - QDRANT_SVC_URL=http://qdrant-svc:8000
      - API_KEY=please-change-me
      - VISION_TIMEOUT=300
      - MAX_IMAGE_BYTES=52428800
    depends_on:
      - clip
      - blip
      - yolo
      - qdrant-svc

  qdrant:
    image: qdrant/qdrant:latest
    ports:
      - "6333:6333"
    volumes:
      - qdrant_data:/qdrant/storage
    environment:
      - QDRANT__SERVICE__GRPC_PORT=6334

  qdrant-svc:
    build:
      context: .
      dockerfile: qdrant/Dockerfile
    environment:
      - QDRANT_HOST=qdrant
      - QDRANT_PORT=6333
      - CLIP_URL=http://clip:8000
      - COLLECTION_NAME=images
      - VECTOR_DIM=512
    depends_on:
      - qdrant
      - clip

  clip:
    build:
      context: .
      dockerfile: clip/Dockerfile
    environment:
      - MODEL_NAME=ViT-B-32
      - MODEL_PRETRAINED=openai

  blip:
    build:
      context: .
      dockerfile: blip/Dockerfile
    environment:
      #- BLIP_MODEL=Salesforce/blip-image-captioning-base
      - BLIP_MODEL=Salesforce/blip-image-captioning-small

  yolo:
    build:
      context: .
      dockerfile: yolo/Dockerfile
    environment:
      - YOLO_MODEL=yolov8n.pt

volumes:
  qdrant_data: