Better [INFO] messages

This commit is contained in:
SevaSk 2023-05-16 19:06:21 -04:00
parent 645834e96b
commit c43e801277

View File

@ -14,11 +14,11 @@ class BaseRecorder:
self.source = source self.source = source
self.source_name = source_name self.source_name = source_name
def adjust_for_noise(self): def adjust_for_noise(self, device_name, msg):
print(f"[INFO] Adjusting for ambient noise from {self.source_name}. Please make some noise...") print(f"[INFO] Adjusting for ambient noise from {device_name}. " + msg)
with self.source: with self.source:
self.recorder.adjust_for_ambient_noise(self.source) self.recorder.adjust_for_ambient_noise(self.source)
print(f"[INFO] Completed ambient noise adjustment for {self.source_name}.") print(f"[INFO] Completed ambient noise adjustment for {device_name}.")
def record_into_queue(self, audio_queue): def record_into_queue(self, audio_queue):
def record_callback(_, audio:sr.AudioData) -> None: def record_callback(_, audio:sr.AudioData) -> None:
@ -30,7 +30,7 @@ class BaseRecorder:
class DefaultMicRecorder(BaseRecorder): class DefaultMicRecorder(BaseRecorder):
def __init__(self): def __init__(self):
super().__init__(source=sr.Microphone(sample_rate=16000), source_name="You") super().__init__(source=sr.Microphone(sample_rate=16000), source_name="You")
self.adjust_for_noise() self.adjust_for_noise("Default Mic", "Please make some noise from the Default Mic...")
class DefaultSpeakerRecorder(BaseRecorder): class DefaultSpeakerRecorder(BaseRecorder):
def __init__(self): def __init__(self):
@ -52,4 +52,4 @@ class DefaultSpeakerRecorder(BaseRecorder):
chunk_size=pyaudio.get_sample_size(pyaudio.paInt16), chunk_size=pyaudio.get_sample_size(pyaudio.paInt16),
channels=default_speakers["maxInputChannels"]) channels=default_speakers["maxInputChannels"])
super().__init__(source=source, source_name="Speaker") super().__init__(source=source, source_name="Speaker")
self.adjust_for_noise() self.adjust_for_noise("Default Speaker", "Please make or play some noise from the Default Speaker...")