Update app.py

app.py

@@ -6,6 +6,7 @@ import face_recognition
 import os
 from datetime import datetime
 import streamlit as st
+from streamlit_webrtc import VideoTransformerBase, webrtc_streamer
 
 # Set page title and description
 st.set_page_config(
@@ -15,7 +16,7 @@ st.set_page_config(
     initial_sidebar_state="collapsed"
 )
 st.title("Attendance System Using Face Recognition 📷")
-st.markdown("This app recognizes faces in an image, verifies Aadhaar card details, and updates attendance records with the current timestamp & Location.")
+st.markdown("This app recognizes faces in a live camera feed, verifies Aadhaar card details, and updates attendance records with the current timestamp & Location.")
 
 # Load images for face recognition
 Images = []
@@ -44,20 +45,26 @@ encodeListknown = findEncodings(Images)
 # Function to validate Aadhaar card number
 def validate_aadhaar(aadhaar):
     # Implement your Aadhaar card validation logic here
-    # For simplicity, let's assume any 12-digit number is a valid Aadhaar card
-    return len(aadhaar) == 12 and aadhaar.isdigit()
+    # For simplicity, let's assume any 4-digit number is a valid Aadhaar card
+    return len(aadhaar) == 4 and aadhaar.isdigit()
+
+# Create a Streamlit component to capture images from the camera
+webrtc_ctx = webrtc_streamer(
+    key="example",
+    video_transformer_factory=None,  # No need for video transformation for this example
+    async_transform=True
+)
 
-# Take picture using the camera and input Aadhaar card details
-img_file_buffer = st.file_uploader("Upload an image", type=["jpg", "jpeg"])
-aadhaar_number = st.text_input("Enter Aadhaar Number:")
+# Main Streamlit app logic
+if webrtc_ctx.video_transformer:
+    st.sidebar.markdown("# Capture Image")
+    aadhaar_number = st.sidebar.text_input("Enter Aadhaar Number:")
 
-# Face recognition code...
+    # Use the camera feed to capture images
+    frame = webrtc_ctx.video_transformer.get_frame()
 
-if img_file_buffer is not None:
-    # Validate Aadhaar card number
-    if validate_aadhaar(aadhaar_number):
-        test_image = Image.open(img_file_buffer)
-        image = np.asarray(test_image)
+    if frame is not None:
+        image = np.array(frame.to_image())
 
         imgS = cv2.resize(image, (0, 0), None, 0.25, 0.25)
         imgS = cv2.cvtColor(imgS, cv2.COLOR_BGR2RGB)
@@ -98,6 +105,5 @@ if img_file_buffer is not None:
 
         if name == "Unknown":
             st.info("Face not detected. Please try again.")
-    else:
-        st.error("Invalid Aadhaar card number. Please enter a valid 12-digit Aadhaar number.")
-
+else:
+    st.warning("Webcam not available. Please make sure your camera is connected and accessible.")