Add environment configuration and entrypoint script for video processing

- Create .env.example for API credentials and device configuration
- Update Dockerfile to copy entrypoint script
- Implement entrypoint.sh to manage camera settings based on robot type
- Modify docker-compose.yml to include environment variables for robot type and devices
- Enhance supervisord.conf with improved logging and restart policies

Author: openminddev

.env.example

@@ -0,0 +1,16 @@
+# OpenMind API Credentials
+OM_API_KEY_ID=your_api_key_id
+OM_API_KEY=your_api_key
+
+# Robot Configuration
+# Set ROBOT_TYPE to "go2" to automatically enable front camera streaming
+# Options: "default", "go2"
+ROBOT_TYPE=default
+
+# Camera and Microphone Devices
+CAMERA_INDEX=/dev/video0
+MICROPHONE_INDEX=default_mic_aec
+
+# Advanced: Manual override for front camera (auto-detected from ROBOT_TYPE if not set)
+# Uncomment to manually control front camera regardless of ROBOT_TYPE
+# ENABLE_FRONT_CAMERA=true

Dockerfile

@@ -99,11 +99,8 @@ RUN python3 -m pip install --ignore-installed pip setuptools wheel packaging \
 ENV VIRTUAL_ENV=/app/om1_video_processor/.venv
 ENV PATH="/app/om1_video_processor/.venv/bin:${PATH}"
 
-RUN echo '#!/bin/bash' > /entrypoint.sh && \
-    echo 'set -e' >> /entrypoint.sh && \
-    echo 'echo ">> Using baked Python virtualenv at $VIRTUAL_ENV"' >> /entrypoint.sh && \
-    echo 'echo ">> Starting supervisord..."' >> /entrypoint.sh && \
-    echo 'exec /usr/bin/supervisord -c /etc/supervisor/conf.d/supervisord.conf' >> /entrypoint.sh && \
-    chmod +x /entrypoint.sh
+COPY entrypoint.sh /entrypoint.sh
+RUN chmod +x /entrypoint.sh
 
 ENTRYPOINT ["/entrypoint.sh"]
+

README.md

@@ -43,12 +43,16 @@ This tool uses the OM1 modules to create an intelligent video streaming pipeline
    export OM_API_KEY="your_api_key"
    ```
 
-3. (Optional) Configure camera and microphone devices:
+3. (Optional) Configure robot type and devices:
    ```bash
+   export ROBOT_TYPE="go2"              # Set to "go2" to enable front camera
    export CAMERA_INDEX="/dev/video0"    # Default camera device
    export MICROPHONE_INDEX="default_mic_aec"     # Default microphone device
    ```
 
+> [!NOTE]
+> When `ROBOT_TYPE=go2`, the front camera stream is automatically enabled. You can also manually control this with `ENABLE_FRONT_CAMERA=true`.
+
 > [!NOTE]
 > Please refer to the [OpenMind Avatar documentation](https://github.com/OpenMind/OM1-avatar) for the audio and video device configuration details.
 

docker-compose.yml

@@ -13,6 +13,7 @@ services:
     runtime: nvidia
     shm_size: "8g"
     environment:
+      - ROBOT_TYPE=${ROBOT_TYPE:-g1}
       - CAMERA_INDEX=${CAMERA_INDEX:-/dev/video0}
       - MICROPHONE_INDEX=${MICROPHONE_INDEX:-default_mic_aec}
       - MIC_RNNOISE_MODEL_PATH=${MIC_RNNOISE_MODEL_PATH:-/app/om1_video_processor/models/rnnoise.rnnn}

entrypoint.sh

@@ -0,0 +1,18 @@
+#!/bin/bash
+set -e
+
+echo ">> Using baked Python virtualenv at $VIRTUAL_ENV"
+
+# Auto-enable front camera for go2 robot type
+if [ "${ROBOT_TYPE}" = "go2" ] && [ -z "${ENABLE_FRONT_CAMERA}" ]; then
+    export ENABLE_FRONT_CAMERA=true
+    echo ">> ROBOT_TYPE=go2 detected, enabling front camera"
+elif [ -z "${ENABLE_FRONT_CAMERA}" ]; then
+    export ENABLE_FRONT_CAMERA=false
+fi
+
+echo ">> Front camera enabled: ${ENABLE_FRONT_CAMERA}"
+echo ">> Starting supervisord..."
+
+# Start supervisord
+exec /usr/bin/supervisord -c /etc/supervisor/conf.d/supervisord.conf

video_processor/supervisord.conf

@@ -5,6 +5,10 @@ chmod=0700
 [supervisord]
 nodaemon=true
 user=root
+loglevel=info
+logfile=/dev/stdout
+logfile_maxbytes=0
+pidfile=/var/run/supervisord.pid
 
 [supervisorctl]
 serverurl=unix:///var/run/supervisor.sock
@@ -15,6 +19,8 @@ supervisor.rpcinterface_factory = supervisor.rpcinterface:make_main_rpcinterface
 [program:mediamtx]
 command=mediamtx /mediamtx.yml
 autorestart=true
+autorestart=unexpected
+exitcodes=0
 startsecs=5
 stdout_logfile=/dev/stdout
 stdout_logfile_maxbytes=0
@@ -24,8 +30,10 @@ priority=100
 
 [program:top_camera_local]
 command=uv run om_face_recog_stream --device %(ENV_CAMERA_INDEX)s --draw-boxes --draw-names --show-fps --no-window --gallery %(ENV_GALLERY_DIR)s  --embeds-dir %(ENV_EMBEDS_DIR)s --http-host 0.0.0.0 --http-port 6793
-autorestart=true
+autorestart=unexpected
+exitcodes=0
 startsecs=10
+startretries=999999999
 stdout_logfile=/dev/stdout
 stdout_logfile_maxbytes=0
 stderr_logfile=/dev/stderr
@@ -34,8 +42,10 @@ priority=200
 
 [program:top_camera_cloud]
 command=ffmpeg -rtsp_transport tcp -i rtsp://localhost:8554/top_camera -c copy -f rtsp -rtsp_transport tcp rtsp://api-video-ingest.openmind.org:8554/top_camera/%(ENV_OM_API_KEY_ID)s?api_key=%(ENV_OM_API_KEY)s
-autorestart=true
+autorestart=unexpected
+exitcodes=0
 startsecs=15
+startretries=999999999
 stdout_logfile=/dev/stdout
 stdout_logfile_maxbytes=0
 stderr_logfile=/dev/stderr
@@ -44,8 +54,11 @@ priority=300
 
 [program:front_camera_cloud]
 command=ffmpeg -rtsp_transport tcp -i rtsp://localhost:8554/front_camera -c copy -f rtsp -rtsp_transport tcp rtsp://api-video-ingest.openmind.org:8554/front_camera/%(ENV_OM_API_KEY_ID)s?api_key=%(ENV_OM_API_KEY)s
-autorestart=true
+autostart=%(ENV_ENABLE_FRONT_CAMERA)s
+autorestart=unexpected
+exitcodes=0
 startsecs=15
+startretries=999999999
 stdout_logfile=/dev/stdout
 stdout_logfile_maxbytes=0
 stderr_logfile=/dev/stderr
@@ -54,8 +67,10 @@ priority=310
 
 [program:down_camera_cloud]
 command=ffmpeg -rtsp_transport tcp -i rtsp://localhost:8554/down_camera -c copy -f rtsp -rtsp_transport tcp rtsp://api-video-ingest.openmind.org:8554/down_camera/%(ENV_OM_API_KEY_ID)s?api_key=%(ENV_OM_API_KEY)s
-autorestart=true
+autorestart=unexpected
+exitcodes=0
 startsecs=15
+startretries=999999999
 stdout_logfile=/dev/stdout
 stdout_logfile_maxbytes=0
 stderr_logfile=/dev/stderr
@@ -64,8 +79,10 @@ priority=320
 
 [program:mic_local]
 command=ffmpeg -f pulse -fragment_size 480 -i %(ENV_MICROPHONE_INDEX)s -af "arnndn=m=%(ENV_MIC_RNNOISE_MODEL_PATH)s,aresample=async=1:first_pts=0" -map 0:a -c:a libopus -b:a 128k -ar 48000 -ac 2 -application lowdelay -frame_duration 10 -compression_level 0 -packet_loss 0 -f rtsp -rtsp_transport tcp rtsp://localhost:8554/audio
-autorestart=true
+autorestart=unexpected
+exitcodes=0
 startsecs=10
+startretries=999999999
 stdout_logfile=/dev/stdout
 stdout_logfile_maxbytes=0
 stderr_logfile=/dev/stderr
@@ -74,8 +91,10 @@ priority=250
 
 [program:mic_cloud]
 command=ffmpeg -rtsp_transport tcp -i rtsp://localhost:8554/audio -c copy -f rtsp -rtsp_transport tcp rtsp://api-video-ingest.openmind.org:8554/audio/%(ENV_OM_API_KEY_ID)s?api_key=%(ENV_OM_API_KEY)s
-autorestart=true
+autorestart=unexpected
+exitcodes=0
 startsecs=20
+startretries=999999999
 stdout_logfile=/dev/stdout
 stdout_logfile_maxbytes=0
 stderr_logfile=/dev/stderr