Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import requests
 import io
 import base64
+import time
 
 # Set the base URL for your FastAPI server
 BASE_URL = "https://hammad712-spam.hf.space"
@@ -13,9 +14,9 @@ with st.form("save_contact_form"):
     contact_phone = st.text_input("Enter Contact Phone (e.g., +1-555-901-2345)")
     contact_name = st.text_input("Enter Contact Name (optional)", value="John Doe")
     contact_email = st.text_input("Enter Contact Email (optional)", value="john.doe@example.com")
-    submitted = st.form_submit_button("Save Contact")
+    save_submitted = st.form_submit_button("Save Contact")
     
-if submitted:
+if save_submitted:
     if not contact_phone:
         st.error("Please enter a contact phone number.")
     else:
@@ -36,62 +37,83 @@ if submitted:
 
 st.markdown("---")
 st.markdown("## Call Test")
-call_test_type = st.radio("Select Call Test Type", ("Contact Test", "Spam Test"))
-call_number = st.text_input("Enter Caller Number (e.g., +1-555-901-2345 or +1-555-654-3210):", key="call_number")
 
-if call_test_type == "Spam Test":
-    st.markdown("For spam calls, upload the fake audio message below.")
-    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.")
-    dummy_message = "dummy"
-    audio_file = None
+call_number = st.text_input("Enter Caller Number (e.g., +1-555-654-3210 for spam or +1-555-901-2345 for contact):", key="call_number")
+
+# For this task, we only simulate spam calls if the caller is not found in contacts.
+st.markdown("For spam calls, upload the fake audio message below.")
+audio_file = st.file_uploader("Upload Fake Audio File", type=["m4a", "wav", "mp3"], key="spam_audio")
 
 if st.button("Test Call"):
     if not call_number:
         st.error("Please enter a caller number.")
     else:
-        if call_test_type == "Spam Test":
+        # First, check if the contact is saved by calling GET /contacts/{phone}
+        with st.spinner("Checking contact..."):
+            contact_resp = requests.get(f"{BASE_URL}/contacts/{call_number}")
+        if contact_resp.ok:
+            # Contact exists – show ringing message immediately.
+            st.info(f"Call from {call_number} – Ringing")
+        else:
+            # No saved contact → treat as spam call.
+            placeholder = st.empty()
+            placeholder.markdown(f"**Spam call from {call_number} – Bot replying**")
+            
+            # Ensure an audio file is uploaded.
             if audio_file is None:
                 st.error("Please upload a fake audio file for the spam test.")
-            else:
-                dummy_audio = audio_file.read()
-                files = {"audio": (audio_file.name, io.BytesIO(dummy_audio), audio_file.type)}
-        else:
-            dummy_audio = io.BytesIO(dummy_message.encode("utf-8"))
-            files = {"audio": ("audio.m4a", dummy_audio, "audio/m4a")}
-        data = {"caller_number": call_number}
-        with st.spinner("Processing call..."):
-            try:
-                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}")
+                st.stop()
+            file_bytes = audio_file.read()
+            
+            # ----- Step 1: Call the STT endpoint -----
+            with st.spinner("Processing STT..."):
+                files = {"audio": (audio_file.name, io.BytesIO(file_bytes), audio_file.type)}
+                stt_resp = requests.post(f"{BASE_URL}/stt", files=files)
+                if stt_resp.ok:
+                    stt_data = stt_resp.json()
+                    transcription = stt_data.get("transcription", "No transcription")
                 else:
-                    st.error(f"Error: {response.status_code} {response.text}")
-            except Exception as e:
-                st.error(f"Exception: {str(e)}")
+                    st.error(f"STT Error: {stt_resp.status_code} {stt_resp.text}")
+                    st.stop()
+            time.sleep(5)  # Wait for 5 seconds
+            placeholder.markdown(f"**STT:** Transcribed ‘{transcription}’")
+            
+            # ----- Step 2: Call the LLM endpoint using the transcription as input -----
+            with st.spinner("Processing LLM..."):
+                llm_payload = {"message": transcription}
+                llm_resp = requests.post(f"{BASE_URL}/llm", json=llm_payload)
+                if llm_resp.ok:
+                    llm_data = llm_resp.json()
+                    llm_reply = llm_data.get("reply", "No reply")
+                else:
+                    st.error(f"LLM Error: {llm_resp.status_code} {llm_resp.text}")
+                    st.stop()
+            time.sleep(10)  # Wait additional 10 seconds (total 15 sec from start)
+            placeholder.markdown(f"**LLM:** Bot says ‘{llm_reply}’")
+            
+            # ----- Step 3: Call the TTS endpoint with a fixed text ("Tell me more!") -----
+            with st.spinner("Processing TTS..."):
+                tts_payload = {"text": "Tell me more!"}
+                tts_resp = requests.post(f"{BASE_URL}/tts", json=tts_payload)
+                if tts_resp.ok:
+                    tts_data = tts_resp.json()
+                    tts_audio_b64 = tts_data.get("audio", "")
+                else:
+                    st.error(f"TTS Error: {tts_resp.status_code} {tts_resp.text}")
+                    st.stop()
+            time.sleep(10)  # Wait additional 10 seconds (total 25 sec from start)
+            placeholder.markdown(f"**TTS:** Bot says ‘Tell me more!’")
+            
+            if tts_audio_b64:
+                audio_bytes = base64.b64decode(tts_audio_b64)
+                st.audio(audio_bytes, format="audio/wav")
+            else:
+                st.info("No TTS audio returned.")
 
 st.markdown("---")
 st.markdown("## Text Test")
 text_test_type = st.radio("Select Text Test Type", ("Contact Test", "Spam Test", "Unknown Test"))
-text_number = st.text_input("Enter Sender Number for Text Test (e.g., +1-555-901-2345 for contact or other for unsaved):", key="text_number")
+text_number = st.text_input("Enter Sender Number for Text Test (e.g., +1-555-901-2345 for contact):", key="text_number")
 default_text = "Hi" if text_test_type == "Contact Test" else "Free trip!" if text_test_type == "Spam Test" else "Fix your car!"
 text_message = st.text_input("Enter Text Message", value=default_text, key="text_message")