Zasha1 commited on
Commit
1c4c3ec
·
verified ·
1 Parent(s): 0e6afd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -4
app.py CHANGED
@@ -23,13 +23,28 @@ model = SentenceTransformer('all-MiniLM-L6-v2')
23
 
24
  def record_audio(duration=5, sample_rate=16000):
25
  """
26
- Record audio using sounddevice and return as NumPy array
27
  """
 
 
 
 
 
 
 
28
  st.write("Recording...")
29
- audio = sd.rec(int(duration * sample_rate), samplerate=sample_rate, channels=1, dtype='float32')
30
- sd.wait()
 
 
 
 
 
 
 
31
  st.write("Recording finished.")
32
- return audio.flatten(), sample_rate
 
33
 
34
  def numpy_to_audio_data(audio_data):
35
  """
 
23
 
24
  def record_audio(duration=5, sample_rate=16000):
25
  """
26
+ Record audio using PyAudio and return as NumPy array
27
  """
28
+ p = pyaudio.PyAudio()
29
+ stream = p.open(format=pyaudio.paFloat32,
30
+ channels=1,
31
+ rate=sample_rate,
32
+ input=True,
33
+ frames_per_buffer=1024)
34
+
35
  st.write("Recording...")
36
+ frames = []
37
+ for _ in range(0, int(sample_rate / 1024 * duration)):
38
+ data = stream.read(1024)
39
+ frames.append(np.frombuffer(data, dtype=np.float32))
40
+
41
+ stream.stop_stream()
42
+ stream.close()
43
+ p.terminate()
44
+
45
  st.write("Recording finished.")
46
+ audio_data = np.concatenate(frames)
47
+ return audio_data, sample_rate
48
 
49
  def numpy_to_audio_data(audio_data):
50
  """