Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
-from datasets import load_dataset, Dataset
+from datasets import load_dataset
 from datetime import datetime
 import io
-import os
+from fpdf import FPDF
 
 # Constants
 SCORES_DATASET = "agents-course/unit4-students-scores"
@@ -31,13 +31,23 @@ def add_certificate_entry(username, name):
     ds = ds.add_item(new_entry)
     ds.push_to_hub(CERTIFICATES_DATASET)
 
-# Function to generate certificate IMG
+# Function to generate certificate PDF
 def generate_certificate(name, score):
-    pdf_output = ""
+    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):
+def handle_certificate(name, profile: gr.OAuthProfile):
     if profile is None:
         return "You must be logged in with your Hugging Face account.", None