Merge branch 'main' of https://huggingface.co/spaces/agents-course/unit_1_quiz

app.py

@@ -2,6 +2,10 @@ import os
 from datetime import datetime
 import random
 
+import pandas as pd
+from huggingface_hub import HfApi, hf_hub_download, Repository
+from huggingface_hub.repocard import metadata_load
+
 import gradio as gr
 from datasets import load_dataset, Dataset
 from huggingface_hub import whoami
@@ -12,6 +16,13 @@ EXAM_PASSING_SCORE = os.getenv("EXAM_PASSING_SCORE") or 0.7
 
 ds = load_dataset(EXAM_DATASET_ID, split="train")
 
+DATASET_REPO_URL = "https://huggingface.co/datasets/agents-course/certificates"
+CERTIFIED_USERS_FILENAME = "certified_students.csv"
+CERTIFIED_USERS_DIR = "certificates"
+repo = Repository(
+    local_dir=CERTIFIED_USERS_DIR, clone_from=DATASET_REPO_URL, use_auth_token=os.getenv("HF_TOKEN")
+)
+
 # Convert dataset to a list of dicts and randomly sort
 quiz_data = ds.to_pandas().to_dict("records")
 random.shuffle(quiz_data)
@@ -55,6 +66,24 @@ def on_user_logged_in(token: gr.OAuthToken | None):
             None,  # no token
         ]
 
+def add_certified_user(hf_username, pass_percentage, submission_time):
+  """
+  Add the certified user to the database
+  """
+  print("ADD CERTIFIED USER")
+  repo.git_pull()
+  history = pd.read_csv(os.path.join(CERTIFIED_USERS_DIR, CERTIFIED_USERS_FILENAME))
+
+  # Check if this hf_username is already in our dataset:
+  check = history.loc[history['hf_username'] == hf_username]
+  if not check.empty:
+    history = history.drop(labels=check.index[0], axis=0)
+  
+  new_row = pd.DataFrame({'hf_username': hf_username, 'pass_percentage': pass_percentage, 'datetime': submission_time}, index=[0])
+  history = pd.concat([new_row, history[:]]).reset_index(drop=True)
+    
+  history.to_csv(os.path.join(CERTIFIED_USERS_DIR, CERTIFIED_USERS_FILENAME), index=False)
+  repo.push_to_hub(commit_message="Update certified users list")
 
 def push_results_to_hub(user_answers, token: gr.OAuthToken | None):
     """
@@ -92,9 +121,18 @@ def push_results_to_hub(user_answers, token: gr.OAuthToken | None):
     )
     sanitized_name = user_info["name"].replace("-", "000")
     new_ds.push_to_hub(repo_id=repo_id, split=sanitized_name)
+
+    # I'm adding a csv version
+    # The idea, if the user passed, we create a simple row in a csv
+    print("ADD CERTIFIED USER")
+    # Add this user to our database
+    add_certified_user(sanitized_name, grade, submission_time)
+    
     return f"Your responses have been submitted to the Hub! Final grade: {grade:.1%}"
 
 
+
+
 def handle_quiz(question_idx, user_answers, selected_answer, is_start):
     """
     Handle quiz state transitions and store answers
@@ -126,7 +164,7 @@ def handle_quiz(question_idx, user_answers, selected_answer, is_start):
         return [
             "",  # question_text
             gr.update(choices=[], visible=False),  # hide radio choices
-            f"{'✅ Passed!' if grade >= float(EXAM_PASSING_SCORE) else '❌ Did not pass'}",
+            f"{'🎉 Passed! Click now on ✅ Submit to save your exam score!'  if grade >= float(EXAM_PASSING_SCORE) else '❌ Did not pass'}",
             question_idx,
             user_answers,
             gr.update(visible=False),  # start button visibility