student.py

from tkinter import *
from tkinter import ttk
from PIL import Image, ImageTk
from tkinter import messagebox
import mysql.connector
import cv2
import os
import numpy as np
import csv

class Student_Registration:
    def __init__(self, root):
        self.root=root
        self.root.geometry("1180x580+0+0")
        self.root.title("Attendance System")

        #variables
        self.var_Name=StringVar()
        self.var_Roll=StringVar()
        self.var_Sec=StringVar()
        self.var_branch=StringVar()
        self.var_Sem=StringVar()
        self.var_Email=StringVar()
        self.var_ph=StringVar()


        title = Label(self.root, text="Student Registration", font=("Times new roman", 25,"bold"),bg="white")
        title.place(x=0,y=0,width=1180,height=40)

        main_frame = Frame(self.root, bd=2, bg="white")
        main_frame.place(x=0,y=45,width=1180, height = 550)

        #Details Frame
        det_frame=LabelFrame(main_frame, bd=2, relief=RIDGE, text="STUDENT DETAILS", font=("times new roman",14,"bold"), bg="white")
        det_frame.place(x=5, y=10, width=400, height=430)

        #Name
        Name_label=Label(det_frame,text="Name:", font=("times new roman", 14,"bold"),bg="white")
        Name_label.grid(row=1, column=0, padx=15, pady=15, sticky=W)

        Name_Entry=Entry(det_frame, textvariable=self.var_Name,width=25,font=("times new roman",12,"bold"),bg="white")
        Name_Entry.grid(row=1, column=1, padx=2, pady=15, sticky=W)

        #Roll No
        Roll_label=Label(det_frame,text="Roll No:", font=("times new roman", 14,"bold"),bg="white")
        Roll_label.grid(row=2, column=0, padx=10, sticky=W)

        Roll_Entry=Entry(det_frame, textvariable=self.var_Roll,width=25,font=("times new roman",12,"bold"),bg="white")
        Roll_Entry.grid(row=2, column=1, padx=2, pady=10, sticky=W)

        #Section
        sec_label=Label(det_frame,text="Section:", font=("times new roman", 14,"bold"),bg="white")
        sec_label.grid(row=3, column=0, padx=10, pady=15, sticky=W)

        sec_Entry=Entry(det_frame, textvariable=self.var_Sec,width=25,font=("times new roman",12,"bold"),bg="white")
        sec_Entry.grid(row=3, column=1, padx=2, pady=15, sticky=W)

        #Branch
        dep_label=Label(det_frame,text="Branch", font=("times new roman", 14,"bold"),bg="white")
        dep_label.grid(row=4,column=0, padx=10, pady=10, sticky=W)

        dep_combo=ttk.Combobox(det_frame, textvariable=self.var_branch,width=20,font=("times new roman",13,"bold"), state="readonly")
        dep_combo["values"]=("Select Branch", "CSE", "CSIT", "EEE", "ECE", "MECH", "Civil")
        dep_combo.current(0)
        dep_combo.grid(row=4, column=1, padx=2, pady=10, sticky=W)

        #Semester
        sem_label=Label(det_frame,text="Semester", font=("times new roman", 14,"bold"),bg="white")
        sem_label.grid(row=5,column=0, padx=10, pady=15, sticky=W)

        sem_combo=ttk.Combobox(det_frame, textvariable=self.var_Sem,width=20, font=("times new roman",13,"bold"), state="readonly")
        sem_combo["values"]=("Select Semester", "1st", "2nd", "3rd", "4th", "5th", "6th","7th", "8th")
        sem_combo.current(0)
        sem_combo.grid(row=5, column=1, padx=2, pady=10, sticky=W)

        #Email
        Email_label=Label(det_frame,text="Email:", font=("times new roman", 14,"bold"),bg="white")
        Email_label.grid(row=6, column=0, padx=10, sticky=W)

        Email_Entry=Entry(det_frame, textvariable=self.var_Email,width=25,font=("times new roman",12,"bold"),bg="white")
        Email_Entry.grid(row=6, column=1, padx=2, pady=10, sticky=W)

        #Phone
        Phone_label=Label(det_frame,text="Phone:", font=("times new roman", 14,"bold"),bg="white")
        Phone_label.grid(row=7, column=0, padx=10 ,pady=10, sticky=W)

        Phone_Entry=Entry(det_frame, textvariable=self.var_ph,width=25,font=("times new roman",12,"bold"),bg="white")
        Phone_Entry.grid(row=7, column=1, padx=2, pady=10, sticky=W)


        #radiobuttons
        self.var_radio1=StringVar()
        radiobt_1=ttk.Radiobutton(det_frame, variable=self.var_radio1, text="Take photo Sample", value="Yes")
        radiobt_1.grid(row=9,column=0, padx=15, pady=15)

        radiobt_1=ttk.Radiobutton(det_frame, variable=self.var_radio1,text="No photo Sample", value="No")
        radiobt_1.grid(row=9,column=1, padx=2, pady=15)

        #Buttons
        Bt_frame=Frame(main_frame,bd=2,relief=RIDGE,bg="white")
        Bt_frame.place(x=5, y=450, width=404,height=35)
        
        save_photo_bt=Button(Bt_frame,text="Save", command=self.add_data ,width=10,font=("times new roman", 12,"bold"),bg="white")
        save_photo_bt.grid(row=0,column=0)

        delete_bt=Button(Bt_frame,text="DELETE", command=self.delete_data,width=10,font=("times new roman", 12,"bold"),bg="white")
        delete_bt.grid(row=0,column=1)

        reset_bt=Button(Bt_frame,text="Reset", command=self.reset_data, width=10,font=("times new roman", 12,"bold"),bg="white")
        reset_bt.grid(row=0,column=2)

        Update_bt=Button(Bt_frame,text="Update",command=self.update_data, width=10,font=("times new roman", 12,"bold"),bg="white")
        Update_bt.grid(row=0,column=3)

        #Button
        Bt_frame1=Frame(main_frame,bd=2,relief=RIDGE,bg="white")
        Bt_frame1.place(x=5, y=485, width=402,height=35)

        take_photo_bt=Button(Bt_frame1,text="Take Photo", command=self.data_generate, width=21,font=("times new roman", 12,"bold"),bg="white")
        take_photo_bt.grid(row=0,column=0)
                
        train_photo_bt=Button(Bt_frame1,text="Train Photo", command=self.train_classifier,width=21,font=("times new roman", 12,"bold"),bg="white")
        train_photo_bt.grid(row=0,column=1)

        #Details Fetch Frame
        Search_frame=LabelFrame(main_frame, bd=2, relief=RIDGE, text="SEARCH SYSTEM", font=("times new roman",15,"bold"), bg="white")
        Search_frame.place(x=420, y=10, width=750, height=90)

        #Search
        search_label=Label(Search_frame,text="Search by:", font=("times new roman", 14,"bold"), width=10,bg="white")
        search_label.grid(row=0, column=0, padx=2, sticky=W)

        self.var_com_search=StringVar()
        search_combo=ttk.Combobox(Search_frame, textvariable= self.var_com_search, font=("times new roman",14,"bold"), state="readonly", width=10)
        search_combo["values"]=("Select", "Roll_No")
        search_combo.current(0)
        search_combo.grid(row=0, column=1, padx=5, pady=10, sticky=W)

        self.var_search=StringVar()
        roll_Entry=Entry(Search_frame,textvariable= self.var_search,width=20,font=("times new roman",14,"bold"),bg="white")
        roll_Entry.grid(row=0, column=2, padx=5, pady=10, sticky=W)

        search_bt=Button(Search_frame,command=self.search_data,text="Search", width=8,font=("times new roman", 14,"bold"),bg="white")
        search_bt.grid(row=0,column=3, padx=5)
                
        showall_bt=Button(Search_frame,command=self.fetch_data,text="Show all", width=10,font=("times new roman", 14,"bold"),bg="white")
        showall_bt.grid(row=0,column=4, padx=5)

        #Student details frame
        table_frame=Frame(main_frame, bd=2, bg="white", relief=RIDGE)
        table_frame.place(x=420, y=110, width=750, height=410)

        scroll_y=ttk.Scrollbar(table_frame, orient=VERTICAL)

        self.student_table=ttk.Treeview(table_frame, column=("name", "rollno", "section","branch", "semester", "email", "phone","photo"), yscrollcommand=scroll_y.set)
        
        scroll_y.pack(side=RIGHT, fill=Y)
        scroll_y.config(command=self.student_table.yview)

        self.student_table.heading("name", text="Name")
        self.student_table.heading("rollno", text="Roll_No")
        self.student_table.heading("section", text="Section")
        self.student_table.heading("branch", text="Branch")
        self.student_table.heading("semester", text="Semester")
        self.student_table.heading("email", text="Email")
        self.student_table.heading("phone", text="Phone")
        self.student_table.heading("photo", text="Photo Sample")

        self.student_table["show"]="headings"

        self.student_table.column("name", width=40)
        self.student_table.column("rollno", width=15)
        self.student_table.column("section", width=3)
        self.student_table.column("branch", width=5)
        self.student_table.column("semester", width=3)
        self.student_table.column("email", width=130)
        self.student_table.column("phone", width=30)
        self.student_table.column("photo", width=25)

        self.student_table.pack(fill=BOTH, expand=1)
        self.student_table.bind("<ButtonRelease>",self.get_cursor)
        self.fetch_data()

    
    #Add Data
    def add_data(self):
        if self.var_branch.get()=="Select branch" or self.var_Name.get()=="" or self.var_Roll.get()=="":
            messagebox.showerror("Error", "All fields are required", parent=self.root)
        else:
            try:
                conn = mysql.connector.connect(host='localhost', username='root', password='Chiku@4009', database='attendance-system')
                my_cursor = conn.cursor()  
                my_cursor.execute("insert into student values(%s,%s,%s,%s,%s,%s,%s,%s)", (
                                                            self.var_Name.get(),
                                                            self.var_Roll.get(),
                                                            self.var_Sec.get(),
                                                            self.var_branch.get(),
                                                            self.var_Sem.get(),
                                                            self.var_Email.get(),
                                                            self.var_ph.get(),
                                                            self.var_radio1.get()
                                                ))
                conn.commit()
                self.fetch_data()
                conn.close()
                messagebox.showinfo("Success", "Details added successfully")
            except Exception as es:
                messagebox.showerror("Error",f"Due To:{str(es)}")   

    def fetch_data(self):
        conn = mysql.connector.connect(host='localhost', username='root', password='Chiku@4009', database='attendance-system')
        my_cursor = conn.cursor()  
        my_cursor.execute("Select * from student")
        data=my_cursor.fetchall()

        if len(data)!=0:
            self.student_table.delete(*self.student_table.get_children())
            for i in data:
                self.student_table.insert("",END, values=i)                                      
            conn.commit()
        conn.close() 


    def get_cursor(self, event=""):
        cursor_focus=self.student_table.focus() 
        content=self.student_table.item(cursor_focus)   
        data=content["values"]

        self.var_Name.set(data[0]),
        self.var_Roll.set(data[1]),
        self.var_Sec.set(data[2]),
        self.var_branch.set(data[3]),
        self.var_Sem.set(data[4]),
        self.var_Email.set(data[5]),
        self.var_ph.set(data[6]),
        self.var_radio1.set(data[7])

    #update
    def update_data(self):
        if self.var_branch.get()=="Select branch" or self.var_Name.get()=="" or self.var_Roll.get()=="":
            messagebox.showerror("Error", "All fields are required")
        else:
            try:
                Update=messagebox.askyesno("Update", "Do you want to update the details?")  
                if Update>0:
                    conn = mysql.connector.connect(host='localhost', username='root', password='Chiku@4009', database='attendance-system')
                    my_cursor = conn.cursor() 
                    my_cursor.execute("Update student set Name=%s, Section=%s,Semester=%s, Branch=%s, Email=%s, Phone=%s, PhotoSample=%s where Roll_No=%s", (
                                                                                                                    self.var_Name.get(),
                                                                                                                    self.var_Sec.get(),
                                                                                                                    self.var_Sem.get(),
                                                                                                                    self.var_branch.get(),
                                                                                                                    self.var_Email.get(),
                                                                                                                    self.var_ph.get(),
                                                                                                                    self.var_radio1.get(),
                                                                                                                    self.var_Roll.get()
                                                                                                                ))
                else:
                    if not Update:
                        return
                messagebox.showinfo("Success", "Student details added successfully", parent=self.root)        
                conn.commit()
                self.fetch_data()
                conn.close()
            except Exception as es:
                messagebox.showerror("Error",f"Due to:{str(es)}")

    #delete
    def delete_data(self):
        if self.var_Roll.get()=="":
            messagebox.showerror("Error", "Roll No is mandatory")
        else:
            try:
                delete=messagebox.askyesno("Delete", "Do you want to delete the details?")   
                if delete>0:
                    conn = mysql.connector.connect(host='localhost', username='root', password='Chiku@4009', database='attendance-system')
                    my_cursor = conn.cursor() 
                    sql="delete from student where Roll_No=%s"
                    val=(self.var_Roll.get(),)
                    my_cursor.execute(sql, val)
                else:
                    if not delete:
                        return
                conn.commit()
                self.fetch_data()
                conn.close()
                messagebox.showinfo("Delete", "Student details deleted successfully", parent=self.root)  
            except Exception as es:
                messagebox.showerror("Error",f"Due to:{str(es)}")    

    #reset
    def reset_data(self):
        self.var_Name.set("")
        self.var_Roll.set("")
        self.var_Sec.set(""),
        self.var_Sem.set("Select Semester")
        self.var_branch.set("Select Branch")
        self.var_Email.set("")
        self.var_ph.set("")
        self.var_radio1.set("")

    #search data
    def search_data(self):
        if self.var_com_search.get()=="" or self.var_search.get()=="":
            messagebox.showerror("Error","Please select option")   
        else:
            try:
                conn = mysql.connector.connect(host='localhost', username='root', password='Chiku@4009', database='attendance-system')
                my_cursor = conn.cursor() 
                my_cursor.execute("select * from student where " +str(self.var_com_search.get())+" LIKE '%"+str(self.var_search.get())+"%'")
                rows=my_cursor.fetchall()

                if len(rows)!=0:
                    self.student_table.delete(*self.student_table.get_children())
                    for i in rows:
                        self.student_table.insert("",END,values=i)


                    conn.commit()
                conn.close()        
            except Exception as es:
                messagebox.showerror("Error",f"Due to:{str(es)}")

    #Photo Sample
    def data_generate(self):
        id = (self.var_Roll.get())
        name = (self.var_Name.get())
        
        if self.var_branch.get()=="Select branch" or self.var_Name.get()=="" or self.var_Roll.get()=="":
            messagebox.showerror("Error", "All fields are required", parent=self.root)
        else:
            try:
                conn = mysql.connector.connect(host='localhost', username='root', password='Chiku@4009', database='attendance-system')
                my_cursor = conn.cursor() 
                my_cursor.execute("select * from student")
                result=my_cursor.fetchall()  

                

                sql="""update student set Name=%s, Section=%s, Semester=%s, Branch=%s, Email=%s, Phone=%s, PhotoSample=%s where Roll_No=%s"""
                
                val= (
                        self.var_Name.get(),
                        self.var_Sec.get(),
                        self.var_Sem.get(),
                        self.var_branch.get(),
                        self.var_Email.get(),
                        self.var_ph.get(),
                        self.var_radio1.get(),

                        self.var_Roll.get()
                    )
                my_cursor.execute(sql, val)  
            
                conn.commit()
                self.fetch_data()
                self.reset_data()  
                conn.close()

                def is_number(s):
                    try:
                        float(s)
                        return True
                    except ValueError:
                        pass
                
                    try:
                        import unicodedata
                        unicodedata.numeric(s)
                        return True
                    except (TypeError, ValueError):
                        pass
                
                    return False

                if(is_number(id)):
                    cam = cv2.VideoCapture(0)
                    harcascadePath = "haarcascade_frontalface_default.xml"
                    face_classifier = detector = cv2.CascadeClassifier(harcascadePath)
                    sampleNum = 0
                    
                    while(True):
                        ret, img = cam.read()
                        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
                        faces = detector.detectMultiScale(gray, 1.3, 5, minSize=(30,30),flags = cv2.CASCADE_SCALE_IMAGE)

                        for(x,y,w,h) in faces:
                            cv2.rectangle(img, (x, y), (x+w, y+h), (10, 159, 255), 2)
                            sampleNum = sampleNum+1
                            #saving the captured face in the dataset folder TrainingImage
                            cv2.imwrite("data\ "+name + "."+id + '.' + str(sampleNum) + ".jpg", gray[y:y + h, x:x + w])
                            cv2.imshow('Scanning Face', img)

                        if cv2.waitKey(30) & 0xFF == ord('q'):
                            break
                        elif sampleNum > 70:
                            break
                        
                    cam.release()
                    cv2.destroyAllWindows()
                    messagebox.showinfo("Result","Generating data set successfully")
                        

            except Exception as es:
                print(es)
                messagebox.showerror("Error",f"Due to: {str(es)}",parent=self.root)
            row = [id, name]
            with open(
                "studentdetails.csv",
                "a+",
            ) as csvFile:
                writer = csv.writer(csvFile, delimiter=",")
                writer.writerow(row)
                csvFile.close()    



    #train data
    def train_classifier(self):
        data_dir=("data")
        path=[os.path.join(data_dir,file) for file in os.listdir(data_dir)]

        faces=[]
        ids=[]

        for image in path:
            img=Image.open(image).convert('L')  #gray scale image
            imageNp=np.array(img,'uint8')
            id=int(os.path.split(image)[-1].split('.')[1])

            faces.append(imageNp)
            ids.append(id)
            cv2.imshow("Training",imageNp)
            cv2.waitKey(1)==13
        ids=np.array(ids)

        #train classifier
        clf=cv2.face.LBPHFaceRecognizer_create()
        clf.train(faces,ids)
        clf.write("classifier.xml")
        cv2.destroyAllWindows()
        messagebox.showinfo("Result","Training datasets completed!!")



if __name__ == "__main__":
    root=Tk()  
    obj=Student_Registration(root)
    root.mainloop()