python SDK
This commit is contained in:
commit
6dc9fceeb5
18 changed files with 482 additions and 0 deletions
15
vocode/output_device/base_output_device.py
Normal file
15
vocode/output_device/base_output_device.py
Normal file
|
|
@ -0,0 +1,15 @@
|
|||
from ..models.audio_encoding import AudioEncoding
|
||||
|
||||
class BaseOutputDevice:
|
||||
|
||||
def __init__(self, sampling_rate: int, audio_encoding: AudioEncoding):
|
||||
self.sampling_rate = sampling_rate
|
||||
self.audio_encoding = audio_encoding
|
||||
|
||||
async def send_async(self, chunk):
|
||||
raise NotImplemented
|
||||
|
||||
async def maybe_send_mark_async(self, message):
|
||||
pass
|
||||
|
||||
|
||||
28
vocode/output_device/speaker_output.py
Normal file
28
vocode/output_device/speaker_output.py
Normal file
|
|
@ -0,0 +1,28 @@
|
|||
import pyaudio
|
||||
|
||||
from .base_output_device import BaseOutputDevice
|
||||
from ..models.audio_encoding import AudioEncoding
|
||||
|
||||
class SpeakerOutput(BaseOutputDevice):
|
||||
|
||||
DEFAULT_SAMPLING_RATE = 44100
|
||||
|
||||
def __init__(self, pa: pyaudio.PyAudio, device_info: dict, audio_encoding: AudioEncoding = AudioEncoding.LINEAR16):
|
||||
self.device_info = device_info
|
||||
sampling_rate = int(self.device_info.get('defaultSampleRate', self.DEFAULT_SAMPLING_RATE))
|
||||
super().__init__(sampling_rate, audio_encoding)
|
||||
self.pa = pa
|
||||
self.stream = self.pa.open(
|
||||
output=True,
|
||||
channels=1,
|
||||
rate=self.sampling_rate,
|
||||
format=pyaudio.paInt16,
|
||||
output_device_index=int(self.device_info['index'])
|
||||
)
|
||||
|
||||
async def send_async(self, chunk):
|
||||
self.stream.write(chunk)
|
||||
|
||||
def terminate(self):
|
||||
self.stream.close()
|
||||
self.pa.close()
|
||||
Loading…
Add table
Add a link
Reference in a new issue