Update app.py

app.py

@@ -1,106 +1,97 @@
-import streamlit as st
-from streamlit_webrtc import webrtc_streamer, VideoTransformerBase
+import os
 import cv2
 import av
-import numpy as np
-import os
+import streamlit as st
+from streamlit_webrtc import webrtc_streamer, VideoTransformerBase
 from datetime import datetime
 
-# Directory to store registered faces
+# Constants
 KNOWN_FACES_DIR = "known_faces"
 os.makedirs(KNOWN_FACES_DIR, exist_ok=True)
 
-# Page configuration
-st.set_page_config(page_title="Face Detection App", layout="wide")
+# Global State
+if "activity_log" not in st.session_state:
+    st.session_state.activity_log = []
 
-# Title and Description
+# Streamlit App Title
 st.title("Face Detection and Registration App")
-st.write("""
-This app allows you to:
-1. Detect faces in real-time using your webcam.
-2. Register new faces with names.
-3. View the list of registered faces.
-4. Show recent activity logs.
-""")
-
-# Initialize log list
-activity_log = []
-
-# Define Video Processor
-class VideoProcessor(VideoTransformerBase):
-    def __init__(self):
+
+st.sidebar.header("Navigation")
+page = st.sidebar.radio("Choose an option:", ["Real-time Face Detection", "Register New Face", "View Registered Faces", "Recent Activity"])
+
+# Video Processor Class
+class FaceDetectionProcessor(VideoTransformerBase):
+    def _init_(self):
         self.face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + "haarcascade_frontalface_default.xml")
-        self.registering_face = False
+        self.register_mode = False
         self.face_name = None
 
-    def set_registering_mode(self, name):
-        self.registering_face = True
+    def set_register_mode(self, name):
+        self.register_mode = True
         self.face_name = name
 
-    def stop_registering(self):
-        self.registering_face = False
+    def stop_register_mode(self):
+        self.register_mode = False
         self.face_name = None
 
     def recv(self, frame):
         img = frame.to_ndarray(format="bgr24")
         gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-
-        # Detect faces
         faces = self.face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(50, 50))
 
-        # Draw rectangles and save face images if in registering mode
         for (x, y, w, h) in faces:
+            # Draw rectangle
             cv2.rectangle(img, (x, y), (x + w, y + h), (0, 255, 0), 2)
-            if self.registering_face:
-                face_img = img[y:y + h, x:x + w]
-                face_path = os.path.join(KNOWN_FACES_DIR, f"{self.face_name}_{datetime.now().strftime('%Y%m%d%H%M%S')}.jpg")
-                cv2.imwrite(face_path, face_img)
-                self.registering_face = False
-                st.session_state.activity_log.append(f"Face registered: {self.face_name} at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
 
-        return av.VideoFrame.from_ndarray(img, format="bgr24")
+            if self.register_mode and self.face_name:
+                # Save the cropped face image
+                face_img = img[y:y+h, x:x+w]
+                face_filename = f"{self.face_name}{datetime.now().strftime('%Y%m%d%H%M%S')}.jpg"
+                face_path = os.path.join(KNOWN_FACES_DIR, face_filename)
+                cv2.imwrite(face_path, face_img)
 
+                st.session_state.activity_log.append(
+                    f"Registered face: {self.face_name} at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+                )
+                self.stop_register_mode()  # Stop registering after saving
+                st.success(f"Face registered successfully as {self.face_name}!")
 
-# Sidebar for navigation
-st.sidebar.title("Options")
-app_mode = st.sidebar.radio("Choose the mode:", ["Real-time Face Detection", "Register a New Face", "View Registered Faces", "Recent Activity"])
+        return av.VideoFrame.from_ndarray(img, format="bgr24")
 
-# Global activity log state
-if "activity_log" not in st.session_state:
-    st.session_state.activity_log = []
 
-# Handle the selected mode
-if app_mode == "Real-time Face Detection":
+# Pages
+if page == "Real-time Face Detection":
     st.header("Real-time Face Detection")
-    webrtc_streamer(key="face_detection", video_processor_factory=VideoProcessor)
+    webrtc_streamer(key="face_detection", video_processor_factory=FaceDetectionProcessor)
+
+elif page == "Register New Face":
+    st.header("Register New Face")
+    name = st.text_input("Enter a name for the face:")
 
-elif app_mode == "Register a New Face":
-    st.header("Register a New Face")
-    name = st.text_input("Enter the name for the new face:")
     if st.button("Start Registration"):
-        if not name.strip():
-            st.error("Name cannot be empty!")
+        if name.strip():
+            st.info(f"Looking for face to register as '{name}'...")
+            processor = webrtc_streamer(key="register_face", video_processor_factory=FaceDetectionProcessor)
+            if processor and processor.video_processor:
+                processor.video_processor.set_register_mode(name)
         else:
-            processor = webrtc_streamer(key="register_face", video_processor_factory=VideoProcessor)
-            if processor.video_processor:
-                processor.video_processor.set_registering_mode(name)
-                st.success(f"Registering face for '{name}'. Please look into the camera.")
+            st.error("Please enter a valid name!")
 
-elif app_mode == "View Registered Faces":
+elif page == "View Registered Faces":
     st.header("Registered Faces")
     faces = os.listdir(KNOWN_FACES_DIR)
+
     if faces:
-        st.write(f"**Total Faces Registered:** {len(faces)}")
-        for face in faces:
-            st.write(face)
+        st.write(f"*Total Registered Faces: {len(faces)}*")
+        for face_file in faces:
+            st.write(face_file)
     else:
-        st.write("No faces have been registered yet.")
+        st.write("No registered faces found.")
 
-elif app_mode == "Recent Activity":
+elif page == "Recent Activity":
     st.header("Recent Activity Log")
     if st.session_state.activity_log:
         for log in reversed(st.session_state.activity_log):
             st.write(log)
     else:
-        st.write("No activity logged yet.")
-
+        st.write("No recent activity to display.")