Update app.py

app.py

@@ -2,7 +2,6 @@ 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"
@@ -40,7 +39,6 @@ st.markdown("## Call Test")
 
 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")
 
@@ -56,8 +54,7 @@ if st.button("Test Call"):
             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**")
+            st.markdown(f"**Spam call from {call_number} – Bot replying**")
             
             # Ensure an audio file is uploaded.
             if audio_file is None:
@@ -75,10 +72,9 @@ if st.button("Test Call"):
                 else:
                     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}’")
+            st.markdown(f"**STT:** Transcribed ‘{transcription}’")
             
-            # ----- Step 2: Call the LLM endpoint using the transcription as input -----
+            # ----- Step 2: Call the LLM endpoint using the STT transcription -----
             with st.spinner("Processing LLM..."):
                 llm_payload = {"message": transcription}
                 llm_resp = requests.post(f"{BASE_URL}/llm", json=llm_payload)
@@ -88,12 +84,11 @@ if st.button("Test Call"):
                 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}’")
+            st.markdown(f"**LLM:** Bot says ‘{llm_reply}’")
             
-            # ----- Step 3: Call the TTS endpoint with a fixed text ("Tell me more!") -----
+            # ----- Step 3: Call the TTS endpoint with the LLM reply -----
             with st.spinner("Processing TTS..."):
-                tts_payload = {"text": "Tell me more!"}
+                tts_payload = {"text": llm_reply}
                 tts_resp = requests.post(f"{BASE_URL}/tts", json=tts_payload)
                 if tts_resp.ok:
                     tts_data = tts_resp.json()
@@ -101,8 +96,7 @@ if st.button("Test Call"):
                 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!’")
+            st.markdown("**TTS:** Bot audio generated from LLM response")
             
             if tts_audio_b64:
                 audio_bytes = base64.b64decode(tts_audio_b64)