Update app.py

app.py

@@ -71,25 +71,28 @@ if option == "Home":
 
 elif option == "Register New Face":
     person_name = st.text_input("Enter the person's name")
-    capture_mode = st.radio("Select input method", ["Use Camera", "Upload Image"])
+    capture_mode = st.radio("Select input method", ["Use Camera", "Upload Image(s)"])
     
     if person_name and st.button("Register Face"):
         person_dir = os.path.join(KNOWN_FACES_DIR, person_name)
         os.makedirs(person_dir, exist_ok=True)
+        
         if capture_mode == "Use Camera":
             st.warning("Switch to a device with a camera for this option.")
-        elif capture_mode == "Upload Image":
-            uploaded_file = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
-            if uploaded_file:
-                img = cv2.imdecode(np.frombuffer(uploaded_file.read(), np.uint8), cv2.IMREAD_COLOR)
-                gray, faces = detect_faces(img)
-                for (x, y, w, h) in faces:
-                    face_img = gray[y:y+h, x:x+w]
-                    resized_img = cv2.resize(face_img, IMG_SIZE)
-                    timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
-                    filepath = os.path.join(person_dir, f"{timestamp}.jpg")
-                    cv2.imwrite(filepath, resized_img)
-                st.success("Face registered successfully!")
+        
+        elif capture_mode == "Upload Image(s)":
+            uploaded_files = st.file_uploader("Upload images", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
+            if uploaded_files:
+                for uploaded_file in uploaded_files:
+                    img = cv2.imdecode(np.frombuffer(uploaded_file.read(), np.uint8), cv2.IMREAD_COLOR)
+                    gray, faces = detect_faces(img)
+                    for (x, y, w, h) in faces:
+                        face_img = gray[y:y+h, x:x+w]
+                        resized_img = cv2.resize(face_img, IMG_SIZE)
+                        timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
+                        filepath = os.path.join(person_dir, f"{timestamp}.jpg")
+                        cv2.imwrite(filepath, resized_img)
+                st.success(f"Faces registered successfully for {person_name}!")
         label_map = train_recognizer()
 
 elif option == "Recognize Faces":