Update app.py

app.py

@@ -7,72 +7,101 @@ BASE_URL = "https://hammad712-spam.hf.space"
 
 st.title("AI Spam Blocker API Tester")
 
+st.markdown("## Save Contact")
+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")
+    
+if submitted:
+    if not contact_phone:
+        st.error("Please enter a contact phone number.")
+    else:
+        payload = [{
+            "phone": contact_phone,
+            "name": contact_name if contact_name else None,
+            "email": contact_email if contact_email else None
+        }]
+        with st.spinner("Saving contact..."):
+            try:
+                response = requests.post(f"{BASE_URL}/contacts", json=payload)
+                if response.ok:
+                    st.success("Contact saved successfully!")
+                else:
+                    st.error(f"Error: {response.status_code} {response.text}")
+            except Exception as e:
+                st.error(f"Exception: {str(e)}")
+
+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):")
+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, enter the fake audio message (as text) below.")
-    call_message = st.text_input("Enter Fake Audio Message", value="You won a prize!")
+    call_message = st.text_input("Enter Fake Audio Message", value="You won a prize!", key="call_message")
 else:
     st.markdown("For contact calls, the message is not used because a saved contact returns a ringing response.")
-    call_message = st.text_input("Enter Dummy Message (ignored for contact test)", value="dummy", disabled=True)
+    call_message = st.text_input("Enter Dummy Message (ignored for contact test)", value="dummy", key="dummy_message", disabled=True)
 
 if st.button("Test Call"):
     if not call_number:
         st.error("Please enter a caller number.")
     else:
-        # For voice calls, we need to simulate an audio file.
-        # We'll create a dummy file (using the call_message as its content)
+        # For voice calls, simulate an audio file using the message text.
         dummy_audio = io.BytesIO(call_message.encode("utf-8"))
         files = {"audio": ("audio.m4a", dummy_audio, "audio/m4a")}
         data = {"caller_number": call_number}
-        try:
-            response = requests.post(f"{BASE_URL}/process-call", data=data, files=files)
-            if response.ok:
-                st.success(f"Response: {response.json()['message']}")
-            else:
-                st.error(f"Error: {response.status_code} {response.text}")
-        except Exception as e:
-            st.error(f"Exception: {str(e)}")
+        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']}")
+                else:
+                    st.error(f"Error: {response.status_code} {response.text}")
+            except Exception as e:
+                st.error(f"Exception: {str(e)}")
 
 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_message = st.text_input("Enter Text Message", value="Hi" if text_test_type == "Contact Test" else "Free trip!" if text_test_type == "Spam Test" else "Fix your car!")
+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")
 
 if st.button("Test Text"):
     if not text_number or not text_message:
         st.error("Please enter both a number and a message.")
     else:
         payload = {"phone": text_number, "message": text_message}
+        with st.spinner("Processing text message..."):
+            try:
+                response = requests.post(f"{BASE_URL}/incoming-message", json=payload)
+                if response.ok:
+                    data = response.json()
+                    if "detail" in data:
+                        st.success(f"Response: {data['detail']}")
+                    else:
+                        st.success("Conversation Result:\n" + data.get("conversation_result", ""))
+                else:
+                    st.error(f"Error: {response.status_code} {response.text}")
+            except Exception as e:
+                st.error(f"Exception: {str(e)}")
+
+st.markdown("---")
+st.markdown("## Setup Test")
+if st.button("Start Setup"):
+    with st.spinner("Setting up call forwarding..."):
         try:
-            response = requests.post(f"{BASE_URL}/incoming-message", json=payload)
+            response = requests.post(f"{BASE_URL}/setup-call-forwarding")
             if response.ok:
-                data = response.json()
-                # For primary or spam texts, we show the detail directly.
-                if "detail" in data:
-                    st.success(f"Response: {data['detail']}")
-                else:
-                    st.success("Conversation Result:\n" + data.get("conversation_result", ""))
+                st.success(f"Response: {response.json()['message']}")
             else:
                 st.error(f"Error: {response.status_code} {response.text}")
         except Exception as e:
             st.error(f"Exception: {str(e)}")
 
-st.markdown("---")
-st.markdown("## Setup Test")
-if st.button("Start Setup"):
-    try:
-        response = requests.post(f"{BASE_URL}/setup-call-forwarding")
-        if response.ok:
-            st.success(f"Response: {response.json()['message']}")
-        else:
-            st.error(f"Error: {response.status_code} {response.text}")
-    except Exception as e:
-        st.error(f"Exception: {str(e)}")
-
 st.markdown("---")
 st.markdown("## Messages History")
 msg_number = st.text_input("Enter Caller Number to Retrieve Conversation History", key="msg_number")
@@ -80,12 +109,13 @@ if st.button("Get Messages"):
     if not msg_number:
         st.error("Please enter a caller number.")
     else:
-        try:
-            response = requests.get(f"{BASE_URL}/messages/{msg_number}")
-            if response.ok:
-                data = response.json()
-                st.write(data)
-            else:
-                st.error(f"Error: {response.status_code} {response.text}")
-        except Exception as e:
-            st.error(f"Exception: {str(e)}")
+        with st.spinner("Retrieving messages..."):
+            try:
+                response = requests.get(f"{BASE_URL}/messages/{msg_number}")
+                if response.ok:
+                    data = response.json()
+                    st.write(data)
+                else:
+                    st.error(f"Error: {response.status_code} {response.text}")
+            except Exception as e:
+                st.error(f"Exception: {str(e)}")