Upload 3 files
Browse files- sample_utils/__init__.py +0 -0
- sample_utils/download.py +50 -0
- sample_utils/turn.py +39 -0
sample_utils/__init__.py
ADDED
File without changes
|
sample_utils/download.py
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import urllib.request
|
2 |
+
from pathlib import Path
|
3 |
+
|
4 |
+
import streamlit as st
|
5 |
+
|
6 |
+
|
7 |
+
# This code is based on https://github.com/streamlit/demo-self-driving/blob/230245391f2dda0cb464008195a470751c01770b/streamlit_app.py#L48 # noqa: E501
|
8 |
+
def download_file(url, download_to: Path, expected_size=None):
|
9 |
+
# Don't download the file twice.
|
10 |
+
# (If possible, verify the download using the file length.)
|
11 |
+
if download_to.exists():
|
12 |
+
if expected_size:
|
13 |
+
if download_to.stat().st_size == expected_size:
|
14 |
+
return
|
15 |
+
else:
|
16 |
+
st.info(f"{url} is already downloaded.")
|
17 |
+
if not st.button("Download again?"):
|
18 |
+
return
|
19 |
+
|
20 |
+
download_to.parent.mkdir(parents=True, exist_ok=True)
|
21 |
+
|
22 |
+
# These are handles to two visual elements to animate.
|
23 |
+
weights_warning, progress_bar = None, None
|
24 |
+
try:
|
25 |
+
weights_warning = st.warning("Downloading %s..." % url)
|
26 |
+
progress_bar = st.progress(0)
|
27 |
+
with open(download_to, "wb") as output_file:
|
28 |
+
with urllib.request.urlopen(url) as response:
|
29 |
+
length = int(response.info()["Content-Length"])
|
30 |
+
counter = 0.0
|
31 |
+
MEGABYTES = 2.0 ** 20.0
|
32 |
+
while True:
|
33 |
+
data = response.read(8192)
|
34 |
+
if not data:
|
35 |
+
break
|
36 |
+
counter += len(data)
|
37 |
+
output_file.write(data)
|
38 |
+
|
39 |
+
# We perform animation by overwriting the elements.
|
40 |
+
weights_warning.warning(
|
41 |
+
"Downloading %s... (%6.2f/%6.2f MB)"
|
42 |
+
% (url, counter / MEGABYTES, length / MEGABYTES)
|
43 |
+
)
|
44 |
+
progress_bar.progress(min(counter / length, 1.0))
|
45 |
+
# Finally, we remove these visual elements by calling .empty().
|
46 |
+
finally:
|
47 |
+
if weights_warning is not None:
|
48 |
+
weights_warning.empty()
|
49 |
+
if progress_bar is not None:
|
50 |
+
progress_bar.empty()
|
sample_utils/turn.py
ADDED
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import logging
|
2 |
+
import os
|
3 |
+
|
4 |
+
import streamlit as st
|
5 |
+
from twilio.base.exceptions import TwilioRestException
|
6 |
+
from twilio.rest import Client
|
7 |
+
|
8 |
+
logger = logging.getLogger(__name__)
|
9 |
+
|
10 |
+
|
11 |
+
def get_ice_servers():
|
12 |
+
"""Use Twilio's TURN server because Streamlit Community Cloud has changed
|
13 |
+
its infrastructure and WebRTC connection cannot be established without TURN server now. # noqa: E501
|
14 |
+
We considered Open Relay Project (https://www.metered.ca/tools/openrelay/) too,
|
15 |
+
but it is not stable and hardly works as some people reported like https://github.com/aiortc/aiortc/issues/832#issuecomment-1482420656 # noqa: E501
|
16 |
+
See https://github.com/whitphx/streamlit-webrtc/issues/1213
|
17 |
+
"""
|
18 |
+
|
19 |
+
# Ref: https://www.twilio.com/docs/stun-turn/api
|
20 |
+
try:
|
21 |
+
account_sid = os.environ["TWILIO_ACCOUNT_SID"]
|
22 |
+
auth_token = os.environ["TWILIO_AUTH_TOKEN"]
|
23 |
+
except KeyError:
|
24 |
+
logger.warning(
|
25 |
+
"Twilio credentials are not set. Fallback to a free STUN server from Google." # noqa: E501
|
26 |
+
)
|
27 |
+
return [{"urls": ["stun:stun.l.google.com:19302"]}]
|
28 |
+
|
29 |
+
client = Client(account_sid, auth_token)
|
30 |
+
|
31 |
+
try:
|
32 |
+
token = client.tokens.create()
|
33 |
+
except TwilioRestException as e:
|
34 |
+
st.warning(
|
35 |
+
f"Error occurred while accessing Twilio API. Fallback to a free STUN server from Google. ({e})" # noqa: E501
|
36 |
+
)
|
37 |
+
return [{"urls": ["stun:stun.l.google.com:19302"]}]
|
38 |
+
|
39 |
+
return token.ice_servers
|