Update app.py

app.py

@@ -1,31 +1,31 @@
 import gradio as gr
 import csv
 from datetime import date
-from huggingface_hub import HfApi, Repository
+from huggingface_hub import HfApi, Repository, hf_hub_download
 import os
 
-repo_id = "omarkashif/attendanceSystem" 
-repo_type = "space" 
-
 class AttendanceTracker:
-    def __init__(self, repo_id, repo_type="space"):
+    def __init__(self, repo_id, dataset_id, repo_type="space"):
         self.courses = ["AI", "DS", "ML"]  # List of courses
         self.repo_id = repo_id
+        self.dataset_id = dataset_id
         self.repo_type = repo_type
 
         self.token = os.getenv("HF_TOKEN")  # Fetch the token from environment variables
 
     def load_original_attendance_list(self, course):
+        # Download the CSV file from the Hugging Face dataset
+        file_path = hf_hub_download(repo_id=self.dataset_id, filename=f"{course}.csv")
+        
         attendance_list = []
-        file_path = f"{course}.csv"
         with open(file_path, "r") as f:
             reader = csv.reader(f)
             for row in reader:
                 attendance_list.append(row[0].lower())
-        return attendance_list
+        return attendance_list, file_path
 
     def mark_attendance(self, attendance_file, course):
-        original_attendance_list = self.load_original_attendance_list(course)
+        original_attendance_list, file_path = self.load_original_attendance_list(course)
 
         # Read attendance file
         attendance_list = []
@@ -45,7 +45,6 @@ class AttendanceTracker:
                 attendance_dict[name.split()[0]] = "Present"
 
         # Save marked attendance to the original file
-        file_path = f"{course}.csv"
         with open(file_path, "r") as f:
             reader = csv.reader(f)
             rows = list(reader)
@@ -66,13 +65,12 @@ class AttendanceTracker:
             writer = csv.writer(f)
             writer.writerows(rows)
 
-        # Push the updated file to the repository
-        self.push_to_hf(course)
+        # Push the updated file to the Hugging Face dataset
+        self.push_to_hf(course, file_path)
 
         return f"Attendance marked successfully for {course}!"
 
-
-    def push_to_hf(self, course):
+    def push_to_hf(self, course, file_path):
         # Push updated CSV to Hugging Face repository
         api = HfApi()
         repo = Repository(
@@ -82,17 +80,13 @@ class AttendanceTracker:
             use_auth_token=self.token
         )
         
-        file_path = f"{course}.csv"
         repo.git_add(file_path)
         repo.git_commit(f"Update {course} attendance")
         repo.git_push()
 
-
-
-        
-
-repo_id = "omarkashif/attendanceSystem" 
-tracker = AttendanceTracker(repo_id)
+repo_id = "omarkashif/attendanceSystem"
+dataset_id = "omarkashif/attendace-ML"  # Replace with your actual dataset ID
+tracker = AttendanceTracker(repo_id, dataset_id)
 
 def upload_and_mark(file, course):
     return tracker.mark_attendance(file, course)