streamlit-webrtc 0.53.11__py3-none-any.whl → 0.54.0__py3-none-any.whl

streamlit_webrtc/shutdown.py

@@ -40,7 +40,7 @@ class SessionShutdownObserver:
     ):
         # Use polling because event-based methods are not available
         # to observe the session lifecycle.
-        while not self._polling_thread_stop_event.wait(1.0):
+        while True:
             app_session = app_session_ref()
             if not app_session:
                 logger.debug("AppSession has removed.")
@@ -51,6 +51,11 @@ class SessionShutdownObserver:
                     app_session.id,
                 )
                 break
+            if self._polling_thread_stop_event.wait(1.0):
+                logger.debug(
+                    "The polling thread should be stopped. Exit the polling loop and return."
+                )
+                return
 
         # Ensure the flag is set
         self._polling_thread_stop_event.set()

streamlit_webrtc/webrtc.py

@@ -4,7 +4,7 @@ import itertools
 import logging
 import queue
 import threading
-from typing import Callable, Generic, Literal, Optional, Union, cast
+from typing import TYPE_CHECKING, Callable, Generic, Literal, Optional, Union, cast
 
 from aiortc import RTCConfiguration, RTCPeerConnection, RTCSessionDescription
 from aiortc.contrib.media import MediaPlayer, MediaRecorder, MediaRelay
@@ -39,6 +39,9 @@ from .process import (
 from .receive import AudioReceiver, VideoReceiver
 from .relay import get_global_relay
 
+if TYPE_CHECKING:
+    import concurrent.futures
+
 __all__ = [
     "AudioProcessorBase",
     "AudioProcessorFactory",
@@ -48,7 +51,7 @@ __all__ = [
     "MediaRecorderFactory",
     "VideoProcessorFactory",
     "WebRtcMode",
-    "TimeoutError",
+    "SignallingTimeoutError",
     "WebRtcWorker",
 ]
 
@@ -63,7 +66,7 @@ class WebRtcMode(enum.Enum):
     SENDRECV = enum.auto()
 
 
-class TimeoutError(Exception):
+class SignallingTimeoutError(Exception):
     pass
 
 
@@ -522,7 +525,7 @@ class WebRtcWorker(Generic[VideoProcessorT, AudioProcessorT]):
             if self.pc.iceConnectionState == "failed":
                 await self.pc.close()
 
-        process_offer_task = loop.create_task(
+        process_offer_task = asyncio.run_coroutine_threadsafe(
             _process_offer_coro(
                 self.mode,
                 self.pc,
@@ -540,10 +543,11 @@ class WebRtcWorker(Generic[VideoProcessorT, AudioProcessorT]):
                 sendback_video=self.sendback_video,
                 sendback_audio=self.sendback_audio,
                 on_track_created=on_track_created,
-            )
+            ),
+            loop=loop,
         )
 
-        def callback(done_task: asyncio.Task):
+        def callback(done_task: "concurrent.futures.Future"):
             e = done_task.exception()
             if e:
                 logger.debug("Error occurred in process_offer")
@@ -574,7 +578,7 @@ class WebRtcWorker(Generic[VideoProcessorT, AudioProcessorT]):
             result = self._answer_queue.get(block=True, timeout=timeout)
         except queue.Empty:
             self.stop(timeout=1)
-            raise TimeoutError(
+            raise SignallingTimeoutError(
                 "Processing offer and initializing the worker "
                 f"has not finished in {timeout} seconds"
             )
@@ -672,6 +676,8 @@ class WebRtcWorker(Generic[VideoProcessorT, AudioProcessorT]):
         self._relayed_source_video_track = None
 
     def stop(self, timeout: Union[float, None] = 1.0):
+        logger.debug("Stopping the WebRTC worker")
+
         self._unset_processors()
         if self._process_offer_thread:
             self._process_offer_thread.join(timeout=timeout)
@@ -680,7 +686,7 @@ class WebRtcWorker(Generic[VideoProcessorT, AudioProcessorT]):
         if self.pc and self.pc.connectionState != "closed":
             loop = get_global_event_loop()
             if loop.is_running():
-                loop.create_task(self.pc.close())
+                asyncio.run_coroutine_threadsafe(self.pc.close(), loop=loop)
             else:
                 loop.run_until_complete(self.pc.close())
 

{streamlit_webrtc-0.53.11.dist-info → streamlit_webrtc-0.54.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamlit-webrtc
-Version: 0.53.11
+Version: 0.54.0
 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>

{streamlit_webrtc-0.53.11.dist-info → streamlit_webrtc-0.54.0.dist-info}/RECORD

@@ -14,12 +14,12 @@ streamlit_webrtc/receive.py,sha256=gzv1uZDTAlxJ0hyJ5LweTr0lVuoisrH5xRevGnYkWwo,2
 streamlit_webrtc/relay.py,sha256=2K2h0uNz50ARUOScYZPYlEmZb0uAi0agaYxI9z3mLeI,896
 streamlit_webrtc/server.py,sha256=5o9E2MRIPoS18lfGPJkyhbd23RWyTwblNVVuAA2kKrA,1489
 streamlit_webrtc/session_info.py,sha256=V1EdzD2I8dWANXdC84EKSz8XgPM-zugtkYsb4hJm8Qo,2313
-streamlit_webrtc/shutdown.py,sha256=rbWMhOIrbOBZDlqqBTzCfs4MuHQ0UEhljkNvvXoeTz0,2171
+streamlit_webrtc/shutdown.py,sha256=PUjMoNZcTRGzZCooCmjWARpeVs5EG_9JXAf1Iay7dBM,2353
 streamlit_webrtc/source.py,sha256=haPqMZ50Xh8tg7Z1yN8Frfk8v7D3oOuKteaD59asbzs,2263
-streamlit_webrtc/webrtc.py,sha256=6fSf-2Irqi9JQ2vBIUzWqlP45rJifqnAFva3optZMfE,27311
+streamlit_webrtc/webrtc.py,sha256=3OpdBhhD6aZzee3p40QzXiUi9kB3SXqNdTe8RkyoFAw,27539
 streamlit_webrtc/frontend/dist/index.html,sha256=mae7QlSxz7tCcSszMZG21Q0qXjbQzPun68zD3bjjO0U,527
 streamlit_webrtc/frontend/dist/assets/index-DMhSSwr2.js,sha256=01mdY7O01shSnWVGKCBSfhgZYSulgZ2uhkWotKlqHIA,582849
-streamlit_webrtc-0.53.11.dist-info/METADATA,sha256=zb2vHjnyOQY5cJrim2VnwRP3081mWmr8LetA3dzVLL4,18642
-streamlit_webrtc-0.53.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-streamlit_webrtc-0.53.11.dist-info/licenses/LICENSE,sha256=pwccNHVA7r4rYofGlMU10aKEU90GLUlQr8uY80PR0NQ,1081
-streamlit_webrtc-0.53.11.dist-info/RECORD,,
+streamlit_webrtc-0.54.0.dist-info/METADATA,sha256=My2AH689--OgW07hvK5muh8tz8v_QsBMsk2WBt-HCbc,18641
+streamlit_webrtc-0.54.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+streamlit_webrtc-0.54.0.dist-info/licenses/LICENSE,sha256=pwccNHVA7r4rYofGlMU10aKEU90GLUlQr8uY80PR0NQ,1081
+streamlit_webrtc-0.54.0.dist-info/RECORD,,