Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 from gradio_client import Client
-from audio_recorder_streamlit import audio_recorder
+from st_audiorec import st_audiorec
 
 
 # Constants
@@ -72,6 +72,12 @@ if prompt := st.chat_input("Ask LLama-2-70b anything..."):
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
     
-audio_bytes = audio_recorder()
-if audio_bytes:
-    st.audio(audio_bytes, format="audio/wav")
+wav_audio_data = st_audiorec()
+
+if wav_audio_data is not None:
+    # Save audio bytes
+    with open("audio.wav", "wb") as f:
+        f.write(audio_bytes)
+    
+    
+