Update app.py

app.py

@@ -4,9 +4,18 @@ import face_recognition
 import cv2
 import numpy as np
 import os
+import sqlite3
 from datetime import datetime
 import CSV
 
+conn = sqlite3.connect('attendance.db')
+c = conn.cursor()
+
+# Create a table for storing attendance if it doesn't exist
+c.execute('''CREATE TABLE IF NOT EXISTS attendance 
+             (id INTEGER PRIMARY KEY AUTOINCREMENT, name TEXT, time TEXT)''')
+
+
 st.title("AIMLJan24 - Face Recognition")
 
 # Load images for face recognition
@@ -36,13 +45,14 @@ encodeListknown = [face_recognition.face_encodings(img)[0] for img in Images]
 # camera to take photo of user in question
 file_name = st.camera_input("Upload image")
 
-def add_attendance(names):
-    current_datetime = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    print('Date',current_datetime)
-    with open(f'Attendance/Attendance-{current_datetime}.csv', 'a') as f:
-        f.write('Name,Time\n')  # Write header if file is newly created
-        for name in names:
-            f.write(f'{name},{current_datetime}\n')
+def add_attendance(name):
+    username = name
+    
+    print(current_datetime)
+    
+    # Insert data into the attendance table
+    c.execute("INSERT INTO attendance (name, time) VALUES (?, ?)", (username, current_datetime))
+    conn.commit()
    
 
 if file_name is not None: