openmpf/openmpf_compose:latest

x-detection-component-base:
  &detection-component-base
  depends_on:
    - workflow-manager
  volumes:
    - shared_data:/opt/mpf/share

services:
  db:
    image: postgres:17-alpine
    environment:
      POSTGRES_DB: mpf
      POSTGRES_USER: mpf
      POSTGRES_PASSWORD: password
    volumes:
      - db_data:/var/lib/postgresql/data
    deploy:
      placement:
        constraints:
          - node.role == manager

  redis:
    image: redis:alpine

  workflow-manager:
    image: openmpf/openmpf_workflow_manager:${TAG:-latest}
    ports:
      - target: 8080
        published: ${WFM_PORT:-8080}
    depends_on:
      - db
      - redis
    volumes:
      - shared_data:/opt/mpf/share
    deploy:
      placement:
        constraints:
          - node.role == manager

  markup:
    image: openmpf/openmpf_markup:${TAG:-latest}
    volumes:
      - shared_data:/opt/mpf/share
    deploy:
      mode: global

  argos-translation:
    <<: *detection-component-base
    image: openmpf/openmpf_argos_translation:${TAG:-latest}

  clip-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_clip_detection:${TAG:-latest}

#  clip-detection-server:
#    image: openmpf/openmpf_clip_detection_server:${TAG:-latest}
#    deploy:
#      mode: global
#      # Expose GPU to the server. At least one GPU is required.
#      resources:
#        reservations:
#          devices:
#            - driver: nvidia
#              device_ids: ['0']
#              capabilities: [gpu]
#    ulimits:
#      memlock: -1  # don't limit locked-in memory (prevent paging)
#      stack: 67108864  # 8 GiB
#    # ports:
#      # - "8001:8001"  # server gRPC port (expose to enable handling requests from outside the stack)
#      # - "8000:8000"  # (optional) HTTP management port
#      # - "8002:8002"  # (optional) Prometheus metrics at http://<host>:<this-port>/metrics
#    command: [tritonserver,
#              --model-repository=/models,
#              --strict-model-config=false,
#              --model-control-mode=explicit,
#              --load-model=vit_l_14,
#              # --log-verbose=1,  # (optional)
#              --grpc-infer-allocation-pool-size=16 ]

  east-text-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_east_text_detection:${TAG:-latest}

  fasttext-language-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_fasttext_language_detection:${TAG:-latest}

  keyword-tagging:
    <<: *detection-component-base
    image: openmpf/openmpf_keyword_tagging:${TAG:-latest}

  # llama-video-summarization:
  #   <<: *detection-component-base
  #   image: openmpf/openmpf_llama_video_summarization:${TAG:-latest}
  #   deploy:
  #     resources:
  #       reservations:
  #         devices:
  #           # Expose GPU to the component. A GPU is required.
  #           - driver: nvidia
  #             device_ids: ['0']
  #             capabilities: [gpu]

  mog-motion-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_mog_motion_detection:${TAG:-latest}

  # nllb-translation:
  #   <<: *detection-component-base
  #   image: openmpf/openmpf_nllb_translation:${TAG:-latest}
  #   deploy:
  #     resources:
  #       reservations:
  #         devices:
  #             # Expose GPU to the component. A GPU is required.
  #           - driver: nvidia
  #             device_ids: ['0']
  #             capabilities: [gpu]

  nlp-text-correction:
    <<: *detection-component-base
    image: openmpf/openmpf_nlp_text_correction:${TAG:-latest}

  oalpr-license-plate-text-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_oalpr_license_plate_text_detection:${TAG:-latest}

  ocv-dnn-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_ocv_dnn_detection:${TAG:-latest}

  ocv-face-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_ocv_face_detection:${TAG:-latest}

  ocv-yolo-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_ocv_yolo_detection:${TAG:-latest}

#  ocv-yolo-detection-server:
#    image: openmpf/openmpf_ocv_yolo_detection_server:${TAG:-latest}
#    deploy:
#      mode: global
#      # Expose GPUs to the server. At least one GPU is required.
#      resources:
#        reservations:
#          devices:
#            - driver: nvidia
#              device_ids: ['0']
#              capabilities: [gpu]
#    ulimits:
#      memlock: -1  # don't limit locked-in memory (prevent paging)
#      stack: 67108864  # 8 GiB
#    # ports:
#    #  - "8001:8001"  # server gRPC port (expose to enable handling requests from outside the stack)
#    #  - "8000:8000"  # (optional) HTTP management port
#    #  - "8002:8002"  # (optional) Prometheus metrics at http://<host>:<this-port>/metrics
#    environment:
#      - LD_PRELOAD=/plugins/libyolo608layerplugin.so
#    volumes:
#      # (optional) Volume for shared memory with OcvYoloDetection component on same host
#      # - "/dev/shm:/dev/shm"
#      # At runtime the server generates a GPU-specific engine file for the YOLO model. Caching it in a volume saves
#      # initialization time. To save space, manually remove generated engine files that are no longer needed.
#      - yolo_engine_files:/models
#    command: [tritonserver,
#              --model-repository=/models,
#              --strict-model-config=false,
#              --model-control-mode=explicit,
#              --load-model=yolo-608,
#              # --log-verbose=1,  # (optional)
#              --grpc-infer-allocation-pool-size=16 ]

  ortools-subject-component:
    <<: *detection-component-base
    image: openmpf/openmpf_ortools_subject_component:${TAG:-latest}

  scene-change-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_scene_change_detection:${TAG:-latest}

  sphinx-speech-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_sphinx_speech_detection:${TAG:-latest}
    environment:
      JAVA_TOOL_OPTIONS: -Xmx1g  # limit Java heap size to 1GB

  subsense-motion-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_subsense_motion_detection:${TAG:-latest}

  tesseract-ocr-text-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_tesseract_ocr_text_detection:${TAG:-latest}

  tika-image-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_tika_image_detection:${TAG:-latest}

  tika-text-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_tika_text_detection:${TAG:-latest}

  transformer-tagging:
    <<: *detection-component-base
    image: openmpf/openmpf_transformer_tagging:${TAG:-latest}

  whisper-speech-detection:
    <<: *detection-component-base
    image: openmpf/openmpf_whisper_speech_detection:${TAG:-latest}

  #  azure-form-detection:
  #    <<: *detection-component-base
  #    image: openmpf/openmpf_azure_form_detection:${TAG:-latest}
  #    environment:
  #      MPF_PROP_ACS_URL: https://eastus.api.cognitive.microsoft.com/formrecognizer/v2.0/layout/analyze
  #      MPF_PROP_ACS_SUBSCRIPTION_KEY: <Your Azure subscription key goes here.>

  # azure-read-text-detection:
  #   <<: *detection-component-base
  #   image: openmpf/openmpf_azure_read_text_detection:${TAG:-latest}
  #   environment:
  #     MPF_PROP_ACS_URL: https://eastus.api.cognitive.microsoft.com/vision/v3.1/read/analyze
  #     MPF_PROP_ACS_SUBSCRIPTION_KEY: <Your Azure subscription key goes here.>

  # azure-speech-detection:
  #   <<: *detection-component-base
  #   image: openmpf/openmpf_azure_speech_detection:${TAG:-latest}
  #   environment:
  #     MPF_PROP_ACS_URL: https://eastus.api.cognitive.microsoft.com/speechtotext/v3.0/transcriptions
  #     MPF_PROP_ACS_SUBSCRIPTION_KEY: <Your Azure subscription key goes here.>
  #     MPF_PROP_ACS_BLOB_CONTAINER_URL: https://myaccount.blob.core.windows.net/mycontainer
  #     MPF_PROP_ACS_BLOB_SERVICE_KEY: <Your Azure blob storage service key goes here.>

  # azure-translation:
  #   <<: *detection-component-base
  #   image: openmpf/openmpf_azure_translation:${TAG:-latest}
  #   environment:
  #     MPF_PROP_ACS_URL: https://api.cognitive.microsofttranslator.com
  #     MPF_PROP_ACS_SUBSCRIPTION_KEY: <Your Azure subscription key goes here.>

volumes:
  shared_data:
  db_data:
  yolo_engine_files: