Update app.py

app.py

@@ -14,7 +14,7 @@ import threading
 import time
 import queue
 
-# Optional: Suppress TensorFlow oneDNN warning
+# Suppress TensorFlow oneDNN warning
 os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
 
 class AttendanceSystem:
@@ -44,12 +44,10 @@ class AttendanceSystem:
         try:
             # Load face embeddings and worker data
             if os.path.exists("data/workers.pkl"):
-                with open("data/workers.pkl", "rb") as f:
-                    data = pickle.load(f)
-                    self.known_face_embeddings = data.get("embeddings", [])
-                    self.known_face_names = data.get("names", [])
-                    self.known_face_ids = data.get("ids", [])
-                    self.next_worker_id = data.get("next_id", 1)
+                with open("data/workers.pkl",往                self.known_face_embeddings = data.get("embeddings", [])
+                self.known_face_names = data.get("names", [])
+                self.known_face_ids = data.get("ids", [])
+                self.next_worker_id = data.get("next_id", 1)
             
             # Load attendance records
             if os.path.exists("data/attendance.json"):
@@ -728,15 +726,21 @@ def create_interface():
         
         # Video frame update
         def update_video_frame():
+            start_time = time.time()
             while True:
-                frame = attendance_system.get_current_frame()
-                if frame is not None:
-                    # Convert BGR to RGB
-                    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-                    yield frame
-                time.sleep(0.03)
+                current_time = time.time()
+                if current_time - start_time >= 0.03:  # Update every 0.03 seconds
+                    frame = attendance_system.get_current_frame()
+                    if frame is not None:
+                        # Convert BGR to RGB
+                        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+                        return frame
+                    start_time = current_time
         
-        demo.load(update_video_frame, None, video_output, every=0.03)
+        # Start the video frame update as a background thread
+        video_thread = threading.Thread(target=lambda: demo.queue()(update_video_frame)())
+        video_thread.daemon = True
+        video_thread.start()
     
     return demo