bug fixed images

pages/{3_Face Encodings.py → 3_Training teh Model.py}

@@ -28,7 +28,8 @@ def get_database():
     return database
 
 def step3_page():
-    st.header("Feature encoding")
+    st.header("Training the Model")
+    st.subheader("Face encoding")
     st.markdown(
         """
     ### What is Face Encoding?
@@ -54,26 +55,29 @@ def step3_page():
     )
 
     st.info(
-        "Now it's your turn to create face encodings to each of the faces in the known-face database that you have created in the previous step!"
+        "Now it's your turn to train your model to create face encodings to each of the faces in the known-face database that you have created in the previous step!"
     )
 
     img_dir = os.path.join(".sessions", get_login()["username"], "known_faces")
-    face_encodings_dir = os.path.join(".sessions", get_login()["username"], "face_encodings")
-    os.makedirs(face_encodings_dir, exist_ok=True)
     images = os.listdir(img_dir)
     if len(images) > 0:
-        st.subheader("Let's see your saved faces in your known-face database.")
+        st.subheader("First, 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("_")[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.")
-    # face_encodings_dict = {}
-    if st.button("Create Face Encodings"):
-        database = get_database()
-        my_bar = st.progress(0, text="Generating face encodings...")
+    st.subheader("Now it's your turn to train the model! Click on the button below to train the model with your data to generate face encodings!")
+    if st.button("Train Model"):
+        my_bar = st.progress(0, text="Training....")
         if len(images) > 0:
+            database = get_database()
+            for i in range(1000):
+                my_bar.progress(i / 10, text="Training....")
+            my_bar.progress(100, text="Successfully Trained!")
+            st.success("Model trained successfully!")
+            st.info("Now, lets generate face encodings for each face in known-face database using the model you just trained!")
+            my_bar = st.progress(0, text="Generating face encodings...")
             for i, img in enumerate(images):
                 face_image = face_recognition.load_image_file(os.path.join(img_dir, img))
                 my_face_encoding = face_recognition.face_encodings(face_image)
@@ -91,14 +95,10 @@ def step3_page():
                         pkl.dump(database, f)
 
                 my_bar.progress(int((i + 1) / len(images) * 100), text="Generating face encodings...")
-                # 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!")
-            st.write(database)
-            # with open(os.path.join(".sessions", get_login()["username"], "face_encodings.json"), "w") as write_file:
-            #     json.dump(face_encodings_dict, write_file, indent=4)
-            complete_level(LEVEL)
+            if st.button("Complete"):
+                complete_level(LEVEL)
         else:
             my_bar.empty()
             st.error("You have not taken any images yet! Do the previous steps first!")