guilhermesimoesbr/guardian-visao:latest

services:
  guardian-visao-amd:
    container_name: guardian-visao-amd
    image: guilhermesimoesbr/guardian-visao-amd:latest
    profiles: ["amd"]
    devices:
      - /dev/kfd
      - ${CARD:-}
      - ${RENDER:-}
    group_add: [video, render]
    ipc: host
    security_opt: [seccomp:unconfined]
    cap_add: [SYS_PTRACE]
    extra_hosts:
      - "host.docker.internal:host-gateway"
    environment:
      ROCR_VISIBLE_DEVICES: "${ROCR_INDEX:-}"
      HSA_XNACK: "${HSA_XNACK:-}"
      HSA_ENABLE_SDMA: "${HSA_ENABLE_SDMA:-}"
      HSA_OVERRIDE_GFX_VERSION: "${HSA_OVERRIDE_GFX_VERSION:-}"
      OPENCV_OPENCL_RUNTIME: "${OPENCV_OPENCL_RUNTIME:-}"
      YOLO_CONFIG_DIR: /app/data
      API_URL: "http://${GUARDIAN_API_HOST:-host.docker.internal}:${GUARDIAN_API_PORT:-5215}"
      LOG_LEVEL: "${GUARDIAN_VISAO_LOG_LEVEL:-INFO}"
      LOG_LEVEL_YOLO: "${GUARDIAN_VISAO_LOG_LEVEL_YOLO:-WARNING}"
      METRICS_PORT: "${GUARDIAN_VISAO_METRICS_PORT:-9090}"
      DETECTION_BATCH_SIZE: "${GUARDIAN_VISAO_DETECTION_BATCH_SIZE:-20}"
    ports:
      - "9090:9090"
    volumes:
      - guardian-visao-data:/app/data
    shm_size: "1gb"
    init: true
    stop_signal: SIGINT
    stop_grace_period: 30s
    restart: no
    networks:
      - monitoring

  guardian-visao-cuda:
    container_name: guardian-visao-cuda
    image: guilhermesimoesbr/guardian-visao-cuda:latest
    profiles: ["cuda"]
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]
    gpus: all
    extra_hosts:
      - "host.docker.internal:host-gateway"
    environment:
      NVIDIA_VISIBLE_DEVICES: "0"
      NVIDIA_DRIVER_CAPABILITIES: "compute,utility"
      YOLO_CONFIG_DIR: /app/data
      API_URL: "http://${GUARDIAN_API_HOST:-host.docker.internal}:${GUARDIAN_API_PORT:-5215}"
      LOG_LEVEL: "${GUARDIAN_VISAO_LOG_LEVEL:-INFO}"
      LOG_LEVEL_YOLO: "${GUARDIAN_VISAO_LOG_LEVEL_YOLO:-WARNING}"
      METRICS_PORT: "${GUARDIAN_VISAO_METRICS_PORT:-9090}"
      DETECTION_BATCH_SIZE: "${GUARDIAN_VISAO_DETECTION_BATCH_SIZE:-20}"
    ports:
      - "9090:9090"
    volumes:
      - guardian-visao-data:/app/data
    shm_size: "1gb"
    init: true
    stop_signal: SIGINT
    stop_grace_period: 30s
    restart: no
    networks:
      - monitoring

  guardian-visao-prometheus:
    image: guilhermesimoesbr/guardian-visao-prometheus:latest
    container_name: guardian-visao-prometheus
    profiles: ["cuda", "amd"]
    ports:
      - "9091:9090"  # Prometheus UI
    volumes:
      - guardian-visao-prometheus-data:/prometheus
    command:
      - '--config.file=/etc/prometheus/prometheus.yml'
      - '--storage.tsdb.path=/prometheus'
      - '--web.console.libraries=/usr/share/prometheus/console_libraries'
      - '--web.console.templates=/usr/share/prometheus/consoles'
    networks:
      - monitoring
    restart: unless-stopped

  guardian-visao-grafana:
    image: guilhermesimoesbr/guardian-visao-grafana:latest
    container_name: guardian-visao-grafana
    profiles: ["cuda", "amd"]
    ports:
      - "3123:3000"  # Grafana UI
    volumes:
      - guardian-visao-grafana-data:/var/lib/grafana
    environment:
      - GF_SECURITY_ADMIN_PASSWORD=abc@1234
      - GF_USERS_ALLOW_SIGN_UP=false
      - GF_PATHS_PROVISIONING=/etc/grafana/provisioning
      - GF_LOG_LEVEL=debug  # Enable debug logging temporarily
    networks:
      - monitoring
    restart: unless-stopped
    depends_on:
      - guardian-visao-prometheus



networks:
  monitoring:
    driver: bridge

volumes:
  guardian-visao-data:
    driver: local
    driver_opts:
      type: none
      o: bind
      device: ${GUARDIAN_VISAO_PATH?GUARDIAN_VISAO_PATH is required}
  guardian-visao-prometheus-data:
  guardian-visao-grafana-data: