x-env:
  &template
  image: fcb_photo_review:1.14.1
  restart: always

x-review:
  &review_template
  <<: *template
  volumes:
    - ./log:/app/log
    - ./model:/app/model
  deploy:
    resources:
      reservations:
        devices:
          - device_ids: [ '0', '1' ]
            capabilities: [ 'gpu' ]
            driver: 'nvidia'

x-mask:
  &mask_template
  <<: *template
  volumes:
    - ./log:/app/log
  deploy:
    resources:
      reservations:
        devices:
          - device_ids: [ '1' ]
            capabilities: [ 'gpu' ]
            driver: 'nvidia'

services:
  det_api:
    <<: *template
    build:
      context: .
    container_name: det_api
    hostname: det_api
    volumes:
      - ./log:/app/log
      - ./model:/app/model
#    command: [ 'det_api.py' ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ '0' ]
              capabilities: [ 'gpu' ]
              driver: 'nvidia'

  photo_review_1:
    <<: *review_template
    container_name: photo_review_1
    hostname: photo_review_1
    depends_on:
      - det_api
    command: [ 'photo_review.py', '--clean', 'True' ]

  photo_review_2:
    <<: *review_template
    container_name: photo_review_2
    hostname: photo_review_2
    depends_on:
      - photo_review_1
    command: [ 'photo_review.py' ]

  photo_review_3:
    <<: *review_template
    container_name: photo_review_3
    hostname: photo_review_3
    depends_on:
      - photo_review_2
    command: [ 'photo_review.py' ]

  photo_review_4:
    <<: *review_template
    container_name: photo_review_4
    hostname: photo_review_4
    depends_on:
      - photo_review_3
    command: [ 'photo_review.py' ]

  photo_review_5:
    <<: *review_template
    container_name: photo_review_5
    hostname: photo_review_5
    depends_on:
      - photo_review_4
    command: [ 'photo_review.py' ]

  photo_mask_1:
    <<: *mask_template
    container_name: photo_mask_1
    hostname: photo_mask_1
    depends_on:
      - photo_review_5
    command: [ 'photo_mask.py', '--clean', 'True' ]

  photo_mask_2:
    <<: *mask_template
    container_name: photo_mask_2
    hostname: photo_mask_2
    depends_on:
      - photo_mask_1
    command: [ 'photo_mask.py' ]