Update app.py

app.py

@@ -10,6 +10,10 @@ os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 os.makedirs(DATABASE_DIR, exist_ok=True)
 attendance_system = AttendanceSystem(DATABASE_DIR)
 
+# Initialize session state
+if 'present_students' not in st.session_state:
+    st.session_state.present_students = []
+
 # Streamlit application
 st.title("Attendance System")
 
@@ -33,26 +37,23 @@ st.subheader("Record Attendance")
 attendance_date = st.date_input("Select Date", key="attendance_date")
 image_file = st.file_uploader("Upload Image for Attendance", type=['jpg', 'png', 'jpeg'], key="attendance_image_uploader")
 
-# Initialize present_students outside the button condition
-present_students = []
-
 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.write(present_students)
+        st.session_state.present_students = attendance_system.record_attendance(selected_course, attendance_date, image_path)
+        st.write(st.session_state.present_students)
         # 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.")
-        st.json(present_students)
+        st.json(st.session_state.present_students)
 
 if st.button("Mark Attendance"):
-    if present_students:
+    if st.session_state.present_students:
         # Convert present_students to integers
-        present_students_int = [int(student) for student in present_students]
+        present_students_int = [int(student) for student in st.session_state.present_students]
         csv_contents = record_attendance_new(present_students_int, selected_course)
         st.success("Attendance marked successfully")