Update app.py

app.py

@@ -7,7 +7,7 @@ import os
 import sqlite3
 from datetime import datetime
 
-
+# Establish connection to SQLite database
 conn = sqlite3.connect('attendance.db')
 c = conn.cursor()
 
@@ -15,7 +15,6 @@ c = conn.cursor()
 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
@@ -24,11 +23,6 @@ classnames = []
 directory = "photos"
 myList = os.listdir(directory)
 current_datetime = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-if not os.path.isdir('Attendance'):
-        os.makedirs('Attendance')
-if f'Attendance-{current_datetime}.csv' not in os.listdir('Attendance'):
-        with open(f'Attendance/Attendance-{current_datetime}.csv', 'w') as f:
-            f.write('Name,Time')
 
 st.write("Photographs found in folder : ")
 for cls in myList:
@@ -47,13 +41,8 @@ file_name = st.camera_input("Upload image")
 
 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:
     col1, col2 = st.columns(2)
@@ -94,6 +83,9 @@ if file_name is not None:
             cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
             cv2.putText(image, name, (x1 + 6, y2 - 6), cv2.FONT_HERSHEY_COMPLEX, 1, (255, 255, 255), 2)
             
+            # Store attendance in SQLite database
+            add_attendance(name)
+            
         # Display the image with recognized faces
         st.image(image, use_column_width=True, output_format="PNG")
 
@@ -101,7 +93,5 @@ if file_name is not None:
         st.write("Recognized Names:")
         for i, name in enumerate(recognized_names):
             st.write(f"Face {i+1}: {name}")
-        if len(recognized_names) > 0:
-            add_attendance(recognized_names)
     else:
-        st.warning("No faces detected in the image. Face recognition failed.")
+        st.warning("No faces detected in the image. Face recognition failed.")