jyapayne/vocode-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add assembly ai integration

Browse source
This commit is contained in:
Kian 2023-02-27 10:19:44 -08:00
commit f39af91505
4 changed files with 14 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

4
vocode/helpers.py
View file

@ -11,7 +11,7 @@ def _get_device_prompt(device_infos: list[dict]) -> str:
Choice: """.format(
"\n".join(f"{index}: {device['name']}" for index, device in enumerate(device_infos)))
def create_microphone_input_and_speaker_output(use_first_available_device=False) -> tuple[MicrophoneInput, SpeakerOutput]:
def create_microphone_input_and_speaker_output(use_first_available_device=False, mic_sampling_rate=None) -> tuple[MicrophoneInput, SpeakerOutput]:
pa = pyaudio.PyAudio()
num_devices = pa.get_device_count()
devices = list(map(pa.get_device_info_by_index, range(num_devices)))
@ -24,7 +24,7 @@ def create_microphone_input_and_speaker_output(use_first_available_device=False)
input_device_info = input_device_infos[int(input(_get_device_prompt(input_device_infos)))]
output_device_info = output_device_infos[int(input(_get_device_prompt(output_device_infos)))]
logger.info("Using microphone input device: %s", input_device_info['name'])
microphone_input = MicrophoneInput(pa, input_device_info)
microphone_input = MicrophoneInput(pa, input_device_info, sampling_rate=mic_sampling_rate)
logger.info("Using speaker output device: %s", output_device_info['name'])
speaker_output = SpeakerOutput(pa, output_device_info)
return microphone_input, speaker_output