attendance-1 / app.py
ravinder2024's picture
Update app.py
1631231 verified
import streamlit as st
import sqlite3
import cv2
import datetime
import pandas as pd
from PIL import Image
import os
# Create a directory for storing photos
os.makedirs("photos", exist_ok=True)
def create_database():
"""Creates the attendance database."""
conn = sqlite3.connect('attendance.db')
c = conn.cursor()
c.execute('''CREATE TABLE IF NOT EXISTS attendance (
id INTEGER PRIMARY KEY AUTOINCREMENT,
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()
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, emotion, timestamp) VALUES (?, ?, ?, ?, ?)",
(rollno, name, photo_path, emotion, timestamp))
conn.commit()
conn.close()
def get_attendance_records():
"""Fetches all attendance records."""
conn = sqlite3.connect('attendance.db')
c = conn.cursor()
c.execute("SELECT rollno, name, photo_path, emotion, timestamp FROM attendance")
records = c.fetchall()
conn.close()
return records
def capture_photo():
"""Captures a photo using the webcam."""
cap = cv2.VideoCapture(0)
st.info("Click 'Capture' to take a photo.")
while cap.isOpened():
ret, frame = cap.read()
if not ret:
st.error("Failed to capture image.")
break
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
st.image(frame, channels="RGB")
if st.button("Capture"):
cap.release()
cv2.destroyAllWindows()
return frame
if st.button("Cancel"):
cap.release()
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")
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:
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()
data = []
for rollno, name, photo_path, emotion, timestamp in records:
photo = Image.open(photo_path) if os.path.exists(photo_path) else None
date, time = timestamp.split(" ")
data.append((rollno, photo, name, emotion, date, time))
df = pd.DataFrame(data, columns=["Roll No", "Photo", "Name", "Emotion", "Date", "Time"])
st.write("### Attendance Records")
for idx, row in df.iterrows():
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