Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import csv
 from datetime import date
-from datasets import load_dataset
 from huggingface_hub import HfApi, HfFolder, hf_hub_download
 import os
 
@@ -35,11 +34,15 @@ class AttendanceTracker:
         local_file_path = self.load_original_attendance_list(course)
         
         # Load existing data from CSV
-        existing_data = []
+        existing_data = {}
         with open(local_file_path, "r") as f:
             reader = csv.reader(f)
             header = next(reader)
-            existing_data = list(reader)
+            for row in reader:
+                if row:
+                    name = row[0].lower()  # Convert to lower case for case-insensitive matching
+                    statuses = row[1:]
+                    existing_data[name] = statuses
 
         # Read the new attendance file
         new_attendance_list = []
@@ -51,34 +54,31 @@ class AttendanceTracker:
         current_date = date.today().strftime("%d-%m-%Y")
         new_column_name = self.get_new_column_name(header, current_date)
 
-        # Update header with new column name if it does not exist
-        if new_column_name not in header:
-            header.append(new_column_name)
-
         # Update existing records with new attendance data
-        name_to_index = {row[0].lower(): i for i, row in enumerate(existing_data)}
-        
-        for row in existing_data:
-            name = row[0].lower()
-            if name in new_attendance_list:
-                # Mark present in the new column
-                row.append("Present")
-            else:
-                # Mark absent in the new column
-                row.append("Absent")
-        
-        # Append new rows for any new names in the attendance list that weren't previously in the file
         for name in new_attendance_list:
-            if name not in name_to_index:
-                new_row = [name] + ["Absent"] * (len(header) - 1)  # Default to "Absent" for previous dates
-                new_row.append("Present")  # Mark present for the new column
-                existing_data.append(new_row)
+            if name in existing_data:
+                # Append new status for the new column
+                existing_data[name].append("Present")
+            else:
+                # Add status "Absent" for the new date for students not in the new attendance list
+                for student_name in existing_data:
+                    existing_data[student_name].append("Absent")
+
+        # Prepare updated data including the new date column
+        updated_rows = []
+        updated_header = header[:]
+        if new_column_name not in updated_header:
+            updated_header.append(new_column_name)
+
+        for name, statuses in existing_data.items():
+            row = [name] + statuses
+            updated_rows.append(row)
 
         # Save the updated attendance list locally
         with open(local_file_path, "w", newline="") as f:
             writer = csv.writer(f)
-            writer.writerow(header)
-            writer.writerows(existing_data)
+            writer.writerow(updated_header)
+            writer.writerows(updated_rows)
 
         # Upload the updated file back to the Hugging Face dataset
         api = HfApi()
@@ -107,4 +107,4 @@ iface = gr.Interface(
     description="Upload a CSV file of today's attendance to update the attendance file for the selected course."
 )
 
-iface.launch()
+iface.launch()