Update app.py

app.py

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