Update app.py

app.py

@@ -146,7 +146,7 @@ def add_person(name, image_path, instagram_handle, email=None):
     except Exception as e:
         return f"Failed to add person: {str(e)}"
 
-# Update recognize_face function to handle multiple face encodings
+# Update recognize_face function to include email in the result
 def recognize_face(image_path):
     if not image_path:
         return "Please upload an image."
@@ -176,7 +176,8 @@ def recognize_face(image_path):
             for name, insta_handle, email in matches:
                 insta_link = f"https://www.instagram.com/{insta_handle}/"
                 insta_link_html = f'<a href="{insta_link}" target="_blank"><font color="red">{insta_handle}</font></a>'
-                results.append(f"- It's a picture of {name}! Insta handle: {insta_link_html}, Email: {email}")
+                email_link_html = f'<a href="#" onclick="set_recipient_email(\'{email}\')">{email}</a>'
+                results.append(f"- It's a picture of {name}! Insta handle: {insta_link_html}, Email: {email_link_html}")
             log_action(st.session_state.auth_state["user"].email, "Recognized face")
             return "\n".join(results)
         else:
@@ -266,17 +267,13 @@ def recognize_face_ui():
         result = recognize_face(image_path)
         st.write(result, unsafe_allow_html=True)
 
-        # Extract email from the result
-        if "Email" in result:
-            email_start_index = result.find("Email: ") + len("Email: ")
-            email_end_index = result.find("</font>", email_start_index)
-            email = result[email_start_index:email_end_index]
+        # Add JavaScript to set the recipient's email when the email is clicked
+        st.write('<script>')
+        st.write('function set_recipient_email(email) {')
+        st.write('document.getElementById("receiver_email").value = email;')
+        st.write('}')
+        st.write('</script>')
 
-            # Add a link to the email
-            message_ui_url = f"http://localhost:8501/?p=Insta%27s%20EYE&p=Insta%27s%20EYE&p=Insta%27s%20EYE&em={email}"
-            st.markdown(f"Click [here](<{message_ui_url}>) to message {email}")
-
-# Streamlit interface for recognizing face optimal
 def recognize_face_optimal_ui():
     st.title("🔍 Recognize Face (Optimal)")
     image_path = st.file_uploader("Upload Image", help="Upload an image for optimal face recognition")
@@ -343,12 +340,12 @@ def get_messages(user_email):
         return None
 
 # Streamlit interface for messaging
-def messaging_ui(recipient_email):
+def messaging_ui():
     st.title("💬 Messaging")
 
     if st.session_state.auth_state["signed_in"]:
         sender_email = st.session_state.auth_state["user"].email
-        receiver_email = st.text_input("Receiver's Email", value=recipient_email, help="Enter the receiver's email address")
+        receiver_email = st.text_input("Receiver's Email", help="Enter the receiver's email address", id="receiver_email")
         message_content = st.text_area("Message Content")
 
         if st.button("Send Message"):
@@ -528,7 +525,7 @@ def display_tour_steps(steps):
 def main():
     st.sidebar.title("Options")
     option = st.sidebar.radio("Select Option", ["Add Person", "Recognize Face", "Delete Person", "Recognize Face (Optimal)", "Tour Guide", "Feedback", "Messaging", "History", "Delete User"])
-
+ 
     if option == "Add Person":
         add_person_ui()
     elif option == "Recognize Face":
@@ -542,20 +539,11 @@ def main():
     elif option == "Feedback":
         feedback_ui()
     elif option == "Messaging":
-        if "em" in st.session_state and st.session_state["em"]:
-            messaging_ui(st.session_state["em"])
-            st.session_state["em"] = None
-        else:
-            messaging_ui()
+        messaging_ui()
     elif option == "History":
         display_history(st.session_state.auth_state["user"].email)
     elif option == "Delete User":
         delete_user_ui()
-
-    # Handle the case when the user clicks on the email link
-    if "em" in st.session_state and st.session_state["em"]:
-        messaging_ui(st.session_state["em"])
-        st.session_state["em"] = None
         
 # Run the tour guide
 if __name__ == "__main__":