bug fixed images

pages/2_Face Detection and Creating Database.py

@@ -56,7 +56,7 @@ def step2_page():
         But remember, we should always ask for permission before taking someone's picture. We can use a smartphone or a digital camera to capture pictures, and it's important to take pictures of different people. This will help our application to have a good known-faces database!
         """
     )
-    st.info("Who is taking the picture?")
+    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")
     input_type = st.radio("Select the Input Type", ["Image", "Camera"])
 
@@ -96,13 +96,16 @@ def step2_page():
                     face_name = st.text_input('Who is this?',
                                               "This is a placeholder", key="text_" + str(i))
                     if st.button("Save", key="button_" + str(i)):
-                        img_name = str(uuid.uuid4()) + f"_{face_name}_{i}" + ".jpg"
-                        img_path = os.path.join(img_dir, img_name)
-                        if os.path.exists(img_path):
-                            st.error("Face already added!")
+                        if face_name == "This is a placeholder":
+                            st.error("Please provide a name for the face!")
                         else:
-                            pil_image.save(img_path)
-                            st.success("Face added successfully!")
+                            img_name = str(uuid.uuid4()) + f"_{face_name}_{i}" + ".jpg"
+                            img_path = os.path.join(img_dir, img_name)
+                            if os.path.exists(img_path):
+                                st.error("Face already added!")
+                            else:
+                                pil_image.save(img_path)
+                                st.success("Face added successfully!")
 
         else:
             st.info("Your face is identified successfully!")
@@ -116,30 +119,28 @@ def step2_page():
             st.image(pil_image, use_column_width="auto")
             st.write(face_name)
             if st.button("Save", key="button"):
-                img_name = str(uuid.uuid4()) + f"_{face_name}_0" + ".jpg"
-                img_path = os.path.join(img_dir, img_name)
-                if os.path.exists(img_path):
-                    st.error("Face already added!")
+                if face_name == "This is a placeholder":
+                    st.error("Please provide a name for the face!")
                 else:
-                    pil_image.save(img_path)
-                    st.success("Face added successfully!")
+                    img_name = str(uuid.uuid4()) + f"_{face_name}_0" + ".jpg"
+                    img_path = os.path.join(img_dir, img_name)
+                    if os.path.exists(img_path):
+                        st.error("Face already added!")
+                    else:
+                        pil_image.save(img_path)
+                        st.success("Face added successfully!")
 
 
-    st.info("Now lets see your known-faces database!")
+    st.subheader("Let's see your saved faces in your known-face database.")
     images = os.listdir(img_dir)
     if len(images) <= 0:
-        st.write("No faces have been added yet.")
+        st.error("No faces have been added yet.")
     else:
         cols = st.columns(len(images))
-        for i in range(len(images)):
-            col = cols[i]
-
-            # display faces
-            with col:
-                face = face_recognition.load_image_file(images[i])
-                face_name = images[i].split("_")[1]
-                st.image(face, use_column_width="auto")
-                st.write(face_name)
+        for i, img in enumerate(images):
+            face_name = img.split("_")[1]
+            cols[i].image(os.path.join(img_dir, img), use_column_width="auto")
+            cols[i].write(face_name)
 
     if st.button("Clear All"):
         for img in images: