bug fixed images - path

pages/2_Face Detection and Creating Database.py

@@ -58,6 +58,8 @@ def step2_page():
     )
     st.subheader("Who is taking the picture?")
     face_name = st.text_input('Specify name to save it in the known-face database', "This is a placeholder", key="name")
+    ID = st.text_input('Image ID',
+                              "This is a placeholder", key="int")
     input_type = st.radio("Select the Input Type", ["Image", "Camera"])
 
     if input_type == "Camera":
@@ -95,11 +97,15 @@ def step2_page():
                     col.image(pil_image, use_column_width=True)
                     face_name = st.text_input('Who is this?',
                                               "This is a placeholder", key="text_" + str(i))
+                    ID = st.text_input('Image ID',
+                                       "This is a placeholder", key="int_" + str(i))
                     if st.button("Save", key="button_" + str(i)):
                         if face_name == "This is a placeholder":
                             st.error("Please provide a name for the face!")
+                        elif ID == "This is a placeholder":
+                            st.error("Please provide an ID!")
                         else:
-                            img_name = str(uuid.uuid4()) + f"_{face_name}_{i}" + ".jpg"
+                            img_name = f"{face_name}_{ID}" + ".jpg"
                             img_path = os.path.join(img_dir, img_name)
                             if os.path.exists(img_path):
                                 st.error("Face already added!")
@@ -117,12 +123,14 @@ def step2_page():
             face_image = image[top:bottom, left:right]
             pil_image = Image.fromarray(face_image)
             st.image(pil_image, use_column_width="auto")
-            st.write(face_name)
+            st.write(f"Name: {face_name}")
             if st.button("Save", key="button"):
                 if face_name == "This is a placeholder":
                     st.error("Please provide a name for the face!")
+                elif ID == "This is a placeholder":
+                    st.error("Please provide an ID!")
                 else:
-                    img_name = str(uuid.uuid4()) + f"_{face_name}_0" + ".jpg"
+                    img_name = f"{face_name}_{ID}" + ".jpg"
                     img_path = os.path.join(img_dir, img_name)
                     if os.path.exists(img_path):
                         st.error("Face already added!")
@@ -138,9 +146,9 @@ def step2_page():
     else:
         cols = st.columns(len(images))
         for i, img in enumerate(images):
-            face_name = img.split("_")[1]
+            face_name = img.split("_")[0]
             cols[i].image(os.path.join(img_dir, img), use_column_width="auto")
-            cols[i].write(face_name)
+            cols[i].write(f"Name: {face_name}")
 
     if st.button("Clear All"):
         for img in images:

pages/3_Face Encodings.py

@@ -52,7 +52,7 @@ def step3_page():
         st.subheader("Let's see your saved faces in your known-face database.")
         cols = st.columns(len(images))
         for i, img in enumerate(images):
-            face_name = img.split("_")[1]
+            face_name = img.split("_")[0]
             cols[i].image(os.path.join(img_dir, img), use_column_width=True)
             cols[i].write(face_name)
     st.subheader("Lets create face encodings for the known-faces.")
@@ -64,7 +64,7 @@ def step3_page():
                 face_image = face_recognition.load_image_file(os.path.join(img_dir, img))
                 my_face_encoding = face_recognition.face_encodings(face_image)
                 my_bar.progress(int((i + 1) / len(images) * 100), text="Generating face encodings...")
-                np.save(os.path.join(face_encodings_dir, img.split("_")[1]+".npy"), my_face_encoding)
+                np.save(os.path.join(face_encodings_dir, img.split("_")[0]+".npy"), my_face_encoding)
                 # face_encodings_dict[img.split("_")[1]] = my_face_encoding.tolist()
             my_bar.progress(100, text="Successfully encoded all the known faces!")
             st.success("Face encoding completed successfully!")