Fix the object detection demo to use Open Relay TURN server

pages/1_object_detection.py

@@ -15,6 +15,7 @@ import streamlit as st
 from streamlit_webrtc import WebRtcMode, webrtc_streamer
 
 from sample_utils.download import download_file
+from sample_utils.turn import get_ice_servers
 
 HERE = Path(__file__).parent
 ROOT = HERE.parent
@@ -137,7 +138,7 @@ def video_frame_callback(frame: av.VideoFrame) -> av.VideoFrame:
 webrtc_ctx = webrtc_streamer(
     key="object-detection",
     mode=WebRtcMode.SENDRECV,
-    rtc_configuration={"iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]},
+    rtc_configuration={"iceServers": get_ice_servers()},
     video_frame_callback=video_frame_callback,
     media_stream_constraints={"video": True, "audio": False},
     async_processing=True,

sample_utils/turn.py

@@ -0,0 +1,26 @@
+import logging
+import os
+
+import requests
+import streamlit as st
+
+logger = logging.getLogger(__name__)
+
+
+OPEN_RELAY_API_HOST = os.environ.get("OPEN_RELAY_API_HOST")
+OPEN_RELAY_API_KEY = os.environ.get("OPEN_RELAY_API_KEY")
+
+
+@st.cache_data
+def get_ice_servers():
+    if not OPEN_RELAY_API_HOST or not OPEN_RELAY_API_KEY:
+        logger.warning(
+            "Open Relay API host or key is not set. Fallback to a free STUN server from Google."  # noqa: E501
+        )
+        return [{"urls": ["stun:stun.l.google.com:19302"]}]
+
+    # Get response from Open Relay API
+    response = requests.get(
+        f"https://{OPEN_RELAY_API_HOST}/api/v1/turn/credentials?apiKey={OPEN_RELAY_API_KEY}"  # noqa: E501
+    )
+    return response.json()