add custom name functionality

app.py

@@ -46,32 +46,32 @@ def on_user_logged_in(token: gr.OAuthToken | None):
     if token is not None:
         return [
             gr.update(visible=False),  # login_btn
-            gr.update(visible=True),   # start_btn
+            gr.update(visible=True),  # start_btn
             gr.update(visible=False),  # next_btn
             gr.update(visible=False),  # submit_btn
-            "",                        # question_text
+            "",  # question_text
             gr.update(choices=[], visible=False),  # radio_choices
             "Click 'Start' to begin the quiz",  # status_text
-            0,                         # question_idx
-            [],                        # user_answers
+            0,  # question_idx
+            [],  # user_answers
             gr.update(visible=False),  # certificate_img
             gr.update(visible=False),  # linkedin_btn
-            token,                     # user_token
+            token,  # user_token
         ]
     else:
         return [
-            gr.update(visible=True),   # login_btn
+            gr.update(visible=True),  # login_btn
             gr.update(visible=False),  # start_btn
             gr.update(visible=False),  # next_btn
             gr.update(visible=False),  # submit_btn
-            "",                        # question_text
+            "",  # question_text
             gr.update(choices=[], visible=False),  # radio_choices
-            "",                        # status_text
-            0,                         # question_idx
-            [],                        # user_answers
+            "",  # status_text
+            0,  # question_idx
+            [],  # user_answers
             gr.update(visible=False),  # certificate_img
             gr.update(visible=False),  # linkedin_btn
-            None,                      # user_token
+            None,  # user_token
         ]
 
 
@@ -168,7 +168,10 @@ async def upload_certificate_to_hub(username: str, certificate_img) -> str:
 
 
 async def push_results_to_hub(
-    user_answers, token: gr.OAuthToken | None, profile: gr.OAuthProfile | None
+    user_answers,
+    custom_name: str | None,
+    token: gr.OAuthToken | None,
+    profile: gr.OAuthProfile | None,
 ):
     """Handle quiz completion and certificate generation."""
     if token is None or profile is None:
@@ -177,6 +180,7 @@ async def push_results_to_hub(
             gr.update(visible=True, value="Please login first"),
             gr.update(visible=False),
             gr.update(visible=False),
+            gr.update(visible=False),  # hide custom name input
         )
 
     # Calculate grade
@@ -186,19 +190,21 @@ async def push_results_to_hub(
 
     if grade < float(EXAM_PASSING_SCORE):
         return (
-            gr.update(
-                visible=True,
-                value=f"You scored {grade:.1%}. Please try again to achieve at least "
-                f"{float(EXAM_PASSING_SCORE):.1%}",
-            ),
+            gr.update(visible=True, value=f"You scored {grade:.1%}..."),
             gr.update(visible=False),
             gr.update(visible=False),
+            gr.update(visible=False),  # hide custom name input
         )
 
     try:
+        # Use custom name if provided, otherwise use profile name
+        name = (
+            custom_name.strip() if custom_name and custom_name.strip() else profile.name
+        )
+
         # Generate certificate
         certificate_img, _ = generate_certificate(
-            name=profile.name, profile_url=profile.picture
+            name=name, profile_url=profile.picture
         )
 
         # Start certificate upload asynchronously
@@ -222,16 +228,16 @@ async def push_results_to_hub(
             gr.update(visible=True, value=result_message),
             gr.update(visible=True, value=certificate_img),
             gr.update(visible=True),
+            gr.update(visible=True),  # show custom name input
         )
 
     except Exception as e:
         print(f"Error generating certificate: {e}")
         return (
-            gr.update(
-                visible=True, value=f"🎉 Congratulations! You passed with {grade:.1%}!"
-            ),
+            gr.update(visible=True, value=f"🎉 You passed with {grade:.1%}!"),
             gr.update(visible=False),
             gr.update(visible=False),
+            gr.update(visible=False),  # hide custom name input
         )
 
 
@@ -359,6 +365,15 @@ with gr.Blocks() as demo:
         next_btn = gr.Button("Next ⏭️", visible=False)
         submit_btn = gr.Button("🎓 Get your certificate", visible=False)
 
+    with gr.Row(variant="panel"):
+        custom_name_input = gr.Textbox(
+            label="Custom Name for Certificate",
+            placeholder="Enter name as you want it to appear on the certificate",
+            info="Leave empty to use your Hugging Face profile name",
+            visible=False,
+            value=None,
+        )
+
     # Wire up the event handlers
     login_btn.click(
         fn=on_user_logged_in,
@@ -415,8 +430,22 @@ with gr.Blocks() as demo:
 
     submit_btn.click(
         fn=push_results_to_hub,
-        inputs=[user_answers],
-        outputs=[status_text, certificate_img, linkedin_btn],
+        inputs=[
+            user_answers,
+            custom_name_input,
+        ],
+        outputs=[
+            status_text,
+            certificate_img,
+            linkedin_btn,
+            custom_name_input,
+        ],
+    )
+
+    custom_name_input.submit(
+        fn=push_results_to_hub,
+        inputs=[user_answers, custom_name_input],
+        outputs=[status_text, certificate_img, linkedin_btn, custom_name_input],
     )
 
 if __name__ == "__main__":