bug fixed image read

pages/2_Face Detection and Creating Database.py

@@ -56,7 +56,8 @@ 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("Select an image to continue!")
+    st.info("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"])
 
     if input_type == "Camera":
@@ -66,36 +67,74 @@ def step2_page():
     if picture:
         image = face_recognition.load_image_file(picture)
         st.image(image)
+        my_bar = st.progress(0, text="Detecting faces in given images...")
+        for i in range(100):
+            my_bar.progress(i, text="Detecting faces in given images...")
 
         # Find all the faces in the image using the default HOG-based model.
         # This method is fairly accurate, but not as accurate as the CNN model and not GPU accelerated.
         # See also: find_faces_in_picture_cnn.py
         face_locations = face_recognition.face_locations(image)
-        st.write("Algorithm found {} face(s) in this photograph.".format(len(face_locations)))
-
-
-        cols = st.columns(len(face_locations))
-        for i in range(len(face_locations)):
-            col = cols[i]
-            face = face_locations[i]
+        my_bar.progress(100, text="Algorithm found {} face(s) in this photograph.".format(len(face_locations)))
+
+        if len(face_locations) <= 0:
+            st.error("No faces have been detected, please upload a valid/ clear photo!")
+        elif len(face_locations) > 1:
+            st.info("More than one face has been detected, please identify each face by providing a name.")
+            cols = st.columns(len(face_locations))
+            for i in range(len(face_locations)):
+                col = cols[i]
+                face = face_locations[i]
+                # display faces
+                with col:
+                    # Print the location of each face in this image
+                    top, right, bottom, left = face
+                    # You can access the actual face itself like this:
+                    face_image = image[top:bottom, left:right]
+                    pil_image = Image.fromarray(face_image)
+                    col.image(pil_image, use_column_width=True)
+                    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!")
+                        else:
+                            pil_image.save(img_path)
+                            st.success("Face added successfully!")
+
+        else:
+            st.info("Your face is identified successfully!")
+            face = face_locations[0]
             # display faces
-            with col:
-                st.header("Face {}".format(i))
-                # Print the location of each face in this image
-                top, right, bottom, left = face
-                # You can access the actual face itself like this:
-                face_image = image[top:bottom, left:right]
-                pil_image = Image.fromarray(face_image)
-                col.image(pil_image, use_column_width=True)
-                face_name = st.text_input('Specify name to save it in the known-face database', "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)
+            # Print the location of each face in this image
+            top, right, bottom, left = face
+            # You can access the actual face itself like this:
+            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)
+            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!")
+                else:
                     pil_image.save(img_path)
                     st.success("Face added successfully!")
 
+    st.write("Now lets see your known-faces database!")
     images = os.listdir(img_dir)
-    st.write(images)
+    if len(images) <= 0:
+        st.error("No faces have been added yet.")
+    else:
+        cols = st.columns(len(images))
+        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:
             os.remove(os.path.join(img_dir, img))

utils/levels.py

@@ -33,4 +33,4 @@ def complete_level(level):
         with open(f".sessions/{get_login()['username']}/level.txt", "w") as f:
             f.write(str(st.session_state['level']))
         st.balloons()
-        st.success(f'You have completed Level {level}! You can now move on to the next level.')
+        st.success(f'You have completed Level {level}! You can now move on to the next level.')