Update app.py

app.py

@@ -398,6 +398,10 @@ class AttendanceSystem:
                 buffer_key = f"{x}_{y}_{w}_{h}"
                 current_time = time.time()
                 
+                # Create a buffer key for this face location
+                buffer_key = f"{x}_{y}_{w}_{h}"
+                current_time = time.time()
+                
                 if self.known_face_embeddings:
                     # Find best match using euclidean distance
                     match_index, match_distance = self._find_best_match(embedding_array)
@@ -410,6 +414,15 @@ class AttendanceSystem:
                         worker_name = self.known_face_names[match_index]
                         color = (0, 255, 0)  # Green for recognized
                         
+                        # Use buffer for consistent detections
+                        if buffer_key not in self.face_recognition_buffer:
+                            self.face_recognition_buffer[buffer_key] = {'count': 1, 'last_time': current_time, 'worker_id': worker_id}
+                        else:
+                            self.face_recognition_buffer[buffer_key]['count'] += 1
+                            self.face_recognition_buffer[buffer_key]['last_time'] = current_time
+                        
+                        color = (0, 255, 0)  # Green for recognized
+                        
                         # Use buffer for consistent detections
                         if buffer_key not in self.face_recognition_buffer:
                             self.face_recognition_buffer[buffer_key] = {'count': 1, 'last_time': current_time, 'worker_id': worker_id}
@@ -434,13 +447,10 @@ class AttendanceSystem:
                             color = (0, 165, 255)  # Orange for potential new worker
                             print(f"    -> Attempting auto-registration...")
                             new_worker = self._register_worker_auto(face_image, embedding)
-                            if new_worker:
                                 worker_id, worker_name = new_worker[0], new_worker[1]
                                 color = (0, 255, 0)  # Change to green after successful registration
                                 if self.mark_attendance(worker_id, worker_name):
                                     self.last_recognition_time[worker_id] = time.time()
-                            else:
-                                print("    -> Auto-registration skipped (duplicate or session limit)")
                         else:
                             print("    -> Distance too close to existing faces, skipping auto-registration.")
                 else:
@@ -448,13 +458,10 @@ class AttendanceSystem:
                     color = (0, 165, 255)  # Orange for new worker
                     print("    -> No known faces in database. Auto-registering...")
                     new_worker = self._register_worker_auto(face_image, embedding)
-                    if new_worker:
                         worker_id, worker_name = new_worker[0], new_worker[1]
                         color = (0, 255, 0)  # Change to green after successful registration
                         if self.mark_attendance(worker_id, worker_name):
                             self.last_recognition_time[worker_id] = time.time()
-                    else:
-                        print("    -> Auto-registration failed")
                 
                 # Clean old buffer entries
                 current_time = time.time()