Update app.py

app.py

@@ -1,4 +1,8 @@
-import streamlit as st  
+
+import streamlit as st
+import sounddevice as sd
+import numpy as np
+import wave
 import gradio as gr  
 import numpy as np      
 from audiorecorder import audiorecorder      
@@ -32,9 +36,30 @@ st.title("Patent Claims Extraction")
 api_key = st.text_input("Enter your OpenAI API Key:", type="password")
 
 # Audio Recording
-audio = st.audio_recorder("Click to record audio", "Click to stop recording")
-
-submit_button = st.button("Use this audio")
+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())
+
+    with st.spinner("Recording..."):
+        with sd.InputStream(callback=audio_callback):
+            st.text("Recording audio. Click 'Stop Recording' when finished.")
+            st.button("Stop Recording")
+
+    st.success("Recording stopped")
+
+    if 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.tobytes())
 
 if submit_button:
     model = whisper.load_model("base")