Update app.py

app.py

@@ -5,6 +5,7 @@ import numpy as np
 from keras.models import load_model
 from PIL import Image
 import sqlite3
+from datetime import datetime
 from huggingface_hub import HfApi
 
 # Constants
@@ -47,7 +48,9 @@ def create_table():
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
                 name TEXT NOT NULL,
                 roll_number TEXT NOT NULL UNIQUE,
-                image_path TEXT NOT NULL
+                image_path TEXT NOT NULL,
+                emotion TEXT,
+                timestamp TEXT
             )
         """)
         conn.commit()
@@ -61,6 +64,16 @@ def insert_student(name, roll_number, image_path):
     except sqlite3.IntegrityError:
         st.warning("Roll number already exists!")
 
+def insert_attendance(roll_number, emotion):
+    timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+    with sqlite3.connect(DB_FILE) as conn:
+        conn.execute("""
+            UPDATE students
+            SET emotion = ?, timestamp = ?
+            WHERE roll_number = ?
+        """, (emotion, timestamp, roll_number))
+        conn.commit()
+
 def get_all_students():
     with sqlite3.connect(DB_FILE) as conn:
         cursor = conn.execute("SELECT * FROM students")
@@ -147,6 +160,14 @@ elif choice == "Face Recognition and Emotion Detection":
                 emotion_label = detect_faces_and_emotions(img_array)
                 if emotion_label:
                     st.success(f"Emotion Detected: {emotion_label}")
+                    # Update the attendance
+                    roll_number = st.text_input("Enter Roll Number for Attendance")
+                    if st.button("Record Attendance"):
+                        if roll_number:
+                            insert_attendance(roll_number, emotion_label)
+                            st.success(f"Attendance recorded for Roll No: {roll_number}")
+                        else:
+                            st.warning("Please enter the roll number.")
                 else:
                     st.warning("No face detected.")
             except Exception as e:
@@ -162,6 +183,14 @@ elif choice == "Face Recognition and Emotion Detection":
                 emotion_label = detect_faces_and_emotions(img_array)
                 if emotion_label:
                     st.success(f"Emotion Detected: {emotion_label}")
+                    # Update the attendance
+                    roll_number = st.text_input("Enter Roll Number for Attendance")
+                    if st.button("Record Attendance"):
+                        if roll_number:
+                            insert_attendance(roll_number, emotion_label)
+                            st.success(f"Attendance recorded for Roll No: {roll_number}")
+                        else:
+                            st.warning("Please enter the roll number.")
                 else:
                     st.warning("No face detected.")
             except Exception as e:
@@ -173,6 +202,6 @@ elif choice == "View Attendance":
     students = get_all_students()
     if students:
         for student in students:
-            st.write(f"Name: {student[1]}, Roll Number: {student[2]}")
+            st.write(f"Name: {student[1]}, Roll Number: {student[2]}, Emotion: {student[4]}, Timestamp: {student[5]}")
     else:
         st.warning("No students registered.")