Update app.py

app.py

@@ -32,23 +32,20 @@ if uploaded_file is not None:
 # Take Attendance  
 st.subheader("Record Attendance")  
 attendance_date = st.date_input("Select Date")  
-image_file = st.file_uploader("Upload Image for Attendance", type=['jpg', 'png', 'jpeg'])  
-
-# Initialize present_students outside the button condition
-present_students = []
+image_file = st.file_uploader("Upload Image for Attendance", type=['jpg', 'png', 'jpeg'])
 
 if st.button("Take Attendance"):
     if image_file is not None and selected_course and attendance_date:
         image_path = os.path.join(UPLOAD_FOLDER, image_file.name)
         with open(image_path, "wb") as f:
             f.write(image_file.getbuffer())
-        present_students = attendance_system.record_attendance(selected_course, attendance_date, image_path)
+        st.session_state.present_students = attendance_system.record_attendance(selected_course, attendance_date, image_path)
 
 # Display the count of present students
-length = len(present_students)
-ty = type(present_students).__name__  # Get the name of the type as a string
+length = len(st.session_state.present_students)
+ty = type(st.session_state.present_students).__name__  # Get the name of the type as a string
 st.success(f"{length} students' attendance completed successfully. This is a {ty}.")
-st.json(present_students)
+st.json(st.session_state.present_students)
 
 # Mark attendance section
 st.subheader("Mark Attendance")
@@ -64,7 +61,7 @@ if selected_roll_numbers:
     st.write("Selected Roll Numbers to Add:", selected_roll_numbers)
 
 # Combine the present students with the selected roll numbers
-present_students_str = [str(student) for student in present_students]
+present_students_str = [str(student) for student in st.session_state.present_students]
 combined_students = list(set(present_students_str) | set(selected_roll_numbers))
 
 # Optionally, show final list of students
@@ -75,7 +72,9 @@ if st.button("Mark Attendance"):
     # Convert combined_students back to integers for record_attendance_new function
     combined_students_int = [int(student) for student in combined_students]
     record_attendance_new(combined_students_int, selected_course)
-    st.success("Attendance marked successfully")
+    st.success(f"{len(combined_students_int)} students' attendance is going to record.")
+    st.success(f"{len(combined_students_int)} students' roll number is going to record. Students present in combined_student: {combined_students_int}")
+
 # Add to database  
 st.subheader("Add Student to Database")  
 roll_number = st.text_input("Enter Roll Number")