Update app.py

app.py

@@ -61,6 +61,11 @@ def insert_student(name, roll_number, image_path):
     except sqlite3.IntegrityError:
         st.warning("Roll number already exists!")
 
+def get_all_students():
+    with sqlite3.connect(DB_FILE) as conn:
+        cursor = conn.execute("SELECT * FROM students")
+        return cursor.fetchall()
+
 # Face and Emotion Detection
 def detect_faces_and_emotions(image):
     gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
@@ -69,11 +74,15 @@ def detect_faces_and_emotions(image):
 
     for (x, y, w, h) in faces:
         face = gray_image[y:y+h, x:x+w]
-        resized_face = cv2.resize(face, (48, 48))
-        normalized_face = resized_face / 255.0
-        reshaped_face = np.reshape(normalized_face, (1, 48, 48, 1))
+        resized_face = cv2.resize(face, (48, 48))  # Resize face to 48x48
+        # Convert grayscale to RGB by duplicating the channel
+        rgb_face = cv2.cvtColor(resized_face, cv2.COLOR_GRAY2RGB)  # Convert grayscale to RGB
+        normalized_face = rgb_face / 255.0  # Normalize pixel values to [0, 1]
+        reshaped_face = np.reshape(normalized_face, (1, 48, 48, 3))  # Ensure the input shape is (1, 48, 48, 3)
+        
+        # Predict the emotion
         emotion_prediction = emotion_model.predict(reshaped_face)
-        emotion_label = np.argmax(emotion_prediction)
+        emotion_label = np.argmax(emotion_prediction)  # Get the index of the highest probability
         return EMOTION_LABELS[emotion_label]
     return None
 
@@ -81,7 +90,7 @@ def detect_faces_and_emotions(image):
 st.title("Student Registration and Emotion Detection")
 create_table()
 
-menu = ["Register Student", "Face Recognition and Emotion Detection"]
+menu = ["Register Student", "Face Recognition and Emotion Detection", "View Attendance"]
 choice = st.sidebar.selectbox("Menu", menu)
 
 if choice == "Register Student":
@@ -91,20 +100,38 @@ if choice == "Register Student":
         name = st.text_input("Name")
         roll_number = st.text_input("Roll Number")
         image_file = st.file_uploader("Upload Image", type=["jpg", "jpeg", "png"])
+        use_webcam = st.checkbox("Use Webcam for Face Registration")
         submitted = st.form_submit_button("Register")
 
         if submitted:
-            if name and roll_number and image_file:
-                try:
-                    img = Image.open(image_file)
-                    img_path = os.path.join(KNOWN_FACES_DIR, f"{roll_number}.png")
-                    img.save(img_path)
-                    insert_student(name, roll_number, img_path)
-                    st.success("Student Registered Successfully!")
-                except Exception as e:
-                    st.error(f"Error: {e}")
+            if name and roll_number:
+                if use_webcam:
+                    st.info("Use the camera input widget to capture an image.")
+                    camera_image = st.camera_input("Capture Image")
+                    if camera_image:
+                        try:
+                            img = Image.open(camera_image)
+                            img = img.convert("RGB")
+                            img_path = os.path.join(KNOWN_FACES_DIR, f"{roll_number}.png")
+                            img.save(img_path)
+                            insert_student(name, roll_number, img_path)
+                            st.success(f"Student {name} Registered Successfully!")
+                        except Exception as e:
+                            st.error(f"Error: {e}")
+                elif image_file:
+                    try:
+                        img = Image.open(image_file)
+                        img = img.convert("RGB")
+                        img_path = os.path.join(KNOWN_FACES_DIR, f"{roll_number}.png")
+                        img.save(img_path)
+                        insert_student(name, roll_number, img_path)
+                        st.success(f"Student {name} Registered Successfully!")
+                    except Exception as e:
+                        st.error(f"Error: {e}")
+                else:
+                    st.warning("Please upload an image or use the webcam to register the face.")
             else:
-                st.warning("Please fill in all fields and upload an image.")
+                st.warning("Please fill in all fields.")
 
 elif choice == "Face Recognition and Emotion Detection":
     st.subheader("Recognize Faces and Detect Emotions")
@@ -139,3 +166,13 @@ elif choice == "Face Recognition and Emotion Detection":
                     st.warning("No face detected.")
             except Exception as e:
                 st.error(f"Error: {e}")
+
+elif choice == "View Attendance":
+    st.subheader("View Registered Students (Attendance)")
+
+    students = get_all_students()
+    if students:
+        for student in students:
+            st.write(f"Name: {student[1]}, Roll Number: {student[2]}")
+    else:
+        st.warning("No students registered.")