Update app.py

app.py

@@ -1,24 +1,72 @@
+# face attendance using streamlit
+
 import streamlit as st
 import cv2
+from deepface import DeepFace
+import threading
+import datetime
+import pymongo
+
+camera = cv2.VideoCapture(0)
+window = st.image([])
+
+myclient = pymongo.MongoClient("mongodb://localhost:27017/")
+mydb = myclient["camera"]
+mycol = mydb["attendance"]
+
+run = st.checkbox('Run')
+
+person1 = cv2.imread('./bill.jpeg')
+person2 = cv2.imread('./elon.jpeg')
+attendant = set()
+
+face_match = False
+
+def recogniser(name):
+    if name not in attendant:
+        print(name)
+        attendant.add(name)
+        add = {
+            "name": name,
+            "time": datetime.datetime.now().ctime()
+        }
+        mycol.insert_one(add)
+
+def check_face(frame):
+    global face_match
+    try:
+        if DeepFace.verify(frame, person1)['verified']:
+            face_match = True
+            recogniser('bill')
+
+        elif DeepFace.verify(frame, person2)['verified']:
+            face_match = True
+            recogniser('elon')
+        else:
+            face_match = False
+
+    except ValueError:
+        face_match = False
+
+
+counter = 0
+
+while run:
+    success, frame = camera.read()
+    img = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+
+    if success:
+        if counter % 30 == 0:
+            try:
+                threading.Thread(target=check_face, args=(img.copy(),)).start()
+            except ValueError:
+                pass
+        counter += 1
+
+        if face_match:
+            cv2.putText(img, 'Match !', (20, 450), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 255, 0), 3)
+
+        else:
+            cv2.putText(img, 'No Match !', (20, 450), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 0, 255), 3)
 
-def main():
-    st.set_page_config(page_title="Streamlit WebCam App")
-    st.title("Webcam Display Steamlit App")
-    st.caption("Powered by OpenCV, Streamlit")
-    cap = cv2.VideoCapture(0)
-    frame_placeholder = st.empty()
-    stop_button_pressed = st.button("Stop")
-    while cap.isOpened() and not stop_button_pressed:
-        ret, frame = cap.read()
-        if not ret:
-            st.write("Video Capture Ended")
-            break
-        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        frame_placeholder.image(frame,channels="RGB")
-        if cv2.waitKey(1) & 0xFF == ord("q") or stop_button_pressed:
-            break
-    cap.release()
-    cv2.destroyAllWindows()
-
-if __name__ == "__main__":
-    main()
+    window.image(img)