move to sounddevice

vocode/output_device/speaker_output.py

@@ -1,4 +1,5 @@
-import pyaudio
+import sounddevice as sd
+import numpy as np
 
 from .base_output_device import BaseOutputDevice
 from ..models.audio_encoding import AudioEncoding
@@ -7,22 +8,20 @@ class SpeakerOutput(BaseOutputDevice):
 
     DEFAULT_SAMPLING_RATE = 44100
 
-    def __init__(self, pa: pyaudio.PyAudio, device_info: dict, sampling_rate: int = None, audio_encoding: AudioEncoding = AudioEncoding.LINEAR16):
+    def __init__(self, device_info: dict, sampling_rate: int = None, audio_encoding: AudioEncoding = AudioEncoding.LINEAR16):
         self.device_info = device_info
-        sampling_rate = sampling_rate or int(self.device_info.get('defaultSampleRate', self.DEFAULT_SAMPLING_RATE))
+        sampling_rate = sampling_rate or int(self.device_info.get('default_samplerate', self.DEFAULT_SAMPLING_RATE))
         super().__init__(sampling_rate, audio_encoding)
-        self.pa = pa
-        self.stream = self.pa.open(
-            output=True,
+        self.stream = sd.OutputStream(
             channels=1,
-            rate=self.sampling_rate,
-            format=pyaudio.paInt16,
-            output_device_index=int(self.device_info['index'])
+            samplerate=self.sampling_rate,
+            dtype=np.int16,
+            device=int(self.device_info['index'])
         )
+        self.stream.start()
 
     async def send_async(self, chunk):
-        self.stream.write(chunk)
+        self.stream.write(np.frombuffer(chunk, dtype=np.int16))
 
     def terminate(self):
-        self.stream.close()
-        self.pa.close()
+        self.stream.close()