CTRL C fix

adapters/audio_player_adapter.py

@@ -74,7 +74,19 @@ def __play(self, file_path: Path, device_index: int) -> None:
         try:
             data, samplerate = sf.read(str(file_path), dtype="float32")
             sd.play(data, samplerate=samplerate, device=device_index)
-            sd.wait()
+
+            while sd.get_stream().active:
+                try:
+                    sd.sleep(100)
+                except KeyboardInterrupt:
+                    self._logger.info("Audio playback interrupted by user")
+                    sd.stop()
+                    raise
+
+        except KeyboardInterrupt:
+            self._logger.info("Audio playback interrupted")
+            sd.stop()
+            raise
         except Exception as exception:  # pylint: disable=broad-exception-caught
             self._logger.exception(
                 f"Error occurred while playing audio: {exception}",)

services/conversation_service.py

@@ -46,6 +46,23 @@ def __init__(self, bypass_jailbreak_result: bool = True) -> None:
     async def run_jailbreak_flow(self, prompt_text: str) -> None:
         """Run the complete jailbreak flow.
 
+        Args:
+            prompt_text: Text of the jailbreak prompt to execute.
+        """
+        try:
+            await self.__execute_jailbreak_flow(prompt_text)
+        except KeyboardInterrupt:
+            self.__logger.info("🛑 Application interrupted by user")
+            await self.__mic_bridge.stop()
+            raise
+        except Exception as exception:
+            self.__logger.exception(f"Unexpected error: {exception}")
+            await self.__mic_bridge.stop()
+            raise
+
+    async def __execute_jailbreak_flow(self, prompt_text: str) -> None:
+        """Execute the main jailbreak flow logic.
+
         Args:
             prompt_text: Text of the jailbreak prompt to execute.
         """
@@ -65,6 +82,10 @@ async def run_jailbreak_flow(self, prompt_text: str) -> None:
             await self.__maintain_mic_forwarding()
             return
 
+        await self.__process_jailbreak_evaluation()
+
+    async def __process_jailbreak_evaluation(self) -> None:
+        """Process jailbreak evaluation flow."""
         self.__logger.info("🔴 Recording model's response...")
         transcript = await self.__record_and_transcribe_response()
         self.__logger.info("🧠 Evaluating jailbreak attempt...")

utils/audio_utils.py

@@ -48,8 +48,6 @@ async def wait_for_silence(self) -> None:
                 raise RuntimeError(
                     "BOT_OUTPUT_DEVICE must be set when running with --verify.",
                 )
-            self.__logger.info(
-                "BOT_OUTPUT_DEVICE is not set. Skipping silence check.",)
             return
 
         input_index = self.__find_device_index()