Update app.py

app.py

@@ -1,24 +1,27 @@
 import gradio as gr
-from gradio import Interface
 import csv
 from datetime import date
+from huggingface_hub import HfApi, Repository
+import os
 
 class AttendanceTracker:
-    def __init__(self):
-        self.original_attendance_list = []
-        self.attendance_file_path = ""
+    def __init__(self, repo_id):
+        self.courses = ["AI", "DS", "ML"]  # List of courses
+        self.repo_id = repo_id
+        self.token = os.getenv("HF_TOKEN")  # Fetch the token from environment variables
 
-        # Load original attendance list
-        self.load_original_attendance_list()
-
-    def load_original_attendance_list(self):
-        # Load original attendance list from a file (e.g. original_attendance_list.csv)
-        with open("original_attendance.csv", "r") as f:
+    def load_original_attendance_list(self, course):
+        attendance_list = []
+        file_path = f"attendance/{course}.csv"
+        with open(file_path, "r") as f:
             reader = csv.reader(f)
             for row in reader:
-                self.original_attendance_list.append(row[0].lower())
+                attendance_list.append(row[0].lower())
+        return attendance_list
+
+    def mark_attendance(self, attendance_file, course):
+        original_attendance_list = self.load_original_attendance_list(course)
 
-    def mark_attendance(self, attendance_file):
         # Read attendance file
         attendance_list = []
         with open(attendance_file.name, "r") as f:
@@ -29,7 +32,7 @@ class AttendanceTracker:
         current_date = date.today().strftime("%d-%m-%Y")
 
         # Create a dictionary to store attendance
-        attendance_dict = {name.split()[0]: "Absent" for name in self.original_attendance_list if len(name.split()) > 0}
+        attendance_dict = {name.split()[0]: "Absent" for name in original_attendance_list if len(name.split()) > 0}
 
         # Mark present students
         for name in attendance_list:
@@ -37,7 +40,8 @@ class AttendanceTracker:
                 attendance_dict[name.split()[0]] = "Present"
 
         # Save marked attendance to the original file
-        with open("original_attendance.csv", "r") as f:
+        file_path = f"attendance/{course}.csv"
+        with open(file_path, "r") as f:
             reader = csv.reader(f)
             rows = list(reader)
 
@@ -46,29 +50,44 @@ class AttendanceTracker:
 
         # Update the rows with attendance data
         for row in range(1, len(rows)):
-            if rows[row] and len(rows[row][0].split()) > 0 and rows[row][0].split()[0].lower() in attendance_dict:
-                rows[row].append(attendance_dict[rows[row][0].split()[0].lower()])
-            else:
-                rows[row].append("Absent")
+            if rows[row] and len(rows[row][0].split()) > 0:
+                if rows[row][0].split()[0].lower() in attendance_dict:
+                    rows[row].append(attendance_dict[rows[row][0].split()[0].lower()])
+                else:
+                    rows[row].append("Absent")
 
         # Save the updated attendance list
-        with open("original_attendance.csv", "w", newline="") as f:
+        with open(file_path, "w", newline="") as f:
             writer = csv.writer(f)
             writer.writerows(rows)
 
-        return "Attendance marked successfully!"
+        # Push the updated file to the repository
+        self.push_to_hf(course)
+
+        return f"Attendance marked successfully for {course}!"
+
+    def push_to_hf(self, course):
+        # Push updated CSV to Hugging Face repository
+        api = HfApi()
+        repo = Repository(local_dir=".", clone_from=self.repo_id, use_auth_token=self.token)
+        
+        file_path = f"attendance/{course}.csv"
+        repo.git_add(file_path)
+        repo.git_commit(f"Update {course} attendance")
+        repo.git_push()
 
-tracker = AttendanceTracker()
+repo_id = "omarkashif/attendanceSystem" 
+tracker = AttendanceTracker(repo_id)
 
-def upload_and_mark(file):
-    return tracker.mark_attendance(file)
+def upload_and_mark(file, course):
+    return tracker.mark_attendance(file, course)
 
-iface = Interface(
+iface = gr.Interface(
     fn=upload_and_mark,
-    inputs=gr.File(label="Upload Attendance File (CSV)"),
+    inputs=[gr.File(label="Upload Attendance File (CSV)"), gr.Dropdown(choices=["AI", "DS", "ML"], label="Select Course")],
     outputs=gr.Textbox(label="Status"),
     title="Attendance Tracker",
-    description="Upload a CSV file of today's attendance to update the original attendance file."
+    description="Upload a CSV file of today's attendance to update the attendance file for the selected course."
 )
 
 iface.launch()