MariaK's picture
Update app.py
f7db876
raw
history blame
7.86 kB
import gradio as gr
from huggingface_hub import HfApi, hf_hub_download, Repository
from huggingface_hub.repocard import metadata_load
from gradio_client import Client
from PIL import Image, ImageDraw, ImageFont
from datetime import date
import time
import os
import pandas as pd
import json
api = HfApi()
HF_TOKEN = os.environ.get("HF_TOKEN")
# Private dataset repo containing the list of already certified users
DATASET_REPO_URL = "https://huggingface.co/datasets/MariaK/audio-course"
CERTIFIED_USERS_FILENAME = "usernames.csv"
# Private space to check if a user has passed.
SPACE_ID = "MariaK/Check-Audio-Course-Progress"
def check_if_passed(username):
"""
Check if given user passed enough assignments
:param username: User HF username
"""
passed = False
certificate_type = ""
client = Client(SPACE_ID, hf_token=HF_TOKEN)
result = client.predict(username, fn_index=0)
with open(result) as json_data:
data = json.load(json_data)
df = pd.DataFrame(data['data'])
if len(df[df.iloc[:,0] == 'βœ…']) == 4:
passed = True
certificate_type = "excellence"
elif len(df[df.iloc[:,0] == 'βœ…']) == 3:
passed = True
certificate_type = "completion"
return passed, certificate_type
def generate_certificate(certificate_template, first_name, last_name):
"""
Generates certificate from the template
:param certificate_template: type of the certificate to generate
:param first_name: first name entered by user
:param last_name: last name entered by user
"""
im = Image.open(certificate_template)
d = ImageDraw.Draw(im)
name_font = ImageFont.truetype("Quattrocento-Regular.ttf", 100)
date_font = ImageFont.truetype("Quattrocento-Regular.ttf", 48)
name = str(first_name) + " " + str(last_name)
print("NAME", name)
# Debug line name
#d.line(((200, 740), (1800, 740)), "gray")
#d.line(((1000, 0), (1000, 1400)), "gray")
# Name
d.text((1000, 740), name, fill="black", anchor="mm", font=name_font)
# Debug line date
#d.line(((1500, 0), (1500, 1400)), "gray")
# Date of certification
d.text((1480, 1170), str(date.today()), fill="black", anchor="mm", font=date_font)
pdf = im.convert('RGB')
pdf.save('certificate.pdf')
return im, "./certificate.pdf"
def add_certified_user(hf_username, first_name, last_name, certificate_type):
"""
Add the certified user to the database
"""
print("ADD CERTIFIED USER")
repo = Repository(local_dir="usernames", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN)
repo.git_pull()
history = pd.read_csv(os.path.join("usernames", CERTIFIED_USERS_FILENAME))
# Check if this hf_username is already in our dataset:
check = history.loc[history['hf_username'] == hf_username]
if not check.empty:
history = history.drop(labels=check.index[0], axis=0)
new_row = pd.DataFrame({'hf_username': hf_username, 'first_name': first_name, 'last_name': last_name, 'certificate_type': certificate_type, 'datetime': time.time()}, index=[0])
history = pd.concat([new_row, history[:]]).reset_index(drop=True)
history.to_csv(os.path.join("usernames", CERTIFIED_USERS_FILENAME), index=False)
repo.push_to_hub(commit_message="Update certified users list")
def create_certificate(passed, certificate_type, hf_username, first_name, last_name):
"""
Generates certificate, adds message, saves username of the certified user
:param passed: boolean whether the user passed enough assignments
:param certificate_type: type of the certificate - completion or excellence
:param first_name: first name entered by user
:param last_name: last name entered by user
"""
if passed and certificate_type == "excellence":
# Generate a certificate of
certificate, pdf = generate_certificate("./certificate-excellence.png", first_name, last_name)
# Add this user to our database
add_certified_user(hf_username, first_name, last_name, certificate_type)
# Add a message
message = """
Congratulations, you successfully completed the Hugging Face Audio Course πŸŽ‰! \n
Since you pass 100% of the hands-on you get a Certificate of Excellence πŸŽ“. \n
You can download your certificate below ⬇️ \n
Don't hesitate to share your certificate image below on Twitter and Linkedin (you can tag me @mariakhalusova and @huggingface) πŸ€—
"""
elif passed and certificate_type == "completion":
# Generate a certificate of completion
certificate, pdf = generate_certificate("./certificate-completion.png", first_name, last_name)
# Add this user to our database
add_certified_user(hf_username, first_name, last_name, certificate_type)
# Add a message
message = """
Congratulations, you successfully completed the Hugging Face Deep Reinforcement Learning Course πŸŽ‰! \n
Since you pass 80% of the hands-on you get a Certificate of Completion πŸŽ“. \n
You can download your certificate below ⬇️ \n
Don't hesitate to share your certificate image below on Twitter and Linkedin (you can tag me @mariakhalusova and @huggingface) πŸ€— \n
You can try to get a Certificate of Excellence if you pass 100% of the hands-on, don't hesitate to check which unit you didn't pass and update these models.
"""
else:
# Not passed yet
certificate = Image.new("RGB", (100, 100), (255, 255, 255))
pdf = "./fail.pdf"
# Add a message
message = """
You didn't pass the minimum of 3 out of 4 of the hands-on to get a certificate of completion.
For more information about the certification process [check the course page on certification](https://huggingface.co/learn/audio-course/chapter8/certification).
Use the [self-evaluation space](https://huggingface.co/spaces/MariaK/Check-my-progress-Audio-Course) to see which assignments have not been completed.
"""
return certificate, message, pdf
def certification(hf_username, first_name, last_name):
passed, certificate_type = check_if_passed(hf_username)
certificate, message, pdf = create_certificate(passed, certificate_type, hf_username, first_name, last_name)
print("MESSAGE", message)
if passed:
visible = True
else:
visible = False
return message, pdf, certificate, output_row.update(visible=visible)
with gr.Blocks() as demo:
gr.Markdown(f"""
# Get your Hugging Face Audio Course Certificate πŸŽ“
The certification process is completely free:
- To get a *certificate of completion*: you need to **pass 3 out of 4 hands-on assignments**.
- To get a *certificate of excellence*: you need to **pass 4 out of 4 hands-on assignments**.
For more information about the certification process [check the course page on certification](https://huggingface.co/learn/audio-course/chapter8/certification).
Don't hesitate to share your certificate on Twitter (tag me @mariakhalusova and @huggingface) and on LinkedIn.
""")
hf_username = gr.Textbox(placeholder="MariaK", label="Your Hugging Face Username (case sensitive)")
first_name = gr.Textbox(placeholder="Jane", label="Your First Name")
last_name = gr.Textbox(placeholder="Doe", label="Your Last Name")
check_progress_button = gr.Button(value="Check if I pass and get the certificate")
output_text = gr.components.Textbox()
with gr.Row(visible=True) as output_row:
output_pdf = gr.File()
output_img = gr.components.Image(type="pil")
check_progress_button.click(fn=certification, inputs=[hf_username, first_name, last_name], outputs=[output_text, output_pdf, output_img, output_row])
demo.launch(debug=True)