Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from datasets import load_dataset
 from datetime import datetime
 import io
-
+from PIL import Image, ImageDraw, ImageFont
 
 # Constants
 SCORES_DATASET = "agents-course/unit4-students-scores"
@@ -33,8 +33,25 @@ def add_certificate_entry(username, name):
 
 # Function to generate certificate PDF
 def generate_certificate(name, score):
-    pdf_output = ""
-    return pdf_output
+    """Generate certificate image and PDF."""
+    certificate_path = os.path.join(
+        os.path.dirname(__file__), "templates", "certificate.png"
+    )
+    im = Image.open(certificate_path)
+    d = ImageDraw.Draw(im)
+
+    name_font = ImageFont.truetype("Quattrocento-Regular.ttf", 100)
+    date_font = ImageFont.truetype("Quattrocento-Regular.ttf", 48)
+
+    name = name.title()
+    d.text((1000, 740), name, fill="black", anchor="mm", font=name_font)
+
+    d.text((1480, 1170), str(date.today()), fill="black", anchor="mm", font=date_font)
+
+    pdf = im.convert("RGB")
+    pdf.save("certificate.pdf")
+
+    return im, "certificate.pdf"
 
 # Main function to handle certificate generation
 def handle_certificate(name, profile: gr.OAuthProfile):
@@ -66,12 +83,13 @@ with gr.Blocks() as demo:
         name_input = gr.Text(label="Enter your name")
     generate_btn = gr.Button("Get my certificate")
     output_text = gr.Textbox(label="Result")
-    cert_file = gr.File(label="Your Certificate", file_types=[".pdf"])
+    cert_image = gr.Image(label="Your Certificate")
+    cert_file = gr.File(label="Download Certificate (PDF)", file_types=[".pdf"])
 
     generate_btn.click(
         fn=handle_certificate,
         inputs=[name_input],
-        outputs=[output_text, cert_file]
+        outputs=[output_text, cert_image, cert_file]
     )
 
 demo.launch()