mirror of
https://github.com/SevaSk/ecoute.git
synced 2024-12-18 20:27:56 +00:00
58 lines
2.4 KiB
Python
58 lines
2.4 KiB
Python
import custom_speech_recognition as sr
|
|
import pyaudiowpatch as pyaudio
|
|
from datetime import datetime
|
|
|
|
RECORD_TIMEOUT = 3
|
|
ENERGY_THRESHOLD = 1000
|
|
DYNAMIC_ENERGY_THRESHOLD = False
|
|
|
|
class DefaultMicRecorder:
|
|
def __init__(self):
|
|
self.recorder = sr.Recognizer()
|
|
self.recorder.energy_threshold = ENERGY_THRESHOLD
|
|
self.recorder.dynamic_energy_threshold = DYNAMIC_ENERGY_THRESHOLD
|
|
self.source = sr.Microphone(sample_rate=16000)
|
|
self.num_channels = 1
|
|
|
|
with self.source:
|
|
self.recorder.adjust_for_ambient_noise(self.source)
|
|
|
|
def record_into_queue(self, audio_queue):
|
|
def record_callback(_, audio:sr.AudioData) -> None:
|
|
data = audio.get_raw_data()
|
|
audio_queue.put(("You", data, datetime.utcnow()))
|
|
|
|
self.recorder.listen_in_background(self.source, record_callback, phrase_time_limit=RECORD_TIMEOUT)
|
|
|
|
class DefaultSpeakerRecorder:
|
|
def __init__(self):
|
|
self.recorder = sr.Recognizer()
|
|
self.recorder.energy_threshold = ENERGY_THRESHOLD
|
|
self.recorder.dynamic_energy_threshold = DYNAMIC_ENERGY_THRESHOLD
|
|
|
|
with pyaudio.PyAudio() as p:
|
|
wasapi_info = p.get_host_api_info_by_type(pyaudio.paWASAPI)
|
|
self.default_speakers = p.get_device_info_by_index(wasapi_info["defaultOutputDevice"])
|
|
|
|
if not self.default_speakers["isLoopbackDevice"]:
|
|
for loopback in p.get_loopback_device_info_generator():
|
|
if self.default_speakers["name"] in loopback["name"]:
|
|
self.default_speakers = loopback
|
|
break
|
|
else:
|
|
print("No loopback device")
|
|
|
|
self.source = sr.Microphone(sample_rate=int(self.default_speakers["defaultSampleRate"]),
|
|
speaker=True,
|
|
chunk_size= pyaudio.get_sample_size(pyaudio.paInt16))
|
|
self.num_channels = self.default_speakers["maxInputChannels"]
|
|
|
|
with self.source:
|
|
self.recorder.adjust_for_ambient_noise(self.source)
|
|
|
|
def record_into_queue(self, audio_queue):
|
|
def record_callback(_, audio:sr.AudioData) -> None:
|
|
data = audio.get_raw_data()
|
|
audio_queue.put(("Speaker", data, datetime.utcnow()))
|
|
|
|
self.recorder.listen_in_background(self.source, record_callback, phrase_time_limit=RECORD_TIMEOUT) |