Update app.py

app.py

@@ -1,27 +1,58 @@
 import streamlit as st
 import requests
+import numpy as np
+import sounddevice as sd
+import wave
 import io
 
 # ✅ Set Streamlit Page Config
 st.set_page_config(page_title="Sai Vahini AI Assistant", layout="centered")
 
-# ✅ Render API URL (Replace with your deployed API URL)
+# ✅ Render API URL (Ensure this matches your deployed API on Render)
 RENDER_API_URL = "https://saivahini.onrender.com/process_audio"
 
 # ✅ UI Header
 st.markdown("<h1 style='text-align: center; color: #ff5733;'>Sai Vahini AI Voice Assistant 🕉️</h1>", unsafe_allow_html=True)
 
-# ✅ Upload Audio File
-uploaded_audio = st.file_uploader("🎤 Upload an audio file (WAV only)", type=["wav"])
+# ✅ Audio recording parameters
+DURATION = 5  # Recording duration in seconds
+SAMPLE_RATE = 16000
+
+# ✅ Function to record audio
+def record_audio():
+    """Records live audio and saves it as a WAV file"""
+    st.info("🎤 Recording... Speak now!")
+    audio = sd.rec(int(DURATION * SAMPLE_RATE), samplerate=SAMPLE_RATE, channels=1, dtype=np.int16)
+    sd.wait()  # Wait until recording is finished
+    st.success("✅ Recording completed!")
+
+    # ✅ Save the audio as a WAV file
+    audio_bytes = io.BytesIO()
+    with wave.open(audio_bytes, "wb") as wf:
+        wf.setnchannels(1)
+        wf.setsampwidth(2)
+        wf.setframerate(SAMPLE_RATE)
+        wf.writeframes(audio.tobytes())
+
+    audio_bytes.seek(0)
+    return audio_bytes
+
+# ✅ Record button
+if st.button("🎤 Record Live Audio"):
+    audio_file = record_audio()
+    st.session_state["audio_data"] = audio_file
+
+# ✅ Play recorded audio before sending
+if "audio_data" in st.session_state:
+    st.audio(st.session_state["audio_data"], format="audio/wav")
 
 # ✅ Process Button
-if st.button("✅ Process Audio"):
-    if uploaded_audio is not None:
+if st.button("✅ Process Recorded Audio"):
+    if "audio_data" in st.session_state:
         with st.spinner("🔄 Sending audio to AI model..."):
             try:
                 # ✅ Send recorded audio to Render API
-                files = {"file": (uploaded_audio.name, uploaded_audio, "audio/wav")}
-                response = requests.post(RENDER_API_URL, files=files)
+                response = requests.post(RENDER_API_URL, files={"file": ("audio.wav", st.session_state["audio_data"], "audio/wav")})
 
                 # ✅ Handle API response
                 if response.status_code == 200:
@@ -49,4 +80,4 @@ if st.button("✅ Process Audio"):
                 st.error(f"❌ Failed to connect to API: {str(e)}")
 
     else:
-        st.error("⚠️ Please upload an audio file first!")
+        st.error("⚠️ No audio recorded. Click 'Record Live Audio' first!")