PepperWizard/docker-compose.yml at main · Action-Prediction-Lab/PepperWizard · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
# Basic stack for PepperWizard → Pepper robot.
#
# Tracking and proprioception services are optional and live in `docker-compose.dev.yml`, enable them with:
#
#   docker compose -f docker-compose.yml -f docker-compose.dev.yml up
#
# First-run setup:
#   ./setup.sh
#   cp .env.example .env && $EDITOR .env
#   docker compose up

services:
  pepper-robot-env:
    image: ghcr.io/action-prediction-lab/pepper-box:latest
    network_mode: host
    volumes:
      - ./.env:/home/pepperdev/py3-naoqi-bridge/robot.env
      - ${HOME}/.pepperbox/pynaoqi-python2.7-2.5.7.1-linux64:/opt/pynaoqi-python2.7-2.5.7.1-linux64:ro
    env_file:
      - .env
    environment:
      - PYTHONUNBUFFERED=1
    command: [ "/home/pepperdev/entrypoint.sh" ]
    restart: on-failure

  pepper-wizard:
    image: ghcr.io/action-prediction-lab/pepper-wizard:latest
    build: .
    env_file:
      - .env
    environment:
      - HF_HUB_OFFLINE=1
      - TRANSFORMERS_OFFLINE=1
      - PYTHONPATH=/usr/src/app
    command: [ "python", "-m", "pepper_wizard.main", "--proxy-ip", "localhost", "--proxy-port", "5000" ]
    depends_on:
      pepper-robot-env:
        condition: service_started
      stt-service:
        condition: service_healthy
    stdin_open: true
    tty: true
    network_mode: host
    volumes:
      - .:/usr/src/app
    restart: on-failure

  stt-service:
    image: ghcr.io/action-prediction-lab/pepper-stt:latest
    build:
      context: .
      dockerfile: stt-service/Dockerfile
    runtime: ${STT_DOCKER_RUNTIME:-runc}
    network_mode: host
    volumes:
      - ./stt-service:/app
      - ./pepper_wizard/config:/app/pepper_config:ro
      - ${XDG_RUNTIME_DIR:-/run/user/1000}/pulse:/run/user/1000/pulse
      - ${HOME}/.config/pulse/cookie:/root/.config/pulse/cookie:ro
    environment:
      - PULSE_SERVER=unix:/run/user/1000/pulse/native
      - PYTHONUNBUFFERED=1
      - HF_HOME=/opt/asr-cache
      - NVIDIA_VISIBLE_DEVICES=${STT_GPU_DEVICES:-none}
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
    healthcheck:
      # Round-trips a real ZMQ ping so pepper-wizard only starts after STT
      # is answering on :5562 (and the backend model is loaded).
      test: ["CMD", "python", "-c", "import sys, zmq; ctx=zmq.Context.instance(); s=ctx.socket(zmq.REQ); s.setsockopt(zmq.RCVTIMEO,2000); s.setsockopt(zmq.SNDTIMEO,2000); s.setsockopt(zmq.LINGER,0); s.connect('tcp://localhost:5562'); s.send_json({'action':'ping'}); sys.exit(0 if s.recv_json().get('status')=='ok' else 1)"]
      interval: 2s
      timeout: 3s
      retries: 15
      start_period: 60s
    restart: on-failure