Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
-import pyaudio
-import wave
+import sounddevice as sd
 import numpy as np
+import wave
 import whisper
 import os
 import streamlit.components.v1 as components
@@ -37,27 +37,28 @@ record_audio = st.checkbox("Record Audio")
 if record_audio:
     audio_frames = []
 
-    p = pyaudio.PyAudio()
-    stream = p.open(format=pyaudio.paInt16, channels=1, rate=44100, input=True, frames_per_buffer=1024)
+    def audio_callback(indata, frames, time, status):
+        if status:
+            print(status, flush=True)
+        if any(indata):
+            audio_frames.append(indata.copy())
 
-    st.text("Recording audio. Click 'Stop Recording' when finished.")
-    while st.button("Stop Recording"):
-        data = stream.read(1024)
-        audio_frames.append(data)
+    if st.button("Stop Recording"):
+        sd.stop()
 
-    stream.stop_stream()
-    stream.close()
-    p.terminate()
+    with st.spinner("Recording..."):
+        with sd.InputStream(callback=audio_callback):
+            st.text("Recording audio. Click 'Stop Recording' when finished.")
 
     st.success("Recording stopped")
 
     if audio_frames:
-        audio_data = b''.join(audio_frames)
+        audio_data = np.concatenate(audio_frames, axis=0)
         with wave.open("recorded_audio.wav", "wb") as wf:
             wf.setnchannels(1)
             wf.setsampwidth(2)
             wf.setframerate(44100)
-            wf.writeframes(audio_data)
+            wf.writeframes(audio_data.tobytes())
 
 # Moved the submit_button check here
 if 'submit_button' in st.session_state: