Update app.py

app.py

@@ -4,7 +4,6 @@ import cv2
 import datetime
 import pandas as pd
 from PIL import Image
-import io
 import os
 
 # Create a directory for storing photos
@@ -19,24 +18,36 @@ def create_database():
                 rollno TEXT,
                 name TEXT,
                 photo_path TEXT,
+                emotion TEXT,
                 timestamp TEXT
                 )''')
     conn.commit()
     conn.close()
 
+def detect_emotion(frame):
+    """Dummy emotion detection function."""
+    # Placeholder for actual emotion detection logic
+    return "Happy"
+
 def mark_attendance(rollno, name, frame):
     """Marks attendance for the user."""
     conn = sqlite3.connect('attendance.db')
     c = conn.cursor()
-    timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    timestamp = datetime.datetime.now()
+    date_str = timestamp.strftime("%Y-%m-%d")
+    time_str = timestamp.strftime("%H:%M:%S")
 
     # Save the captured photo
     photo_path = f"photos/{rollno}_{name.replace(' ', '_')}.jpg"
     photo = Image.fromarray(frame)
     photo.save(photo_path)
 
+    # Detect emotion (dummy function)
+    emotion = detect_emotion(frame)
+
     # Insert record into the database
-    c.execute("INSERT INTO attendance (rollno, name, photo_path, timestamp) VALUES (?, ?, ?, ?)", (rollno, name, photo_path, timestamp))
+    c.execute("INSERT INTO attendance (rollno, name, photo_path, emotion, timestamp) VALUES (?, ?, ?, ?, ?)",
+              (rollno, name, photo_path, emotion, timestamp))
     conn.commit()
     conn.close()
 
@@ -44,7 +55,7 @@ def get_attendance_records():
     """Fetches all attendance records."""
     conn = sqlite3.connect('attendance.db')
     c = conn.cursor()
-    c.execute("SELECT rollno, name, photo_path, timestamp FROM attendance")
+    c.execute("SELECT rollno, name, photo_path, emotion, timestamp FROM attendance")
     records = c.fetchall()
     conn.close()
     return records
@@ -91,14 +102,14 @@ elif menu == "Database":
     st.header("Attendance Records")
     records = get_attendance_records()
     data = []
-    for rollno, name, photo_path, timestamp in records:
+    for rollno, name, photo_path, emotion, timestamp in records:
         photo = Image.open(photo_path) if os.path.exists(photo_path) else None
-        data.append((rollno, name, photo, timestamp))
+        date, time = timestamp.split(" ")
+        data.append((rollno, photo, name, emotion, date, time))
 
-    df = pd.DataFrame(data, columns=["Roll No", "Name", "Photo", "Timestamp"])
+    df = pd.DataFrame(data, columns=["Roll No", "Photo", "Name", "Emotion", "Date", "Time"])
     st.write("### Attendance Records")
-    st.dataframe(df.drop(columns=["Photo"]))  # Display text columns
-
     for idx, row in df.iterrows():
-        if row["Photo"]:
-            st.image(row["Photo"], caption=f"Roll No: {row['Roll No']} - Name: {row['Name']}")
+        st.image(row["Photo"], caption=f"Roll No: {row['Roll No']} - Name: {row['Name']} - Emotion: {row['Emotion']}")
+    st.table(df.drop(columns=["Photo"]))  # Exclude photo for table display
+