add custom name button

app.py

@@ -21,9 +21,7 @@ def download_profile_picture(profile_url: str):
     return Image.open(BytesIO(response.content))
 
 
-def generate_certificate(
-    certificate_path: str, first_name: str, last_name: str, profile_url: str
-):
+def generate_certificate(certificate_path: str, name: str, profile_url: str):
     """Generate certificate image and PDF."""
     im = Image.open(certificate_path)
     d = ImageDraw.Draw(im)
@@ -31,8 +29,6 @@ def generate_certificate(
     name_font = ImageFont.truetype("Quattrocento-Regular.ttf", 100)
     date_font = ImageFont.truetype("Quattrocento-Regular.ttf", 48)
 
-    name = f"{first_name} {last_name}"
-
     # Capitalize first letter of each name
     name = name.title()
 
@@ -133,25 +129,26 @@ def upload_certificate_to_hub(username: str, certificate_img) -> str:
             return None
 
 
-def check_certification(token: gr.OAuthToken | None, profile: gr.OAuthProfile | None):
+def check_certification(
+    token: gr.OAuthToken | None,
+    profile: gr.OAuthProfile | None,
+    custom_name: str | None = None,
+):
     """Check certification status for logged-in user."""
     if token is None or profile is None:
         gr.Warning("Please log in to Hugging Face before checking certification!")
-        return None, None, None, gr.Row.update(visible=False)
+        return None, None, gr.Row.update(visible=False)
 
     username = profile.username
-    first_name = profile.name.split(sep=" ")[0]
-    try:
-        last_name = " ".join(profile.name.split(sep=" ")[1:])
-    except IndexError:
-        last_name = ""
+
+    # Use custom name if provided, otherwise fall back to profile name
+    name = custom_name.strip() if custom_name and custom_name.strip() else profile.name
     profile_url = profile.picture
 
     if not username:
         return (
             "Please login with your Hugging Face account to check certification status",
             None,
-            None,
             gr.Row.update(visible=False),
         )
 
@@ -163,8 +160,7 @@ def check_certification(token: gr.OAuthToken | None, profile: gr.OAuthProfile |
     if result.passed and result.certificate_path:
         certificate_img, pdf_path = generate_certificate(
             certificate_path=result.certificate_path,
-            first_name=first_name,
-            last_name=last_name,
+            name=name,
             profile_url=profile_url,
         )
 
@@ -193,12 +189,20 @@ with gr.Blocks() as demo:
     
     To earn this certificate, you need to complete <a href="https://hf.co/learn/agents-course/unit1/introduction" alt="Agent Course Unit 1"/>Unit 1 of the Agents Course</a> and **pass 80% of the final quiz**.
 
-    Once you receive your certificate, don’t hesitate to share it on X (and tag @huggingface) as well as on LinkedIn so that we can congratulate you.
+    Once you receive your certificate, don't hesitate to share it on X (and tag @huggingface) as well as on LinkedIn so that we can congratulate you.
     """
     )
 
-    # Add login button
-    gr.LoginButton().activate()
+    with gr.Row():
+        # Add login button
+        gr.LoginButton()
+
+    # Add optional name input
+    custom_name_input = gr.Textbox(
+        label="Custom Name (Optional)",
+        placeholder="Enter your name as you want it to appear on the certificate",
+        info="Leave empty to use your Hugging Face profile name",
+    )
 
     check_progress_button = gr.Button(value="Get My Certificate")
 
@@ -207,6 +211,7 @@ with gr.Blocks() as demo:
 
     check_progress_button.click(
         fn=check_certification,
+        inputs=[custom_name_input],
         outputs=[output_text, output_img],
     ).then(
         fn=join_finishers_org,