connect certificate app with dataset

app.py

@@ -2,14 +2,16 @@ import os
 import requests
 from io import BytesIO
 from datetime import date
+import tempfile
 
 import gradio as gr
 from PIL import Image, ImageDraw, ImageFont
-from huggingface_hub import whoami
+from huggingface_hub import whoami, upload_file
 
 from criteria import check_certification as check_certification_criteria
 from org import join_finishers_org
 
+CERTIFYING_ORG_LINKEDIN_ID = os.getenv("CERTIFYING_ORG_LINKEDIN_ID", "000000")
 
 def download_profile_picture(profile_url: str):
     """Download profile picture from URL."""
@@ -66,22 +68,22 @@ def get_user_info(oauth_token):
         return None, None, None, None
 
 
-def create_linkedin_button(username: str) -> str:
+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
 
-    # URL encode the certificate URL
-    cert_url = "https://huggingface.co/agents-course-finishers"
+    # 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": "Hugging Face Course Certificate",
-        # "organizationId": "40479",  # Hugging Face's LinkedIn Organization ID
         "organizationName": "Hugging Face",
-        "issueYear": str(current_year),
+        "organizationId": CERTIFYING_ORG_LINKEDIN_ID,
+        "organizationIdissueYear": str(current_year),
         "issueMonth": str(current_month),
-        "certUrl": cert_url,
+        "certUrl": certificate_url,
         "certId": username,  # Using username as cert ID
     }
 
@@ -100,6 +102,39 @@ def create_linkedin_button(username: str) -> str:
     """
 
 
+def upload_certificate_to_hub(username: str, certificate_img) -> str:
+    """Upload certificate to the dataset hub and return the URL."""
+    # Save image to temporary file
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
+        certificate_img.save(tmp.name)
+
+        try:
+            # Upload file to hub
+            repo_id = "agents-course/certificates"
+            path_in_repo = f"certificates/{username}/{date.today()}.png"
+
+            upload_file(
+                path_or_fileobj=tmp.name,
+                path_in_repo=path_in_repo,
+                repo_id=repo_id,
+                repo_type="dataset",
+            )
+
+            # Construct the URL to the image
+            cert_url = (
+                f"https://huggingface.co/datasets/{repo_id}/resolve/main/{path_in_repo}"
+            )
+
+            # 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 check_certification(token: gr.OAuthToken | None):
     """Check certification status for logged-in user."""
     if token is None:
@@ -127,8 +162,11 @@ def check_certification(token: gr.OAuthToken | None):
             profile_url=profile_url,
         )
 
+        # Upload certificate to hub and get URL
+        cert_url = upload_certificate_to_hub(username, certificate_img)
+
         # Add LinkedIn button for passed certificates
-        linkedin_button = create_linkedin_button(username)
+        linkedin_button = create_linkedin_button(username, cert_url)
         result_message = f"{result.message}\n\n{linkedin_button}"
     else:
         certificate_img = None