docker-compose.yml

version: '3.4'
x-common:
  &common
  build:
    context: .
    dockerfile: airflow/Dockerfile
  user: "${AIRFLOW_UID}:0"
  env_file:
    - .env
  volumes:
    - ./airflow/dags:/opt/airflow/dags
    - ./airflow/logs:/opt/airflow/logs
    - ./airflow/plugins:/opt/airflow/plugins
    - ./data/:/opt/airflow/data

    - /var/run/docker.sock:/var/run/docker.sock

x-depends-on:
  &depends-on
  depends_on:
    postgres:
      condition: service_healthy
    airflow-init:
      condition: service_completed_successfully

services:
  label_studio_app:
    image: heartexlabs/label-studio:latest
    expose:
      - "8005"
    ports:
      - 8005:8005
    #depends_on:
    ###   - postgres
    environment:
      - DJANGO_DB=default
      - POSTGRE_NAME=comitiva_label_studio
      - POSTGRE_USER=comitivaesperanca
      - POSTGRE_PASSWORD=Pantanal123
      - POSTGRE_PORT=5432
      - POSTGRE_HOST=pantanaldev-dev.postgres.database.azure.com
      - JSON_LOG=1
      - LABEL_STUDIO_PORT=8005
    #      - LOG_LEVEL=DEBUG
    volumes:
      - ./label-studio/data/:/label-studio/data:rw
    command: [ "label-studio", "start" ]

  ngrok:
    image: shkoliar/ngrok:latest
    ports:
      - 4551:4551
    links:
      - label_studio_app
    environment:
      - AUTH_TOKEN=2O55OUg2iXZk1ITXIKqVtYP4Ygj_cRF232Ab14b6e8qcpAdZ
      - DOMAIN=label_studio_app
      - PORT=8005

  jupyter_lab:
    container_name: jupyter_lab
    build:
      context: .
      dockerfile: jupyter/Dockerfile
    ports:
      - "8888:8888"
    environment:
      - JUPYTER_ENABLE_LAB=yes
      - JUPYTER_TOKEN=
    volumes:
      - ./data/:/home/jovyan/work/data
      - ./jupyter/notebooks/:/home/jovyan/work/notebooks

  postgres:
    image: postgres:13
    container_name: postgres
    ports:
      - "5434:5432"
    healthcheck:
      test: [ "CMD", "pg_isready", "-U", "airflow" ]
      interval: 5s
      retries: 5
    env_file:
      - .env

  scheduler:
    <<: *common
    <<: *depends-on
    container_name: airflow-scheduler
    command: scheduler
    restart: on-failure
    ports:
      - "8793:8793"

  webserver:
    <<: *common
    <<: *depends-on
    container_name: airflow-webserver
    restart: always
    command: webserver
    expose:
      - 8080
    ports:
      - "8080:8080"
    healthcheck:
      test:
        [
          "CMD",
          "curl",
          "--fail",
          "http://localhost:8080/health"
        ]
      interval: 30s
      timeout: 30s
      retries: 5

  airflow-init:
    <<: *common
    container_name: airflow-init
    entrypoint: /bin/bash
    command:
      - -c
      - |
        mkdir -p /sources/logs /sources/dags /sources/plugins
        chown -R "${AIRFLOW_UID}:0" /sources/{logs,dags,plugins}
        exec /entrypoint airflow version