phowell
/
pocketflow
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

simplify audio

This commit is contained in:
zachary62 2025-05-13 14:58:16 -04:00
parent 102ef4a0fe
commit fef5bf10a8
1 changed files with 2 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
cookbook/pocketflow-voice-chat/utils/audio_utils.py
View File

@ -1,10 +1,5 @@
import sounddevice as sd import sounddevice as sd
import numpy as np import numpy as np
import time
# import wave # No longer needed for dummy file saving in main for play_audio_file
# import tempfile # No longer needed for dummy file saving in main
# import os # No longer needed for dummy file saving in main
# import soundfile as sf # No longer needed as play_audio_file is removed
DEFAULT_SAMPLE_RATE = 44100 DEFAULT_SAMPLE_RATE = 44100
DEFAULT_CHANNELS = 1 DEFAULT_CHANNELS = 1
@ -39,10 +34,7 @@ def record_audio(sample_rate = DEFAULT_SAMPLE_RATE,
silence_counter = 0 silence_counter = 0
chunks_recorded = 0 chunks_recorded = 0
stream = None with sd.InputStream(samplerate=sample_rate, channels=channels, dtype='float32') as stream:
try:
stream = sd.InputStream(samplerate=sample_rate, channels=channels, dtype='float32')
stream.start()
for i in range(max_chunks): for i in range(max_chunks):
audio_chunk, overflowed = stream.read(chunk_size_frames) audio_chunk, overflowed = stream.read(chunk_size_frames)
@ -76,20 +68,10 @@ def record_audio(sample_rate = DEFAULT_SAMPLE_RATE,
if i == max_chunks - 1 and not is_recording: if i == max_chunks - 1 and not is_recording:
print("No speech detected within the maximum recording duration.") print("No speech detected within the maximum recording duration.")
stream.stop()
stream.close()
return None, sample_rate return None, sample_rate
if not recorded_frames and is_recording: if not recorded_frames and is_recording:
print("Recording started but captured no frames before stopping. This might be due to immediate silence.") print("Recording started but captured no frames before stopping. This might be due to immediate silence.")
except Exception as e:
print(f"Error during recording: {e}")
return None, sample_rate
finally:
if stream and not stream.closed:
stream.stop()
stream.close()
if not recorded_frames: if not recorded_frames:
print("No audio was recorded.") print("No audio was recorded.")