Update app.py

app.py

@@ -6,6 +6,7 @@ import face_recognition
 import os
 from datetime import datetime
 import streamlit as st
+import sqlite3
 
 # Set page title and description
 st.set_page_config(
@@ -17,6 +18,20 @@ st.set_page_config(
 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.")
 
+# Create or connect to the SQLite database
+conn = sqlite3.connect("attendance_database.db")
+cursor = conn.cursor()
+
+# Check if the 'faces' table exists, create it if not
+cursor.execute('''
+    CREATE TABLE IF NOT EXISTS faces (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        aadhaar TEXT,
+        encoding BLOB
+    )
+''')
+conn.commit()
+
 # Load images for face recognition
 Images = []
 classnames = []
@@ -48,9 +63,10 @@ def validate_aadhaar(aadhaar):
     return len(aadhaar) == 4 and aadhaar.isdigit()
 
 # Take picture using the camera and input Aadhaar card details
-img_file_buffer = st.camera_input("Take a picture")
+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):
@@ -91,6 +107,11 @@ if img_file_buffer is not None:
                     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()
+
         # 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")
@@ -98,4 +119,7 @@ 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.")
+        st.error("Invalid Aadhaar card number. Please enter a valid 4-digit Aadhaar number.")
+
+# Close the database connection
+conn.close()