diff --git a/ttsdatasetrecorder.py b/ttsdatasetrecorder.py index 6b8e538..e5e9cc3 100644 --- a/ttsdatasetrecorder.py +++ b/ttsdatasetrecorder.py @@ -66,10 +66,13 @@ def record_audio(self,out_path,time_allowance): self.ids.goto_text_input.disabled = True recording = self.mic.record(samplerate=self.fs, numframes=int(time_allowance * self.fs),channels=1) - recording = recording / numpy.max(numpy.abs(recording)) - - wavwrite(out_path, self.fs, recording) # Save as WAV file - self.recording_indicator = "" + max_volume = numpy.max(numpy.abs(recording)) + if max_volume != 0: + recording = recording / max_volume + wavwrite(out_path, self.fs, recording) # Save as WAV file + self.recording_indicator = "" + else: + self.recording_indicator = "Nothing recorded. Check your microphone." self.ids.record_button.disabled = False self.ids.reading_speed_slider.disabled = False self.ids.goto_text_input.disabled = False