Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import csv
 from datetime import date
 from datasets import load_dataset
-from huggingface_hub import HfApi, HfFolder
+from huggingface_hub import HfApi, HfFolder, hf_hub_download
 import os
 
 class AttendanceTracker:
@@ -13,13 +13,14 @@ class AttendanceTracker:
         self.token = HfFolder.get_token()
 
     def load_original_attendance_list(self, course):
-        # Load the CSV file using datasets package
-        dataset = load_dataset(self.dataset_id, data_files={course: f"{course}.csv"},split=f"{course}")
+        # Download the CSV file using datasets package
+        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
+        return attendance_list, local_file_path
 
     def mark_attendance(self, attendance_file, course):
-        original_attendance_list = self.load_original_attendance_list(course)
+        original_attendance_list, local_file_path = self.load_original_attendance_list(course)
 
         # Read attendance file
         attendance_list = []
@@ -38,22 +39,24 @@ class AttendanceTracker:
             if name.split()[0] in attendance_dict:
                 attendance_dict[name.split()[0]] = "Present"
 
-        # Download the dataset
-        dataset = load_dataset(self.dataset_id, data_files={course: f"{course}.csv"}, split=course)
-
-        # Update the rows with attendance data
+        # Load the dataset and prepare updated data
+        dataset = load_dataset(self.dataset_id, data_files={course: local_file_path}, split=course)
         updated_rows = []
+
+        # 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:
-            row = [entry['Name']]
-            if entry['Name'].split()[0].lower() in attendance_dict:
-                row.append(attendance_dict[entry['Name'].split()[0].lower()])
-            else:
-                row.append("Absent")
+            name = entry['Name']
+            row = [name]
+            # Mark attendance
+            row.append(attendance_dict.get(name.split()[0].lower(), "Absent"))
             updated_rows.append(row)
 
         # Save the updated attendance list locally
-        updated_file_path = f"{course}.csv"
-        with open(updated_file_path, "w", newline="") as f:
+        with open(local_file_path, "w", newline="") as f:
             writer = csv.writer(f)
             writer.writerow(["Name", current_date])
             writer.writerows(updated_rows)
@@ -61,7 +64,7 @@ class AttendanceTracker:
         # Upload the updated file back to the Hugging Face dataset
         api = HfApi()
         api.upload_file(
-            path_or_fileobj=updated_file_path,
+            path_or_fileobj=local_file_path,
             path_in_repo=f"{course}.csv",
             repo_id=self.repo_id,
             repo_type="dataset",
@@ -85,4 +88,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()