Yacinebella commited on
Commit
f12a981
·
verified ·
1 Parent(s): fce60a3

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -210
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()