diff --git a/audio/audio.py b/audio/audio.py index f38faf9..c990ecd 100644 --- a/audio/audio.py +++ b/audio/audio.py @@ -46,6 +46,9 @@ class Audio: def stopRecord(self, play=True, save=False): print("* done recording") + if self.recordStream is None: + return + self.recordStream.stop_stream() self.recordStream.close() self._isRecording = False @@ -58,7 +61,7 @@ class Audio: self.frames = [] - return savedFileName if save else self.frames + return savedFileName if save else None def _saveFile(self): wf = wave.open(WAVE_OUTPUT_FILENAME, "wb") diff --git a/stt/app.py b/stt/app.py index dfe4f43..3824cbe 100644 --- a/stt/app.py +++ b/stt/app.py @@ -3,8 +3,8 @@ import RPi.GPIO as GPIO -import audio.audio -from simple_recorder.apa102 import apa102 +import audio.audio as audio +import simple_recorder.apa102 as apa102 BUTTON = 17 @@ -36,7 +36,8 @@ while True: if audioInstance.isRecording(): turnOffLED() savedFileName = audioInstance.stopRecord(save=True, play=False) - print("saved file: " + savedFileName) + if savedFileName is not None: + print("saved file: " + savedFileName) else: if not audioInstance.isRecording(): turnOnLED()