Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
 import csv
 from datetime import date
-from datasets import load_dataset
-from huggingface_hub import HfApi, HfFolder, hf_hub_download
+from huggingface_hub import HfApi, HfFolder, hf_hub_download, upload_file
 import os
 
 class AttendanceTracker:
@@ -17,6 +16,20 @@ class AttendanceTracker:
         local_file_path = hf_hub_download(repo_id=self.dataset_id, filename=f"{course}.csv", repo_type="dataset", use_auth_token=self.token)
         return local_file_path
 
+    def get_new_column_name(self, existing_header, current_date):
+        # Determine if the current date is already in the header and generate a new column name if needed
+        if current_date not in existing_header:
+            return current_date
+        
+        # Create a new column name with a suffix
+        base_name = current_date
+        suffix = 1
+        new_column_name = f"{base_name}({suffix})"
+        while new_column_name in existing_header:
+            suffix += 1
+            new_column_name = f"{base_name}({suffix})"
+        return new_column_name
+
     def mark_attendance(self, attendance_file, course):
         local_file_path = self.load_original_attendance_list(course)
         
@@ -27,9 +40,8 @@ class AttendanceTracker:
             header = next(reader)
             for row in reader:
                 if row:
-                    name = row[0]
-                    statuses = row[1:]
-                    existing_data[name] = statuses
+                    name = row[0].lower()  # Convert to lower case for case-insensitive matching
+                    existing_data[name] = row[1:]  # Store all existing columns for this name
 
         # Read the new attendance file
         new_attendance_list = []
@@ -37,27 +49,23 @@ class AttendanceTracker:
             reader = csv.reader(f)
             new_attendance_list = [row[0].lower() for row in reader if row]
 
-        # Get current date
+        # Get current date and prepare the new column name
         current_date = date.today().strftime("%d-%m-%Y")
-
-        # 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"]
+        new_column_name = self.get_new_column_name(header, current_date)
 
         # Prepare updated data including the new date column
         updated_rows = []
         updated_header = header[:]
-        if current_date not in header:
-            updated_header.append(current_date)
+        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)
+            # Append new status for the new column
+            if name in new_attendance_list:
+                statuses.append("Present")
+            else:
+                statuses.append("Absent")
+            updated_rows.append([name] + statuses)
 
         # Save the updated attendance list locally
         with open(local_file_path, "w", newline="") as f:
@@ -66,8 +74,7 @@ class AttendanceTracker:
             writer.writerows(updated_rows)
 
         # Upload the updated file back to the Hugging Face dataset
-        api = HfApi()
-        api.upload_file(
+        upload_file(
             path_or_fileobj=local_file_path,
             path_in_repo=f"{course}.csv",
             repo_id=self.repo_id,
@@ -92,4 +99,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()