Update app.py

app.py

@@ -4,8 +4,8 @@ import cv2
 import requests
 import face_recognition
 import os
-from datetime import datetime
 import streamlit as st
+import csv
 
 # Set page title and description
 st.set_page_config(
@@ -20,8 +20,9 @@ st.markdown("This app recognizes faces in an image, verifies Aadhaar card detail
 # Load images for face recognition
 Images = []
 classnames = []
-directory = "photos"
+aadhar_numbers = []  # New list to store Aadhaar numbers
 
+directory = "photos"
 myList = os.listdir(directory)
 
 for cls in myList:
@@ -30,16 +31,8 @@ for cls in myList:
         curImg = cv2.imread(img_path)
         Images.append(curImg)
         classnames.append(os.path.splitext(cls)[0])
-
-def findEncodings(Images):
-    encodeList = []
-    for img in Images:
-        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-        encode = face_recognition.face_encodings(img)[0]
-        encodeList.append(encode)
-    return encodeList
-
-encodeListknown = findEncodings(Images)
+        # Assume Aadhaar number is part of the image filename (e.g., "123456_john.jpg")
+        aadhar_numbers.append(cls.split('_')[0])
 
 # Function to validate Aadhaar card number
 def validate_aadhaar(aadhaar):
@@ -47,6 +40,18 @@ def validate_aadhaar(aadhaar):
     # For simplicity, let's assume any 6-digit number is a valid Aadhaar card
     return len(aadhaar) == 6 and aadhaar.isdigit()
 
+# Function to read names and Aadhaar numbers from a CSV file
+def read_csv(file_path):
+    data = []
+    with open(file_path, mode='r') as file:
+        reader = csv.reader(file)
+        for row in reader:
+            data.append(row)
+    return data
+
+# Load names and Aadhaar numbers from CSV file
+csv_data = read_csv('csv.csv')  # Replace 'your_csv_file.csv' with the actual file path
+
 # Take picture using the camera and input Aadhaar card details
 img_file_buffer = st.camera_input("Take a picture")
 aadhaar_number = st.text_input("Enter Aadhaar Number:")
@@ -91,11 +96,20 @@ if img_file_buffer is not None:
                     else:
                         st.warning("Data not updated")
 
+        # Display the name corresponding to the entered Aadhaar number
+        if name == "Unknown":
+            # Match Aadhaar number with the list
+            aadhar_index = aadhar_numbers.index(aadhaar_number) if aadhaar_number in aadhar_numbers else None
+            if aadhar_index is not None:
+                st.success(f"Match found: {csv_data[aadhar_index][0]}")
+            else:
+                st.warning("Face not detected, and Aadhaar number not found in the database.")
+        else:
+            st.warning("Aadhaar number is valid, but face recognition failed.")
+
         # 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 6-digit Aadhaar number.")