Update for stop()

slackr31337 · Aug 13, 2023 · 8f09683 · 8f09683
1 parent 9777083
commit 8f09683
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 20 deletions.
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -9,6 +9,8 @@
         "pvporcupine",
         "pvrecorder",
         "ratecv",
+        "signame",
+        "signum",
         "SIGTERM",
         "simpleaudio",
         "tomono",

diff --git a/voice_pipeline.py b/voice_pipeline.py
@@ -68,6 +68,7 @@ class PorcupinePipeline:
     _sslcontext = None
     _message_id = 1
     _last_ping = 0
+    _recorder = None
     _devices = {}
     _conversation_id = None
     _followup = False
@@ -100,6 +101,11 @@ def __init__(self, args: argparse.Namespace):
             sys.exit(0)
 
         self._porcupine = get_porcupine(self._state)
+        self._recorder = PvRecorder(
+            device_index=args.audio_device,
+            frame_length=self._porcupine.frame_length,
+        )
+
         self._audio_thread = threading.Thread(
             target=self.read_audio,
             daemon=True,
@@ -138,23 +144,26 @@ def start(self) -> None:
         self._event_loop.run_until_complete(self._start_audio_pipeline())
 
     ##########################################
-    def stop(self, **args) -> None:
+    def stop(self, signum=0, frame=None) -> None:
         """Stop audio thread and loop"""
 
         _LOGGER.info("Stopping")
-        if args:
-            _LOGGER.debug(args)
+        if signum:
+            signame = signal.Signals(signum).name
+            _LOGGER.debug(signame)
 
-        self._state.recording = False
         self._state.running = False
-        self._websocket = None
+        self._state.recording = False
+        self._recorder.stop()
 
         self._audio_thread.join(1)
 
         if hasattr(self._porcupine, "delete"):
             self._porcupine.delete()
 
         self._porcupine = None
+        self._websocket = None
+        sys.exit(0)
 
     ##########################################
     async def _ping(self):
@@ -335,11 +344,14 @@ async def stt_task(self) -> None:
             [msg[EVENT][DATA]["runner_data"].get("stt_binary_handler_id")]
         )
 
-        receive_event_task = asyncio.create_task(self._websocket.receive_json())
+        receive_event_task = asyncio.create_task(
+            self._websocket.receive_json(timeout=WEBSOCKET_TIMEOUT)
+        )
+
         while self._state.connected:
             audio_chunk = await self._state.audio_queue.get()
             if not audio_chunk:
-                _LOGGER.error("No audio chunk in queue")
+                break
 
             # Prefix binary message with handler id
             send_audio_task = asyncio.create_task(
@@ -373,9 +385,9 @@ async def stt_task(self) -> None:
                         event_data.get("message"),
                     )
                     break
-                
+
                 elif event_type == "intent-end":
-                    intent = event_data.get("intent_output",{})
+                    intent = event_data.get("intent_output", {})
                     self._conversation_id = intent.get("conversation_id")
 
                 elif event_type == "stt-end":
@@ -397,10 +409,13 @@ async def stt_task(self) -> None:
                 else:
                     _LOGGER.debug("event_type=%s", event_type)
                     _LOGGER.debug("event=%s", event)
-                    #_LOGGER.debug("event_data=%s", event_data)
+                    # _LOGGER.debug("event_data=%s", event_data)
 
                 receive_event_task = asyncio.create_task(self._websocket.receive_json())
 
+            if not self._state.running:
+                break
+
             if send_audio_task not in done:
                 await send_audio_task
 
@@ -413,16 +428,11 @@ def read_audio(self) -> None:
             ratecv_state = None
 
             _LOGGER.debug("Reading audio")
-            recorder = PvRecorder(
-                device_index=args.audio_device,
-                frame_length=self._porcupine.frame_length,
-            )
-
-            recorder.start()
+            self._recorder.start()
             self._state.recording = False
             while self._state.running:
                 try:
-                    pcm = recorder.read()
+                    pcm = self._recorder.read()
 
                 except OSError as err:
                     _LOGGER.error("Exception: %s", err)
@@ -585,6 +595,3 @@ def get_porcupine(state: State) -> Porcupine:
     audio_pipeline = PorcupinePipeline(args)
     with suppress(KeyboardInterrupt):
         audio_pipeline.start()
-
-    audio_pipeline.stop()
-    sys.exit(0)