streamlit-webrtc 0.62.4__tar.gz → 0.63.1__tar.gz

{streamlit_webrtc-0.62.4 → streamlit_webrtc-0.63.1}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: streamlit-webrtc
-Version: 0.62.4
+Version: 0.63.1
 Summary: Real-time video and audio processing on Streamlit
 Project-URL: Repository, https://github.com/whitphx/streamlit-webrtc
 Author-email: "Yuichiro Tachibana (Tsuchiya)" <t.yic.yt@gmail.com>
 License-Expression: MIT
 License-File: LICENSE
 Requires-Python: !=3.9.7,>=3.9
-Requires-Dist: aioice>=0.10.0
+Requires-Dist: aioice>=0.10.1
 Requires-Dist: aiortc>=1.11.0
 Requires-Dist: packaging>=20.0
 Requires-Dist: streamlit>=0.89.0

{streamlit_webrtc-0.62.4 → streamlit_webrtc-0.63.1}/pyproject.toml

@@ -13,7 +13,7 @@ dependencies = [
     "streamlit>=0.89.0",
     "aiortc>=1.11.0", # aiortc<1.4.0 causes an error with cryptography>=39.0.0. See https://github.com/whitphx/streamlit-webrtc/issues/1164. The fix was introduced into aiortc in https://github.com/aiortc/aiortc/commit/08b0a7e9f5030a9f7e5617382e92560d4ae763a2 that 1.4.0 included.
     "packaging>=20.0",
-    "aioice>=0.10.0",
+    "aioice>=0.10.1",
 ]
 
 [project.urls]

{streamlit_webrtc-0.62.4 → streamlit_webrtc-0.63.1}/streamlit_webrtc/__init__.py

@@ -21,9 +21,19 @@ from .credentials import (
     get_hf_ice_servers,
     get_twilio_ice_servers,
 )
-from .factory import create_mix_track, create_process_track, create_video_source_track
+from .factory import (
+    create_audio_source_track,
+    create_mix_track,
+    create_process_track,
+    create_video_source_track,
+)
 from .mix import MediaStreamMixTrack, MixerCallback
-from .source import VideoSourceCallback, VideoSourceTrack
+from .source import (
+    AudioSourceCallback,
+    AudioSourceTrack,
+    VideoSourceCallback,
+    VideoSourceTrack,
+)
 from .webrtc import (
     AudioProcessorBase,
     AudioProcessorFactory,
@@ -64,7 +74,10 @@ __all__ = [
     "VideoReceiver",
     "VideoSourceTrack",
     "VideoSourceCallback",
+    "AudioSourceTrack",
+    "AudioSourceCallback",
     "create_video_source_track",
+    "create_audio_source_track",
     "WebRtcMode",
     "WebRtcWorker",
     "MediaStreamConstraints",

{streamlit_webrtc-0.62.4 → streamlit_webrtc-0.63.1}/streamlit_webrtc/factory.py

@@ -25,7 +25,12 @@ from .process import (
     VideoProcessTrack,
 )
 from .relay import get_global_relay
-from .source import VideoSourceCallback, VideoSourceTrack
+from .source import (
+    AudioSourceCallback,
+    AudioSourceTrack,
+    VideoSourceCallback,
+    VideoSourceTrack,
+)
 
 _PROCESSOR_TRACK_CACHE_KEY_PREFIX = "__PROCESSOR_TRACK_CACHE__"
 
@@ -204,3 +209,32 @@ def create_video_source_track(
         video_source_track = VideoSourceTrack(callback=callback, fps=fps)
         st.session_state[cache_key] = video_source_track
     return video_source_track
+
+
+_AUDIO_SOURCE_TRACK_CACHE_KEY_PREFIX = "__AUDIO_SOURCE_TRACK_CACHE__"
+
+
+def create_audio_source_track(
+    callback: AudioSourceCallback,
+    key: str,
+    sample_rate: int = 48000,
+    ptime: float = 0.020,
+) -> AudioSourceTrack:
+    cache_key = _AUDIO_SOURCE_TRACK_CACHE_KEY_PREFIX + key
+    if (
+        cache_key in st.session_state
+        and isinstance(st.session_state[cache_key], AudioSourceTrack)
+        and st.session_state[cache_key].kind == "audio"
+        and st.session_state[cache_key].readyState == "live"
+    ):
+        audio_source_track: AudioSourceTrack = st.session_state[cache_key]
+        audio_source_track._callback = callback
+        audio_source_track._sample_rate = sample_rate
+        audio_source_track._ptime = ptime
+        audio_source_track._samples_per_frame = int(sample_rate * ptime)
+    else:
+        audio_source_track = AudioSourceTrack(
+            callback=callback, sample_rate=sample_rate, ptime=ptime
+        )
+        st.session_state[cache_key] = audio_source_track
+    return audio_source_track