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(
@@ -47,12 +48,25 @@ def validate_aadhaar(aadhaar):
     # For simplicity, let's assume any 12-digit number is a valid Aadhaar card
     return len(aadhaar) == 12 and aadhaar.isdigit()
 
+# Create SQLite database connection
+conn = sqlite3.connect('users.db')
+cursor = conn.cursor()
+
+# Create users table if it doesn't exist
+cursor.execute('''
+    CREATE TABLE IF NOT EXISTS users (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        name TEXT NOT NULL,
+        aadhaar TEXT NOT NULL,
+        face_encoding TEXT NOT NULL
+    )
+''')
+conn.commit()
+
 # 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):
@@ -82,6 +96,11 @@ if img_file_buffer is not None:
                 cv2.putText(image, name, (x1 + 6, y2 - 6), cv2.FONT_HERSHEY_COMPLEX, 1, (255, 255, 255), 2)
 
                 if name != "Unknown":
+                    # Insert user information into the database
+                    cursor.execute("INSERT INTO users (name, aadhaar, face_encoding) VALUES (?, ?, ?)",
+                                   (name, aadhaar_number, str(encodeFace)))
+                    conn.commit()
+
                     url = "https://mrvishal7705.000webhostapp.com"
                     url1 = "/update.php"
                     data1 = {'name': name, 'aadhaar': aadhaar_number}
@@ -100,4 +119,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 12-digit Aadhaar number.")
-