Update app.py

app.py

@@ -107,7 +107,7 @@ def detect_and_align_faces(image_path):
     return aligned_face
 
 # Add person to database
-def add_person(name, image_path, instagram_handle):
+def add_person(name, image_path, instagram_handle, email=None):
     try:
         encoding = load_and_encode(image_path)
         if not encoding:
@@ -117,13 +117,17 @@ def add_person(name, image_path, instagram_handle):
         encoding = encoding[0].tolist()
 
         # Save data to Firebase Realtime Database
-        ref.child(name).set({
+        person_data = {
             "encoding": encoding,
             "info": {
                 "instagram_handle": instagram_handle,
                 "instagram_link": f"https://www.instagram.com/{instagram_handle}/"
             }
-        })
+        }
+        if email:
+            person_data["info"]["email"] = email
+
+        ref.child(name).set(person_data)
 
         return f"Success: {name} added to the database!"
     except Exception as e:
@@ -143,15 +147,16 @@ def recognize_face(image_path):
         for name, data in ref.get().items():
             known_encoding = np.array(data["encoding"])
             if face_recognition.compare_faces([known_encoding], unknown_encoding[0])[0]:
-                matches.append((name, data["info"]))
+                info = data["info"]
+                email = info.get("email", "Email not provided")
+                matches.append((name, info["instagram_handle"], email))
 
         if matches:
             results = []
-            for name, info in matches:
-                insta_handle = info["instagram_handle"]
-                insta_link = info["instagram_link"]
+            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}")
+                results.append(f"- It's a picture of {name}! Insta handle: {insta_link_html}, Email: {email}")
             return "\n".join(results)
         else:
             return "Face not found in the database."
@@ -209,12 +214,13 @@ def add_person_ui():
     st.title("Add Person")
     name = st.text_input("Enter Name", help="Enter the name of the person")
     image_path = st.file_uploader("Upload Image", help="Upload an image containing the person's face")
+    email = st.text_input("Enter Email (Optional)", help="Enter the person's email address (optional)")
     instagram_handle = st.text_input("Enter Instagram Handle", help="Enter the person's Instagram handle")
     if st.button("Add Person"):
         if not name or not image_path or not instagram_handle:
-            st.error("Please fill all the fields.")
+            st.error("Please fill all the required fields.")
         else:
-            result = add_person(name, image_path, instagram_handle)
+            result = add_person(name, image_path, instagram_handle, email)
             st.success(result)
 
 # Streamlit interface for recognizing face
@@ -225,6 +231,13 @@ def recognize_face_ui():
         result = recognize_face(image_path)
         st.write(result, unsafe_allow_html=True)
 
+        if "It's a picture of" in result:
+            # Extract email from the result
+            email_start = result.find("Insta handle: ") + len("Insta handle: ")
+            email_end = result.find("</font></a>", email_start)
+            email = result[email_start:email_end]
+            st.write(f"Email: {email}")
+
 # Streamlit interface for recognizing face with optimal ID
 def recognize_face_optimal_ui():
     st.title("Recognize Face (Optimal)")
@@ -234,7 +247,12 @@ def recognize_face_optimal_ui():
         if "not found" in result.lower():  # Check if "not found" is in the result message
             st.error(result)
         else:
-            st.write(result, unsafe_allow_html=True)
+            if "email" in result:
+                name, similarity_score, email = result.split(",")
+                st.write(f"Best match: {name} with a similarity score of {1 - float(similarity_score):.2%}. Email: {email}")
+            else:
+                name, similarity_score = result.split(",")
+                st.write(f"Best match: {name} with a similarity score of {1 - float(similarity_score):.2%}.")
         
 # Streamlit interface for deleting a person
 def delete_person_ui():