Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import requests
 import io
+import base64
 
 # Set the base URL for your FastAPI server
 BASE_URL = "https://hammad712-spam.hf.space"
@@ -43,7 +44,6 @@ if call_test_type == "Spam Test":
     audio_file = st.file_uploader("Upload Fake Audio File", type=["m4a", "wav", "mp3"])
 else:
     st.markdown("For contact calls, the message is not used because a saved contact returns a ringing response.")
-    # For a contact test, we'll create a dummy audio message.
     dummy_message = "dummy"
     audio_file = None
 
@@ -51,7 +51,6 @@ if st.button("Test Call"):
     if not call_number:
         st.error("Please enter a caller number.")
     else:
-        # For Contact Test, use a dummy audio message; for Spam Test, use the uploaded audio file.
         if call_test_type == "Spam Test":
             if audio_file is None:
                 st.error("Please upload a fake audio file for the spam test.")
@@ -59,7 +58,6 @@ if st.button("Test Call"):
                 dummy_audio = audio_file.read()
                 files = {"audio": (audio_file.name, io.BytesIO(dummy_audio), audio_file.type)}
         else:
-            # Create a dummy audio file using dummy_message for contact test.
             dummy_audio = io.BytesIO(dummy_message.encode("utf-8"))
             files = {"audio": ("audio.m4a", dummy_audio, "audio/m4a")}
         data = {"caller_number": call_number}
@@ -68,6 +66,23 @@ if st.button("Test Call"):
                 response = requests.post(f"{BASE_URL}/process-call", data=data, files=files)
                 if response.ok:
                     st.success(f"Response: {response.json()['message']}")
+                    # After process-call is successful, get the audio reply
+                    with st.spinner("Retrieving audio response..."):
+                        audio_reply_resp = requests.get(f"{BASE_URL}/audio-reply/{call_number}")
+                        if audio_reply_resp.ok:
+                            audio_data = audio_reply_resp.json()
+                            st.markdown("### Audio Reply Details")
+                            st.write("**STT Response:**", audio_data.get("stt_response", ""))
+                            st.write("**LLM Reply:**", audio_data.get("llm_reply", ""))
+                            # Decode the base64 audio and display an audio player
+                            audio_b64 = audio_data.get("audio_reply", "")
+                            if audio_b64:
+                                audio_bytes = base64.b64decode(audio_b64)
+                                st.audio(audio_bytes, format="audio/wav")
+                            else:
+                                st.info("No audio reply available.")
+                        else:
+                            st.error(f"Error retrieving audio reply: {audio_reply_resp.status_code} {audio_reply_resp.text}")
                 else:
                     st.error(f"Error: {response.status_code} {response.text}")
             except Exception as e: