Update app.py

app.py

@@ -53,27 +53,37 @@ def detect_and_match_face(frame, known_faces, known_names):
 def main():
     st.title("Face Recognition App")
 
-    known_faces, known_names = load_known_faces()
+    # Login system
+    username = st.sidebar.text_input("Username")
+    password = st.sidebar.text_input("Password", type="password")
 
-    # OpenCV webcam setup
-    video_capture = cv2.VideoCapture(0)
+    if username == "your_username" and password == "your_password":
+        st.sidebar.success("Logged In")
+        st.write("Welcome, {}".format(username))
 
-    while True:
-        # Read each frame from the webcam
-        ret, frame = video_capture.read()
+        known_faces, known_names = load_known_faces()
 
-        # Detect and match faces in the frame
-        frame_with_faces = detect_and_match_face(frame, known_faces, known_names)
+        # OpenCV webcam setup
+        video_capture = cv2.VideoCapture(0)
 
-        # Display the frame with faces
-        st.image(frame_with_faces, channels="BGR", use_column_width=True)
+        while True:
+            # Read each frame from the webcam
+            ret, frame = video_capture.read()
 
-        # Break the loop if the user clicks the 'Stop' button
-        if st.button("Stop"):
-            break
+            # Detect and match faces in the frame
+            frame_with_faces = detect_and_match_face(frame, known_faces, known_names)
 
-    # Release the webcam
-    video_capture.release()
+            # Display the frame with faces
+            st.image(frame_with_faces, channels="BGR", use_column_width=True)
+
+            # Break the loop if the user clicks the 'Stop' button
+            if st.button("Stop"):
+                break
+
+        # Release the webcam
+        video_capture.release()
+    else:
+        st.sidebar.warning("Incorrect username or password")
 
 if __name__ == "__main__":
     main()