Update app.py

app.py

@@ -64,20 +64,25 @@ def train_recognizer():
     faces = []
     labels = []
     for name in os.listdir(KNOWN_FACES_DIR):
-        for filename in os.listdir(os.path.join(KNOWN_FACES_DIR, name)):
-            filepath = os.path.join(KNOWN_FACES_DIR, name, filename)
+        person_dir = os.path.join(KNOWN_FACES_DIR, name)
+        if not os.path.isdir(person_dir):  # Skip files, process only directories
+            continue
+        for filename in os.listdir(person_dir):
+            filepath = os.path.join(person_dir, filename)
+            if not filepath.lower().endswith(('.jpg', '.jpeg', '.png')):  # Ensure valid image formats
+                continue
             image = cv2.imread(filepath, cv2.IMREAD_GRAYSCALE)
             if image is not None:
                 faces.append(image)
                 labels.append(name)
+    if not faces or not labels:
+        raise ValueError("No valid training data found in the known faces directory.")
     label_ids = {name: idx for idx, name in enumerate(set(labels))}
     label_ids_rev = {idx: name for name, idx in label_ids.items()}
     labels = [label_ids[label] for label in labels]
-    if faces:
-        face_recognizer.train(faces, np.array(labels))
+    face_recognizer.train(faces, np.array(labels))
     return label_ids_rev
 
-label_ids_rev = train_recognizer()
 
 # Sidebar options
 sidebar_choice = st.sidebar.selectbox("Choose an option", ["Emotion Detection", "Register New Face", "View Recent Activity"])