Update app.py

app.py

@@ -5,6 +5,10 @@ import datetime
 import pandas as pd
 from PIL import Image
 import io
+import os
+
+# Create a directory for storing photos
+os.makedirs("photos", exist_ok=True)
 
 def create_database():
     """Creates the attendance database."""
@@ -12,18 +16,27 @@ def create_database():
     c = conn.cursor()
     c.execute('''CREATE TABLE IF NOT EXISTS attendance (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
+                rollno TEXT,
                 name TEXT,
+                photo_path TEXT,
                 timestamp TEXT
                 )''')
     conn.commit()
     conn.close()
 
-def mark_attendance(name):
+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")
-    c.execute("INSERT INTO attendance (name, timestamp) VALUES (?, ?)", (name, timestamp))
+
+    # Save the captured photo
+    photo_path = f"photos/{rollno}_{name.replace(' ', '_')}.jpg"
+    photo = Image.fromarray(frame)
+    photo.save(photo_path)
+
+    # Insert record into the database
+    c.execute("INSERT INTO attendance (rollno, name, photo_path, timestamp) VALUES (?, ?, ?, ?)", (rollno, name, photo_path, timestamp))
     conn.commit()
     conn.close()
 
@@ -31,7 +44,7 @@ def get_attendance_records():
     """Fetches all attendance records."""
     conn = sqlite3.connect('attendance.db')
     c = conn.cursor()
-    c.execute("SELECT * FROM attendance")
+    c.execute("SELECT rollno, name, photo_path, timestamp FROM attendance")
     records = c.fetchall()
     conn.close()
     return records
@@ -56,23 +69,36 @@ def capture_photo():
             cv2.destroyAllWindows()
             return None
 
+# Initialize the database
 create_database()
+
 st.title("Attendance System")
 
 menu = st.sidebar.selectbox("Menu", ["Click Photo", "Database"])
 
 if menu == "Click Photo":
     st.header("Mark Attendance by Clicking Photo")
-    name = st.text_input("Enter your name:")
-    if name:
+    rollno = st.text_input("Enter your Roll Number:")
+    name = st.text_input("Enter your Name:")
+    if rollno and name:
         frame = capture_photo()
         if frame is not None:
-            st.success(f"Attendance marked for {name}")
-            mark_attendance(name)
+            mark_attendance(rollno, name, frame)
+            st.success(f"Attendance marked for {name} (Roll No: {rollno})")
             st.image(frame, caption="Captured Image", channels="RGB")
 
 elif menu == "Database":
     st.header("Attendance Records")
     records = get_attendance_records()
-    df = pd.DataFrame(records, columns=["ID", "Name", "Timestamp"])
-    st.dataframe(df)
+    data = []
+    for rollno, name, photo_path, timestamp in records:
+        photo = Image.open(photo_path) if os.path.exists(photo_path) else None
+        data.append((rollno, name, photo, timestamp))
+
+    df = pd.DataFrame(data, columns=["Roll No", "Name", "Photo", "Timestamp"])
+    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']}")