Add HF TURN server configs

main.py

@@ -1,8 +1,8 @@
 import streamlit as st
-from streamlit_webrtc import webrtc_streamer, get_twilio_ice_servers, __version__ as st_webrtc_version
+from streamlit_webrtc import webrtc_streamer, get_hf_ice_servers, get_twilio_ice_servers, __version__ as st_webrtc_version
 
-frontend_ice_type = st.selectbox("Frontend ICE type", ["Empty", "Google STUN", "Twilio TURN"])
-backend_ice_type = st.selectbox("Backend ICE type", ["Empty", "Google STUN", "Twilio TURN"])
+frontend_ice_type = st.selectbox("Frontend ICE type", ["Empty", "Google STUN", "Twilio TURN", "HF TURN only", "HF TURN and Google STUN"])
+backend_ice_type = st.selectbox("Backend ICE type", ["Empty", "Google STUN", "Twilio TURN", "HF TURN only", "HF TURN and Google STUN"])
 
 if frontend_ice_type == "Empty":
     frontend_rtc_configuration = {
@@ -19,6 +19,17 @@ elif frontend_ice_type == "Twilio TURN":
             twilio_token=st.secrets["TWILIO_AUTH_TOKEN"],
         )
     }
+elif frontend_ice_type == "HF TURN only":
+    hf_ice_servers = get_hf_ice_servers(token=st.secrets["HF_TOKEN"])
+    frontend_rtc_configuration = {
+        "iceServers": hf_ice_servers
+    }
+elif frontend_ice_type == "HF TURN and Google STUN":
+    hf_ice_servers = get_hf_ice_servers(token=st.secrets["HF_TOKEN"])
+    ice_servers = hf_ice_servers + [{"urls": ["stun:stun.l.google.com:19302"]}]
+    frontend_rtc_configuration = {
+        "iceServers": ice_servers
+    }
 
 if backend_ice_type == "Empty":
     backend_rtc_configuration = {
@@ -35,6 +46,18 @@ elif backend_ice_type == "Twilio TURN":
             twilio_token=st.secrets["TWILIO_AUTH_TOKEN"],
         )
     }
+elif backend_ice_type == "HF TURN only":
+    hf_ice_servers = get_hf_ice_servers(token=st.secrets["HF_TOKEN"])
+    backend_rtc_configuration = {
+        "iceServers": hf_ice_servers
+    }
+elif backend_ice_type == "HF TURN and Google STUN":
+    hf_ice_servers = get_hf_ice_servers(token=st.secrets["HF_TOKEN"])
+    ice_servers = hf_ice_servers + [{"urls": ["stun:stun.l.google.com:19302"]}]
+    backend_rtc_configuration = {
+        "iceServers": ice_servers
+    }
+
 
 st.write("Frontend ICE configuration:", frontend_rtc_configuration)
 st.write("Backend ICE configuration:", backend_rtc_configuration)