Update app.py

app.py

@@ -6,7 +6,6 @@ import face_recognition
 import os
 from datetime import datetime
 import streamlit as st
-import sqlite3
 
 # Set page title and description
 st.set_page_config(
@@ -18,20 +17,6 @@ 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 = []
@@ -63,10 +48,9 @@ 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.file_uploader("Upload an image", type=["jpg", "jpeg"])
+img_file_buffer = st.camera_input("Take a picture")
 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):
@@ -107,11 +91,6 @@ 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")
@@ -120,6 +99,3 @@ if img_file_buffer is not None:
             st.info("Face not detected. Please try again.")
     else:
         st.error("Invalid Aadhaar card number. Please enter a valid 4-digit Aadhaar number.")
-
-# Close the database connection
-conn.close()