Update app.py

app.py

@@ -34,28 +34,32 @@ cursor.execute('''
 conn.commit()
 
 # Load images for face recognition
-Images = []
-classnames = []
-directory = "photos"
-
-myList = os.listdir(directory)
-
-for cls in myList:
-    if os.path.splitext(cls)[1] in [".jpg", ".jpeg"]:
-        img_path = os.path.join(directory, cls)
-        curImg = cv2.imread(img_path)
-        Images.append(curImg)
-        classnames.append(os.path.splitext(cls)[0])
-
-def findEncodings(Images):
-    encodeList = []
-    for img in Images:
+def load_images(directory):
+    Images = []
+    classnames = []
+
+    myList = os.listdir(directory)
+
+    for cls in myList:
+        if os.path.splitext(cls)[1] in [".jpg", ".jpeg"]:
+            img_path = os.path.join(directory, cls)
+            curImg = cv2.imread(img_path)
+            Images.append(curImg)
+            classnames.append(os.path.splitext(cls)[0])
+
+    return Images, classnames
+
+Images, classnames = load_images("photos")
+
+def find_encodings(images):
+    encode_list = []
+    for img in images:
         img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
         encode = face_recognition.face_encodings(img)[0]
-        encodeList.append(encode)
-    return encodeList
+        encode_list.append(encode)
+    return encode_list
 
-encodeListknown = findEncodings(Images)
+encodeListKnown = find_encodings(Images)
 
 # Function to validate Aadhaar card number
 def validate_aadhaar(aadhaar):
@@ -63,70 +67,62 @@ def validate_aadhaar(aadhaar):
     # For simplicity, let's assume any 4-digit number is a valid Aadhaar card
     return len(aadhaar) == 4 and aadhaar.isdigit()
 
-# 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:")
-
-# Face recognition code...
-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)
-
-        imgS = cv2.resize(image, (0, 0), None, 0.25, 0.25)
-        imgS = cv2.cvtColor(imgS, cv2.COLOR_BGR2RGB)
-        facesCurFrame = face_recognition.face_locations(imgS)
-        encodesCurFrame = face_recognition.face_encodings(imgS, facesCurFrame)
-
-        name = "Unknown"  # Default name for unknown faces
-
-        if len(encodesCurFrame) > 0:
-            for encodeFace, faceLoc in zip(encodesCurFrame, facesCurFrame):
-                matches = face_recognition.compare_faces(encodeListknown, encodeFace)
-                faceDis = face_recognition.face_distance(encodeListknown, encodeFace)
-                matchIndex = np.argmin(faceDis)
-
-                if matches[matchIndex]:
-                    name = classnames[matchIndex].upper()
-
-                y1, x2, y2, x1 = faceLoc
-                y1, x2, y2, x1 = y1 * 4, x2 * 4, y2 * 4, x1 * 4
-                cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
-                cv2.rectangle(image, (x1, y2 - 35), (x2, y2), (0, 255, 0), cv2.FILLED)
-                cv2.putText(image, name, (x1 + 6, y2 - 6), cv2.FONT_HERSHEY_COMPLEX, 1, (255, 255, 255), 2)
-
-                if name != "Unknown":
-                    # Update Aadhaar data
-                    url = "https://attendanceviaface.000webhostapp.com"
-                    url1 = "/update.php"
-                    data1 = {'name': name, 'aadhaar': aadhaar_number}
-                    response = requests.post(url + url1, data=data1)
-
-                    if response.status_code == 200:
-                        st.success("Data updated on: " + url)
-                    else:
-                        st.warning("Data not updated")
+# Function to verify Aadhaar from the database
+def verify_aadhaar_from_database(aadhaar_number):
+    cursor.execute("SELECT * FROM faces WHERE aadhaar = ?", (aadhaar_number,))
+    user_data = cursor.fetchone()
+    return user_data is not None
+
+# Function to perform face recognition and Aadhaar verification
+def recognize_and_verify(image, encodeListKnown, aadhaar_number):
+    imgS = cv2.resize(image, (0, 0), None, 0.25, 0.25)
+    imgS = cv2.cvtColor(imgS, cv2.COLOR_BGR2RGB)
+    facesCurFrame = face_recognition.face_locations(imgS)
+    encodesCurFrame = face_recognition.face_encodings(imgS, facesCurFrame)
+
+    name = "Unknown"  # Default name for unknown faces
+
+    if len(encodesCurFrame) > 0:
+        for encodeFace, faceLoc in zip(encodesCurFrame, facesCurFrame):
+            matches = face_recognition.compare_faces(encodeListKnown, encodeFace)
+            faceDis = face_recognition.face_distance(encodeListKnown, encodeFace)
+            matchIndex = np.argmin(faceDis)
+
+            if matches[matchIndex]:
+                name = classnames[matchIndex].upper()
+
+            y1, x2, y2, x1 = faceLoc
+            y1, x2, y2, x1 = y1 * 4, x2 * 4, y2 * 4, x1 * 4
+            cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
+            cv2.rectangle(image, (x1, y2 - 35), (x2, y2), (0, 255, 0), cv2.FILLED)
+            cv2.putText(image, name, (x1 + 6, y2 - 6), cv2.FONT_HERSHEY_COMPLEX, 1, (255, 255, 255), 2)
+
+            if name != "Unknown":
+                # Update Aadhaar data
+                url = "https://attendanceviaface.000webhostapp.com"
+                url1 = "/update.php"
+                data1 = {'name': name, 'aadhaar': aadhaar_number}
+                response = requests.post(url + url1, data=data1)
+
+                if response.status_code == 200:
+                    st.success("Data updated on: " + url)
+                else:
+                    st.warning("Data not updated")
 
-                    # Store face encoding and Aadhaar number in the database
-                    face_encoding_bytes = pickle.dumps(encodeFace)
-                    cursor.execute("INSERT INTO faces (aadhaar, encoding) VALUES (?, ?)", (aadhaar_number, face_encoding_bytes))
-                    conn.commit()
+                # Store face encoding and Aadhaar number in the database
+                face_encoding_bytes = pickle.dumps(encodeFace)
+                cursor.execute("INSERT INTO faces (aadhaar, encoding) VALUES (?, ?)", (aadhaar_number, face_encoding_bytes))
+                conn.commit()
 
-        # Apply styling with CSS
-        st.markdown('<style>img { animation: pulse 2s infinite; }</style>', unsafe_allow_html=True)
-        st.image(image, use_column_width=True, output_format="PNG")
+    # Apply styling with CSS
+    st.markdown('<style>img { animation: pulse 2s infinite; }</style>', unsafe_allow_html=True)
+    st.image(image, use_column_width=True, output_format="PNG")
 
-        if name == "Unknown":
-            st.info("Face not detected. Please try again.")
-    else:
-        st.error("Invalid Aadhaar card number. Please enter a valid 4-digit Aadhaar number.")
-
-# Section to take attendance by clicking a photo
-st.header("Take Attendance by Clicking Photo")
-capture_button = st.button("Capture Photo for Attendance")
+    if name == "Unknown":
+        st.info("Face not detected. Please try again.")
 
-if capture_button:
+# Function to take attendance by clicking a photo
+def take_attendance_by_photo(encodeListKnown, aadhaar_number):
     # Capture photo using the camera
     captured_image = st.camera_input("Capture a photo for attendance")
 
@@ -143,30 +139,44 @@ if capture_button:
         if len(faces_captured) > 0:
             # Check if any of the captured faces match known faces
             for encode_captured, face_loc in zip(encodes_captured, faces_captured):
-                matches_captured = face_recognition.compare_faces(encodeListknown, encode_captured)
-                face_dis_captured = face_recognition.face_distance(encodeListknown, encode_captured)
+                matches_captured = face_recognition.compare_faces(encodeListKnown, encode_captured)
+                face_dis_captured = face_recognition.face_distance(encodeListKnown, encode_captured)
                 match_index_captured = np.argmin(face_dis_captured)
 
                 if matches_captured[match_index_captured]:
                     name_captured = classnames[match_index_captured].upper()
 
-                    # Check if the user is already present in the database
-                    cursor.execute("SELECT * FROM faces WHERE aadhaar = ?", (aadhaar_number,))
-                    user_data = cursor.fetchone()
-
-                    if user_data:
-                        st.warning(f"{name_captured} is already present in the database.")
-                    else:
-                        # Store face encoding and Aadhaar number in the database
-                        face_encoding_bytes_captured = pickle.dumps(encode_captured)
-                        cursor.execute("INSERT INTO faces (aadhaar, encoding) VALUES (?, ?)", (aadhaar_number, face_encoding_bytes_captured))
-                        conn.commit()
+                    # Verify Aadhaar from the database
+                    if verify_aadhaar_from_database(aadhaar_number):
                         st.success(f"{name_captured}'s attendance recorded.")
+                    else:
+                        st.warning(f"{name_captured} is not registered with Aadhaar {aadhaar_number}.")
                 else:
                     st.warning("No matching face found in the database.")
 
         else:
             st.warning("No face detected in the captured photo. Please try again.")
 
+# 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:")
+
+# Face recognition and Aadhaar verification code...
+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)
+        recognize_and_verify(image, encodeListKnown, aadhaar_number)
+    else:
+        st.error("Invalid Aadhaar card number. Please enter a valid 4-digit Aadhaar number.")
+
+# Section to take attendance by clicking a photo
+st.header("Take Attendance by Clicking Photo")
+capture_button = st.button("Capture Photo for Attendance")
+
+if capture_button:
+    take_attendance_by_photo(encodeListKnown, aadhaar_number)
+
 # Close the database connection
 conn.close()