Update app.py

app.py

@@ -2,49 +2,57 @@ import gradio as gr
 from datasets import load_dataset, Dataset
 from datetime import datetime
 import io
+from fpdf import FPDF
 import os
 
-# Load the datasets
+# Constants
 SCORES_DATASET = "agents-course/unit4-students-scores"
 CERTIFICATES_DATASET = "agents-course/course-certificates-of-excellence"
-THRESOLD_SCORE = 45
+THRESHOLD_SCORE = 45
 
-# Check the score based on username
+# Function to check user score
 def check_user_score(username):
     score_data = load_dataset(SCORES_DATASET, split="train", download_mode="force_redownload")
     matches = [row for row in score_data if row["username"] == username]
     return matches[0] if matches else None
 
-
-# Check if this user already generated a certificate
+# Function to check if certificate entry exists
 def has_certificate_entry(username):
     cert_data = load_dataset(CERTIFICATES_DATASET, split="train", download_mode="force_redownload")
     return any(row["username"] == username for row in cert_data)
 
-
+# Function to add certificate entry
 def add_certificate_entry(username, name):
-    # Create a new entry
     new_entry = {
         "username": username,
         "name": name,
         "date": datetime.now().strftime("%Y-%m-%d"),
     }
-
-    # Download the dataset, append and push
     ds = load_dataset(CERTIFICATES_DATASET, split="train")
-    updated = ds.add_item(new_entry)
-    updated.push_to_hub(CERTIFICATES_DATASET)
+    ds = ds.add_item(new_entry)
+    ds.push_to_hub(CERTIFICATES_DATASET)
 
+# Function to generate certificate PDF
 def generate_certificate(name, score):
-    pass
-
-
-def handle_certificate(name, request: gr.Request):
-    username = request.username
-
-    if not username:
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=24)
+    pdf.cell(200, 20, txt="Certificate of Excellence", ln=True, align="C")
+    pdf.set_font("Arial", size=16)
+    pdf.cell(200, 10, txt=f"Presented to {name}", ln=True, align="C")
+    pdf.cell(200, 10, txt=f"For passing Unit 4 with a score of {score}", ln=True, align="C")
+    pdf.cell(200, 10, txt=f"Date: {datetime.now().strftime('%Y-%m-%d')}", ln=True, align="C")
+    pdf_output = io.BytesIO()
+    pdf.output(pdf_output)
+    pdf_output.seek(0)
+    return pdf_output
+
+# Main function to handle certificate generation
+def handle_certificate(name, profile: gr.OAuthProfile | None):
+    if profile is None:
         return "You must be logged in with your Hugging Face account.", None
 
+    username = profile.username
     user_score = check_user_score(username)
 
     if not user_score:
@@ -52,18 +60,19 @@ def handle_certificate(name, request: gr.Request):
 
     score = user_score["score"]
 
-    if score < THRESOLD_SCORE:
-        return f"Your score is {score}. You need at least {THRESOLD_SCORE} to pass.", None
+    if score < THRESHOLD_SCORE:
+        return f"Your score is {score}. You need at least {THRESHOLD_SCORE} to pass.", None
 
-    # Passed: check if already in certificate dataset
     if not has_certificate_entry(username):
         add_certificate_entry(username, name)
 
     certificate = generate_certificate(name, score)
     return "Congratulations! Here's your certificate:", certificate
 
-with gr.Blocks(auth=True) as demo:
+# Gradio interface
+with gr.Blocks() as demo:
     gr.Markdown("# 🎓 Unit 4 Certificate Generator")
+    gr.LoginButton()
     with gr.Row():
         name_input = gr.Text(label="Enter your name")
     generate_btn = gr.Button("Get my certificate")
@@ -72,8 +81,8 @@ with gr.Blocks(auth=True) as demo:
 
     generate_btn.click(
         fn=handle_certificate,
-        inputs=[name_input],
+        inputs=[name_input, gr.OAuthProfile()],
         outputs=[output_text, cert_file]
     )
 
-demo.launch()
+demo.launch()