Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
-from streamlit_webrtc import webrtc_streamer, WebRtcMode
-import av
-import wave
 import requests
-import io
 import numpy as np
+import sounddevice as sd
+import wave
+import io
 
+# ✅ Set page title and layout
 st.set_page_config(page_title="Sai Vahini AI Assistant", layout="centered")
 
 # ✅ Render API URL (Ensure this matches your deployed API on Render)
@@ -14,74 +14,16 @@ 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)
 
-# ✅ WebRTC audio processing
-def audio_frame_callback(frame):
-    """Handles incoming audio frames from WebRTC"""
-    audio = frame.to_ndarray(format="s16le")
-    audio_bytes = audio.tobytes()
-    if "frames" not in st.session_state:
-        st.session_state.frames = []
-    st.session_state.frames.append(audio_bytes)
-    return av.AudioFrame.from_ndarray(audio, format="s16", layout="mono")
-
-# ✅ WebRTC Streamer for recording
-st.write("🎤 **Click below to start speaking...**")
-webrtc_streamer(
-    key="audio-recorder",
-    mode=WebRtcMode.SENDRECV,
-    audio_frame_callback=audio_frame_callback,
-    media_stream_constraints={"audio": True, "video": False},
-)
-
-# ✅ Check if audio frames exist
-if "frames" not in st.session_state or not st.session_state.frames:
-    st.warning("⚠️ No audio recorded. Click the button above to start recording.")
-
-# ✅ Process Button
-if st.button("✅ Process Recorded Audio"):
-    if "frames" in st.session_state and st.session_state.frames:
-        with st.spinner("🔄 Processing your voice..."):
-            try:
-                # ✅ Convert recorded audio frames into WAV format
-                audio_bytes = io.BytesIO()
-                with wave.open(audio_bytes, "wb") as wf:
-                    wf.setnchannels(1)
-                    wf.setsampwidth(2)
-                    wf.setframerate(16000)
-                    wf.writeframes(b''.join(st.session_state.frames))
-
-                audio_bytes.seek(0)  # Reset buffer pointer
-
-                # ✅ Send recorded audio to Render API
-                response = requests.post(RENDER_API_URL, files={"file": ("audio.wav", audio_bytes, "audio/wav")})
-
-                # ✅ Handle API response
-                if response.status_code == 200:
-                    result = response.json()
-                    st.success("✅ AI Response:")
-                    st.write("📝 **Transcription:**", result.get("transcription", "No transcription"))
-                    st.write("🤖 **Answer:**", result.get("response", "No response found."))
-
-                    # ✅ Fetch and play AI-generated voice response
-                    audio_response_url = result.get("audio")
-                    if audio_response_url:
-                        st.write(f"🔊 **AI-generated voice response:**")
-                        audio_response = requests.get(audio_response_url)
-                        if audio_response.status_code == 200:
-                            st.audio(audio_response.content, format="audio/wav")
-                        else:
-                            st.error(f"❌ Failed to load AI audio ({audio_response.status_code})")
-                    else:
-                        st.warning("⚠️ No audio response received from API.")
-                    
-                    # ✅ Clear session state for new recording
-                    st.session_state.frames = []
-                
-                else:
-                    st.error(f"❌ API Error: {response.status_code} - {response.text}")
+# ✅ Audio recording parameters
+DURATION = 5  # Seconds
+SAMPLE_RATE = 16000
 
-            except requests.exceptions.RequestException as e:
-                st.error(f"❌ Failed to connect to API: {str(e)}")
+# ✅ Function to record audio
+def record_audio():
+    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!")
 
-    else:
-        st.error("⚠️ No audio captured. Please record first!")
+    # ✅ Save the audio as a WAV file
+    audio_bytes = io.BytesIO