version: "3.9"
services:
    nginx:
        image: imagestored/nginx${TAG:-}
        ports:
            - "3000:8080" #Change to X:8080 to access on port X
        depends_on:
            - frontend

    frontend:
        image: imagestored/frontend${TAG:-}
        depends_on:
            - backend

    backend:
        image: imagestored/backend${TAG:-}
        restart: always
        environment:
            PGSTRING: postgres://postgres:example@db:5432/postgres # This string has to resolve to a Postgres database on version 11
        volumes:
            - ./media/:/code/media
        depends_on:
            - db

    db:
        image: postgres:11
        restart: always
        environment:
            POSTGRES_PASSWORD: example
        volumes:
            - ./data/:/var/lib/postgresql/data

    detectron2: # You can use either detectron2, which is slower but more accurate, or yolo, which is faster but has less accurate results, and is very lightweight.
        image: imagestored/detectron2${TAG:-}
        environment:
            BASE_ADDRESS: backend:4000 # Set this to the address the backend will be accessible on
            FETCH_INTERVAL: 10 # This is the interval the labeler will poll for new images in seconds
        depends_on:
            - backend
        profiles:
            - detectron

    yolo: # You can use either detectron2, which is slower but more accurate, or yolo, which is faster but has less accurate results, and is very lightweight. 
        image: imagestored/yolo${TAG:-}
        environment:
            BASE_ADDRESS: backend:4000 # Set this to the address the backend will be accessible on
            FETCH_INTERVAL: 10 # This is the interval the labeler will poll for new images in seconds
        depends_on:
            - backend
        profiles:
            - yolo

    search: # CLIP-based search engine to replace the default search feature
        image: imagestored/search${TAG:-}
        restart: always
        environment:
            BASE_ADDRESS: backend:4000 # Set this to the address the backend will be accessible on
            PORT: 5000 # Pick a port to run the search-backend on
        volumes:
            - ./imageFeatures/:/code/features
        depends_on:
            - backend
        profiles:
            - search

    face: # Facial recognition
        image: imagestored/face${TAG:-}
        restart: always
        environment:
            BASE_ADDRESS: backend:4000 # Set this to the address the backend will be accessible on
            PORT: 5000 # Pick a port to run the search-backend on
        volumes:
            - ./faceFeatures/:/code/features
        depends_on:
            - backend
        profiles:
            - face

    autoimport: # Automatic import of media in the "import" directory
        image: imagestored/autoimport${TAG:-}
        restart: always
        environment:
            BACKEND_URL: backend:4000 # Set this to the address the backend will be accessible on
            SLEEP_TIME: 100 # How long to wait after each upload (ms)
            POLLING_RATE: 10000 # How often to check for new files (ms)
        depends_on:
            - backend
        volumes:
            - ./import/:/code/import
        profiles:
            - import