Delete app.py
#130
by
Yacinebella
- opened
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()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|