Add to profile LinkedIn button

app.py

@@ -2,6 +2,11 @@ import gradio as gr
 from datasets import load_dataset, Dataset
 from datetime import datetime
 from datetime import date
+import requests
+import tempfile
+import asyncio
+from huggingface_hub import upload_file
+from functools import partial
 import io
 import os
 from PIL import Image, ImageDraw, ImageFont
@@ -13,6 +18,8 @@ login(token=os.environ["HUGGINGFACE_TOKEN"])
 SCORES_DATASET = "agents-course/unit4-students-scores"
 CERTIFICATES_DATASET = "agents-course/course-certificates-of-excellence"
 THRESHOLD_SCORE = 30
+CERTIFYING_ORG_LINKEDIN_ID = os.getenv("CERTIFYING_ORG_LINKEDIN_ID", "000000")
+COURSE_TITLE = os.getenv("COURSE_TITLE", "Hugging Face Agents Course")
 
 # Function to check user score
 def check_user_score(username):
@@ -68,8 +75,77 @@ def generate_certificate(name, score):
 
     return im, "certificate.pdf"
 
+async def upload_certificate_to_hub(username: str, certificate_img) -> str:
+    """Upload certificate to the dataset hub and return the URL asynchronously."""
+    # Save image to temporary file
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
+        certificate_img.save(tmp.name)
+
+        try:
+            # Run upload in a thread pool since upload_file is blocking
+            loop = asyncio.get_event_loop()
+            upload_func = partial(
+                upload_file,
+                path_or_fileobj=tmp.name,
+                path_in_repo=f"certificates/{username}/{date.today()}.png",
+                repo_id="agents-course/final-certificates",
+                repo_type="dataset",
+                token=os.getenv("HF_TOKEN"),
+            )
+            await loop.run_in_executor(None, upload_func)
+
+            # Construct the URL to the image
+            cert_url = (
+                f"https://huggingface.co/datasets/agents-course/final-certificates/"
+                f"resolve/main/certificates/{username}/{date.today()}.png"
+            )
+
+            # Clean up temp file
+            os.unlink(tmp.name)
+            return cert_url
+
+        except Exception as e:
+            print(f"Error uploading certificate: {e}")
+            os.unlink(tmp.name)
+            return None
+
+def create_linkedin_button(username: str, cert_url: str | None) -> str:
+    """Create LinkedIn 'Add to Profile' button HTML."""
+    current_year = date.today().year
+    current_month = date.today().month
+
+    # Use the dataset certificate URL if available, otherwise fallback to default
+    certificate_url = cert_url or "https://huggingface.co/agents-course-finishers"
+
+    linkedin_params = {
+        "startTask": "CERTIFICATION_NAME",
+        "name": COURSE_TITLE,
+        "organizationName": "Hugging Face",
+        "organizationId": CERTIFYING_ORG_LINKEDIN_ID,
+        "issueYear": str(current_year),
+        "issueMonth": str(current_month),
+        "certUrl": certificate_url,
+        "certId": username,  # Using username as cert ID
+    }
+
+    # Build the LinkedIn button URL
+    base_url = "https://www.linkedin.com/profile/add?"
+    params = "&".join(
+        f"{k}={requests.utils.quote(v)}" for k, v in linkedin_params.items()
+    )
+    button_url = base_url + params
+
+    message = f"""
+        <a href="{button_url}" target="_blank" style="display: block; margin: 0 auto; width: fit-content;">
+            <img src="https://download.linkedin.com/desktop/add2profile/buttons/en_US.png" 
+                alt="LinkedIn Add to Profile button"
+                style="height: 40px; width: auto; display: block;" />
+        </a>
+    """
+    return message
+
 # Main function to handle certificate generation
-def handle_certificate(name, profile: gr.OAuthProfile):
+async def handle_certificate(name, profile: gr.OAuthProfile):
     if profile is None:
         return "You must be logged in with your Hugging Face account.", None
 
@@ -86,7 +162,18 @@ def handle_certificate(name, profile: gr.OAuthProfile):
 
     certificate_image, certificate_pdf = generate_certificate(name, score)
     add_certificate_entry(username, name, score)
-    return "Congratulations! Here's your certificate:", certificate_image, certificate_pdf
+    
+    # Start certificate upload asynchronously
+    gr.Info("Uploading your certificate...")
+    cert_url = await upload_certificate_to_hub(username, certificate_image)
+
+    if cert_url is None:
+        gr.Warning("Certificate upload failed, but you still passed!")
+        cert_url = "https://huggingface.co/agents-course"
+
+    linkedin_button = create_linkedin_button(username, cert_url)
+    return "Congratulations! Here's your certificate:", certificate_image, gr.update(value=linkedin_button, visible=True), certificate_pdf
+    
 
 # Gradio interface    
 with gr.Blocks() as demo:
@@ -109,13 +196,15 @@ with gr.Blocks() as demo:
         name_input = gr.Text(label="Enter your name (this will appear on the certificate)")
     generate_btn = gr.Button("Get my certificate")
     output_text = gr.Textbox(label="Result")
+    linkedin_btn = gr.HTML(visible=False)
+
     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_image, cert_file]
+        outputs=[output_text, cert_image, linkedin_btn, cert_file]
     )
 
 demo.launch()