Update app.py

app.py

@@ -5,7 +5,6 @@ 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
@@ -43,16 +42,11 @@ except Exception as e:
 # Database Functions
 def create_table():
     with sqlite3.connect(DB_FILE) as conn:
-        conn.execute("""
-            CREATE TABLE IF NOT EXISTS students (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                name TEXT NOT NULL,
-                roll_number TEXT NOT NULL UNIQUE,
-                image_path TEXT NOT NULL,
-                emotion TEXT,
-                timestamp TEXT
-            )
-        """)
+        conn.execute("""CREATE TABLE IF NOT EXISTS students (
+                        id INTEGER PRIMARY KEY AUTOINCREMENT,
+                        name TEXT NOT NULL,
+                        roll_number TEXT NOT NULL UNIQUE,
+                        image_path TEXT NOT NULL)""")
         conn.commit()
 
 def insert_student(name, roll_number, image_path):
@@ -64,16 +58,6 @@ 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")
@@ -130,7 +114,7 @@ if choice == "Register Student":
                             insert_student(name, roll_number, img_path)
                             st.success(f"Student {name} Registered Successfully!")
                         except Exception as e:
-                            st.error(f"Error: {e}")
+                            st.error(f"Error saving webcam image: {e}")
                 elif image_file:
                     try:
                         img = Image.open(image_file)
@@ -140,7 +124,7 @@ if choice == "Register Student":
                         insert_student(name, roll_number, img_path)
                         st.success(f"Student {name} Registered Successfully!")
                     except Exception as e:
-                        st.error(f"Error: {e}")
+                        st.error(f"Error saving uploaded image: {e}")
                 else:
                     st.warning("Please upload an image or use the webcam to register the face.")
             else:
@@ -160,14 +144,6 @@ 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:
@@ -183,14 +159,6 @@ 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:
@@ -202,6 +170,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]}, Emotion: {student[4]}, Timestamp: {student[5]}")
+            st.write(f"Name: {student[1]}, Roll Number: {student[2]}")
     else:
         st.warning("No students registered.")