Delete app.py
Browse files
app.py
DELETED
|
@@ -1,210 +0,0 @@
|
|
| 1 |
-
import gradio as gr
|
| 2 |
-
from datasets import load_dataset, Dataset
|
| 3 |
-
from datetime import datetime
|
| 4 |
-
from datetime import date
|
| 5 |
-
import requests
|
| 6 |
-
import tempfile
|
| 7 |
-
import asyncio
|
| 8 |
-
from huggingface_hub import upload_file
|
| 9 |
-
from functools import partial
|
| 10 |
-
import io
|
| 11 |
-
import os
|
| 12 |
-
from PIL import Image, ImageDraw, ImageFont
|
| 13 |
-
from huggingface_hub import login
|
| 14 |
-
|
| 15 |
-
login(token=os.environ["HUGGINGFACE_TOKEN"])
|
| 16 |
-
|
| 17 |
-
# Constants
|
| 18 |
-
SCORES_DATASET = "agents-course/unit4-students-scores"
|
| 19 |
-
CERTIFICATES_DATASET = "agents-course/course-certificates-of-excellence"
|
| 20 |
-
THRESHOLD_SCORE = 30
|
| 21 |
-
CERTIFYING_ORG_LINKEDIN_ID = os.getenv("CERTIFYING_ORG_LINKEDIN_ID", "000000")
|
| 22 |
-
COURSE_TITLE = os.getenv("COURSE_TITLE", "Hugging Face Agents Course")
|
| 23 |
-
|
| 24 |
-
# Function to check user score
|
| 25 |
-
def check_user_score(username):
|
| 26 |
-
score_data = load_dataset(SCORES_DATASET, split="train", download_mode="force_redownload")
|
| 27 |
-
matches = [row for row in score_data if row["username"] == username]
|
| 28 |
-
return matches[0] if matches else None
|
| 29 |
-
|
| 30 |
-
# Function to check if certificate entry exists
|
| 31 |
-
def has_certificate_entry(username):
|
| 32 |
-
cert_data = load_dataset(CERTIFICATES_DATASET, split="train", download_mode="force_redownload")
|
| 33 |
-
print(username)
|
| 34 |
-
return any(row["username"] == username for row in cert_data)
|
| 35 |
-
|
| 36 |
-
# Function to add certificate entry
|
| 37 |
-
def add_certificate_entry(username, name, score):
|
| 38 |
-
# Load current dataset
|
| 39 |
-
ds = load_dataset(CERTIFICATES_DATASET, split="train", download_mode="force_redownload")
|
| 40 |
-
|
| 41 |
-
# Remove any existing entry with the same username
|
| 42 |
-
filtered_rows = [row for row in ds if row["username"] != username]
|
| 43 |
-
|
| 44 |
-
# Append the updated/new entry
|
| 45 |
-
new_entry = {
|
| 46 |
-
"username": username,
|
| 47 |
-
"score": score,
|
| 48 |
-
"timestamp": datetime.now().isoformat()
|
| 49 |
-
}
|
| 50 |
-
filtered_rows.append(new_entry)
|
| 51 |
-
|
| 52 |
-
# Rebuild dataset and push
|
| 53 |
-
updated_ds = Dataset.from_list(filtered_rows)
|
| 54 |
-
updated_ds.push_to_hub(CERTIFICATES_DATASET)
|
| 55 |
-
|
| 56 |
-
# Function to generate certificate PDF
|
| 57 |
-
def generate_certificate(name, score):
|
| 58 |
-
"""Generate certificate image and PDF."""
|
| 59 |
-
certificate_path = os.path.join(
|
| 60 |
-
os.path.dirname(__file__), "templates", "certificate.png"
|
| 61 |
-
)
|
| 62 |
-
im = Image.open(certificate_path)
|
| 63 |
-
d = ImageDraw.Draw(im)
|
| 64 |
-
|
| 65 |
-
name_font = ImageFont.truetype("Quattrocento-Regular.ttf", 100)
|
| 66 |
-
date_font = ImageFont.truetype("Quattrocento-Regular.ttf", 48)
|
| 67 |
-
|
| 68 |
-
name = name.title()
|
| 69 |
-
d.text((1000, 740), name, fill="black", anchor="mm", font=name_font)
|
| 70 |
-
|
| 71 |
-
d.text((1480, 1170), str(date.today()), fill="black", anchor="mm", font=date_font)
|
| 72 |
-
|
| 73 |
-
pdf = im.convert("RGB")
|
| 74 |
-
pdf.save("certificate.pdf")
|
| 75 |
-
|
| 76 |
-
return im, "certificate.pdf"
|
| 77 |
-
|
| 78 |
-
async def upload_certificate_to_hub(username: str, certificate_img) -> str:
|
| 79 |
-
"""Upload certificate to the dataset hub and return the URL asynchronously."""
|
| 80 |
-
# Save image to temporary file
|
| 81 |
-
with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
|
| 82 |
-
certificate_img.save(tmp.name)
|
| 83 |
-
|
| 84 |
-
try:
|
| 85 |
-
# Run upload in a thread pool since upload_file is blocking
|
| 86 |
-
loop = asyncio.get_event_loop()
|
| 87 |
-
upload_func = partial(
|
| 88 |
-
upload_file,
|
| 89 |
-
path_or_fileobj=tmp.name,
|
| 90 |
-
path_in_repo=f"certificates/{username}/{date.today()}.png",
|
| 91 |
-
repo_id="agents-course/final-certificates",
|
| 92 |
-
repo_type="dataset",
|
| 93 |
-
token=os.getenv("HF_TOKEN"),
|
| 94 |
-
)
|
| 95 |
-
await loop.run_in_executor(None, upload_func)
|
| 96 |
-
|
| 97 |
-
# Construct the URL to the image
|
| 98 |
-
cert_url = (
|
| 99 |
-
f"https://huggingface.co/datasets/agents-course/final-certificates/"
|
| 100 |
-
f"resolve/main/certificates/{username}/{date.today()}.png"
|
| 101 |
-
)
|
| 102 |
-
|
| 103 |
-
# Clean up temp file
|
| 104 |
-
os.unlink(tmp.name)
|
| 105 |
-
return cert_url
|
| 106 |
-
|
| 107 |
-
except Exception as e:
|
| 108 |
-
print(f"Error uploading certificate: {e}")
|
| 109 |
-
os.unlink(tmp.name)
|
| 110 |
-
return None
|
| 111 |
-
|
| 112 |
-
def create_linkedin_button(username: str, cert_url: str | None) -> str:
|
| 113 |
-
"""Create LinkedIn 'Add to Profile' button HTML."""
|
| 114 |
-
current_year = date.today().year
|
| 115 |
-
current_month = date.today().month
|
| 116 |
-
|
| 117 |
-
# Use the dataset certificate URL if available, otherwise fallback to default
|
| 118 |
-
certificate_url = cert_url or "https://huggingface.co/agents-course-finishers"
|
| 119 |
-
|
| 120 |
-
linkedin_params = {
|
| 121 |
-
"startTask": "CERTIFICATION_NAME",
|
| 122 |
-
"name": COURSE_TITLE,
|
| 123 |
-
"organizationName": "Hugging Face",
|
| 124 |
-
"organizationId": CERTIFYING_ORG_LINKEDIN_ID,
|
| 125 |
-
"issueYear": str(current_year),
|
| 126 |
-
"issueMonth": str(current_month),
|
| 127 |
-
"certUrl": certificate_url,
|
| 128 |
-
"certId": username, # Using username as cert ID
|
| 129 |
-
}
|
| 130 |
-
|
| 131 |
-
# Build the LinkedIn button URL
|
| 132 |
-
base_url = "https://www.linkedin.com/profile/add?"
|
| 133 |
-
params = "&".join(
|
| 134 |
-
f"{k}={requests.utils.quote(v)}" for k, v in linkedin_params.items()
|
| 135 |
-
)
|
| 136 |
-
button_url = base_url + params
|
| 137 |
-
|
| 138 |
-
message = f"""
|
| 139 |
-
<a href="{button_url}" target="_blank" style="display: block; margin: 0 auto; width: fit-content;">
|
| 140 |
-
<img src="https://download.linkedin.com/desktop/add2profile/buttons/en_US.png"
|
| 141 |
-
alt="LinkedIn Add to Profile button"
|
| 142 |
-
style="height: 40px; width: auto; display: block;" />
|
| 143 |
-
</a>
|
| 144 |
-
"""
|
| 145 |
-
return message
|
| 146 |
-
|
| 147 |
-
# Main function to handle certificate generation
|
| 148 |
-
async def handle_certificate(name, profile: gr.OAuthProfile):
|
| 149 |
-
if profile is None:
|
| 150 |
-
return "You must be logged in with your Hugging Face account.", None
|
| 151 |
-
|
| 152 |
-
username = profile.username
|
| 153 |
-
user_score = check_user_score(username)
|
| 154 |
-
|
| 155 |
-
if not user_score:
|
| 156 |
-
return "You need to complete Unit 4 first.", None, None, None
|
| 157 |
-
|
| 158 |
-
score = user_score["score"]
|
| 159 |
-
|
| 160 |
-
if score < THRESHOLD_SCORE:
|
| 161 |
-
return f"Your score is {score}. You need at least {THRESHOLD_SCORE} to pass.", None, None
|
| 162 |
-
|
| 163 |
-
certificate_image, certificate_pdf = generate_certificate(name, score)
|
| 164 |
-
add_certificate_entry(username, name, score)
|
| 165 |
-
|
| 166 |
-
# Start certificate upload asynchronously
|
| 167 |
-
gr.Info("Uploading your certificate...")
|
| 168 |
-
cert_url = await upload_certificate_to_hub(username, certificate_image)
|
| 169 |
-
|
| 170 |
-
if cert_url is None:
|
| 171 |
-
gr.Warning("Certificate upload failed, but you still passed!")
|
| 172 |
-
cert_url = "https://huggingface.co/agents-course"
|
| 173 |
-
|
| 174 |
-
linkedin_button = create_linkedin_button(username, cert_url)
|
| 175 |
-
return "Congratulations! Here's your certificate:", certificate_image, gr.update(value=linkedin_button, visible=True), certificate_pdf
|
| 176 |
-
|
| 177 |
-
|
| 178 |
-
# Gradio interface
|
| 179 |
-
with gr.Blocks() as demo:
|
| 180 |
-
gr.Markdown("# 🎓 Agents Course - Get Your Final Certificate")
|
| 181 |
-
gr.Markdown("Welcome! Follow the steps below to receive your official certificate:")
|
| 182 |
-
gr.Markdown("⚠️ **Note**: Due to high demand, you might experience occasional bugs. If something doesn't work, please try again after a moment!")
|
| 183 |
-
|
| 184 |
-
with gr.Group():
|
| 185 |
-
gr.Markdown("## ✅ How it works")
|
| 186 |
-
gr.Markdown("""
|
| 187 |
-
1. **Sign in** with your Hugging Face account using the button below.
|
| 188 |
-
2. **Enter your full name** (this will appear on the certificate).
|
| 189 |
-
3. Click **'Get My Certificate'** to check your score and download your certificate.
|
| 190 |
-
""")
|
| 191 |
-
gr.Markdown("---")
|
| 192 |
-
gr.Markdown("📝 **Note**: You must have completed [Unit 4](https://huggingface.co/learn/agents-course/unit4/introduction) and your Agent must have scored **above 30** to get your certificate.")
|
| 193 |
-
|
| 194 |
-
gr.LoginButton()
|
| 195 |
-
with gr.Row():
|
| 196 |
-
name_input = gr.Text(label="Enter your name (this will appear on the certificate)")
|
| 197 |
-
generate_btn = gr.Button("Get my certificate")
|
| 198 |
-
output_text = gr.Textbox(label="Result")
|
| 199 |
-
linkedin_btn = gr.HTML(visible=False)
|
| 200 |
-
|
| 201 |
-
cert_image = gr.Image(label="Your Certificate")
|
| 202 |
-
cert_file = gr.File(label="Download Certificate (PDF)", file_types=[".pdf"])
|
| 203 |
-
|
| 204 |
-
generate_btn.click(
|
| 205 |
-
fn=handle_certificate,
|
| 206 |
-
inputs=[name_input],
|
| 207 |
-
outputs=[output_text, cert_image, linkedin_btn, cert_file]
|
| 208 |
-
)
|
| 209 |
-
|
| 210 |
-
demo.launch()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|