Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from streamlit_webrtc import webrtc_streamer
+from streamlit_webrtc import webrtc_streamer, WebRtcMode
 import requests
 import av
 import numpy as np
@@ -8,9 +8,12 @@ import io
 
 st.title("Sai Vahini AI Voice Assistant 🕉️")
 
-api_url = "https://saivahini.onrender.com/"  # Replace with your Render API clearly!
+# Replace with your actual Render API endpoint
+api_url = "https://saivahini.onrender.com/process_audio"
 
-# Function to save audio as WAV clearly
+frames = []
+
+# Function to save audio as WAV file in memory
 def save_audio(frames, sample_rate=16000):
     audio_bytes = io.BytesIO()
     wf = wave.open(audio_bytes, 'wb')
@@ -22,38 +25,42 @@ def save_audio(frames, sample_rate=16000):
     audio_bytes.seek(0)
     return audio_bytes
 
-# WebRTC Configuration (audio-only clearly)
+# Callback to handle audio frames
 def audio_frame_callback(frame):
     audio = frame.to_ndarray(format="s16le")
     frames.append(audio.tobytes())
     return av.AudioFrame.from_ndarray(audio, format="s16", layout="mono")
 
-frames = []
-
-# UI to record voice input
+# Button to start recording audio
 if st.button("🎙️ Record your voice"):
     frames.clear()
-    webrtc_streamer(key="audio-recorder", mode="sendrecv", audio_frame_callback=audio_frame_callback, media_stream_constraints={"audio": True, "video": False})
+    webrtc_streamer(
+        key="audio-recorder",
+        mode=WebRtcMode.SENDRECV,
+        audio_frame_callback=audio_frame_callback,
+        media_stream_constraints={"audio": True, "video": False}
+    )
 
+# Button to stop recording and process audio
 if st.button("✅ Stop & Process Audio"):
     if frames:
-        st.write("🔄 Processing your voice...")
-        audio_file = save_audio(frames)
-        
-        response = requests.post(api_url, files={"file": ("audio.wav", audio_file, "audio/wav")})
-        
-        if response.status_code == 200:
-            result = response.json()
-            st.success("✅ AI Response:")
-            st.write("**Transcription:**", result["transcription"])
-            st.write("**Answer:**", result["response"])
-            
-            audio_response = result["audio"]
-            
-            # Display response audio
-            audio_data = open(audio_response, 'rb').read()
-            st.audio(audio_data, format='audio/wav')
-        else:
-            st.error(f"❌ API Error: {response.status_code}")
+        with st.spinner("🔄 Processing your voice..."):
+            audio_file = save_audio(frames)
+
+            response = requests.post(api_url, files={"file": ("audio.wav", audio_file, "audio/wav")})
+
+            if response.status_code == 200:
+                result = response.json()
+                st.success("✅ AI Response:")
+                st.write("**Transcription:**", result["transcription"])
+                st.write("**Answer:**", result["response"])
+
+                audio_response_url = result.get("audio")
+
+                # If the API returns a URL, fetch and play the audio directly
+                audio_content = requests.get(audio_response_url).content
+                st.audio(audio_content, format='audio/wav')
+            else:
+                st.error(f"❌ API Error: {response.status_code}")
     else:
-        st.error("⚠️ No audio captured. Please record again.")
+        st.error("⚠️ No audio captured. Please record again.")