jolly_time / app.py
Veerammal's picture
Create app.py
dc82029 verified
import streamlit as st
import cv2
import numpy as np
import face_recognition
from PIL import Image
# Simulated database of student details
students_db = {
"student1": "Details for Student 1",
"student2": "Details for Student 2",
}
def load_image(image_file):
img = Image.open(image_file)
return np.array(img)
def recognize_face(image):
# Find all face locations and encodings in the image
face_locations = face_recognition.face_locations(image)
face_encodings = face_recognition.face_encodings(image, face_locations)
recognized_students = []
for face_encoding in face_encodings:
# Compare with known faces (placeholder)
# In a real app, you'd load known face encodings from the database
# Here, it's just a mock example
matches = [True] # Replace this with actual matching logic
if True in matches:
recognized_students.append("student1") # Replace with actual name
return recognized_students
st.title("Student Face Recognition")
uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png"])
if uploaded_file is not None:
# Load and display the image
image = load_image(uploaded_file)
st.image(image, caption='Uploaded Image.', use_column_width=True)
# Recognize faces
recognized_students = recognize_face(image)
if recognized_students:
for student in recognized_students:
st.write(students_db.get(student, "Student not found"))
else:
st.write("No students recognized.")