Update app.py

app.py

@@ -118,18 +118,18 @@ def detect_and_align_faces(image):
     return aligned_faces
 
 # Add person to database
-def add_person(name, image_path, instagram_handle, email=None):
+def add_person(name, image_paths, instagram_handle, email=None):
     try:
-        encoding = load_and_encode(image_path)
-        if not encoding:
-            return "No face found in the provided image."
-
-        # Convert NumPy arrays to lists for JSON serialization
-        encoding = encoding[0].tolist()
+        encodings = []
+        for image_path in image_paths:
+            encoding = load_and_encode(image_path)
+            if not encoding:
+                return f"No face found in the image: {image_path}"
+            encodings.append(encoding[0].tolist())
 
         # Save data to Firebase Realtime Database
         person_data = {
-            "encoding": encoding,
+            "encodings": encodings,
             "info": {
                 "instagram_handle": instagram_handle,
                 "instagram_link": f"https://www.instagram.com/{instagram_handle}/"
@@ -245,14 +245,14 @@ def send_feedback(feedback_data):
 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")
+    image_paths = st.file_uploader("Upload Image(s)", accept_multiple_files=True, help="Upload 1 to 3 images 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:
+        if not name or not image_paths or not instagram_handle:
             st.error("Please fill all the required fields.")
         else:
-            result = add_person(name, image_path, instagram_handle, email)
+            result = add_person(name, image_paths, instagram_handle, email)
             st.success(result)
 
 # Streamlit interface for recognizing face