Update app.py

app.py

@@ -13,52 +13,56 @@ class AttendanceTracker:
         self.token = HfFolder.get_token()
 
     def load_original_attendance_list(self, course):
-        # Download the CSV file using datasets package
+        # Download the CSV file using hf_hub_download
         local_file_path = hf_hub_download(repo_id=self.dataset_id, filename=f"{course}.csv", repo_type="dataset", use_auth_token=self.token)
-        dataset = load_dataset(self.dataset_id, data_files={course: local_file_path}, split=course)
-        attendance_list = [entry['Name'].lower() for entry in dataset]
-        return attendance_list, local_file_path
+        return local_file_path
 
     def mark_attendance(self, attendance_file, course):
-        original_attendance_list, local_file_path = self.load_original_attendance_list(course)
-
-        # Read attendance file
-        attendance_list = []
+        local_file_path = self.load_original_attendance_list(course)
+        
+        # Load existing data from CSV
+        existing_data = {}
+        with open(local_file_path, "r") as f:
+            reader = csv.reader(f)
+            header = next(reader)
+            for row in reader:
+                if row:
+                    name = row[0]
+                    statuses = row[1:]
+                    existing_data[name] = statuses
+
+        # Read the new attendance file
+        new_attendance_list = []
         with open(attendance_file.name, "r") as f:
             reader = csv.reader(f)
-            attendance_list = [row[0].lower() for row in reader if row]
+            new_attendance_list = [row[0].lower() for row in reader if row]
 
         # Get current date
         current_date = date.today().strftime("%d-%m-%Y")
 
-        # Create a dictionary to store attendance
-        attendance_dict = {name.split()[0]: "Absent" for name in original_attendance_list if len(name.split()) > 0}
+        # Update existing records with new attendance data
+        for name in new_attendance_list:
+            name_lower = name.lower()
+            if name_lower in existing_data:
+                existing_data[name_lower].append("Present")
+            else:
+                # Add new record for new student
+                existing_data[name_lower] = ["Absent"] * (len(header) - 1) + ["Present"]
 
-        # Mark present students
-        for name in attendance_list:
-            if name.split()[0] in attendance_dict:
-                attendance_dict[name.split()[0]] = "Present"
-
-        # Load the dataset and prepare updated data
-        dataset = load_dataset(self.dataset_id, data_files={course: local_file_path}, split=course)
+        # Prepare updated data including the new date column
         updated_rows = []
+        updated_header = header[:]
+        if current_date not in header:
+            updated_header.append(current_date)
 
-        # Check if the dataset already has a column for today's date
-        existing_columns = dataset.column_names
-        if current_date not in existing_columns:
-            existing_columns.append(current_date)
-
-        for entry in dataset:
-            name = entry['Name']
-            row = [name]
-            # Mark attendance
-            row.append(attendance_dict.get(name.split()[0].lower(), "Absent"))
+        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(["Name", current_date])
+            writer.writerow(updated_header)
             writer.writerows(updated_rows)
 
         # Upload the updated file back to the Hugging Face dataset