app.py

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, hf_username)
        # 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 2023 Hackathon 🎉! \n 
        Since you contributed to models, datasets, and spaces- 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 @wonhseo, @pseudolab and @huggingface) 🤗
        """
    elif passed and certificate_type == "completion":    
        # Generate a certificate of completion
        certificate, pdf = generate_certificate("./certificate-completion.png", first_name, last_name, hf_username)
        # 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 2023 Hackathon 🎉! \n 
        Since you contributed to at least one model, dataset, or space- 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 @wonhseo, @pseudolab and @huggingface) 🤗 \n
        You can try to get a Certificate of Excellence if you contribute to all types of repos, please don't hesitate to do so.
        """
    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 one contribution to get a certificate of completion. 
          For more information about the certification process, refer to the hackathon page.
          If the results here differ from your contributions, make sure you moved your space to the pseudolab organization.
          """
    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 2023 Hackathon Certificate 🎓
    The certification process is completely free:
    - To get a *certificate of completion*: you need to **contribute to at least one model, dataset, or space**.
    - To get a *certificate of excellence*: you need to **contribute to models, datasets, and spaces**. *(Yes, all three!)*
    
    For more information about the certification process [check the hackathon page on certification](https://pseudo-lab.github.io/huggingface-hackathon23/submit.html#certification).
    
    Don't hesitate to share your certificate on Twitter (tag me [@wonhseo](https://twitter.com/wonhseo), [@pseudolab](https://twitter.com/pseudolab), and [@huggingface](https://twitter.com/huggingface)) and on LinkedIn.
    """)
    
    hf_username = gr.Textbox(placeholder="wseo", label="Your Hugging Face Username (case sensitive)")
    first_name = gr.Textbox(placeholder="Wonhyeong", label="Your First Name")
    last_name = gr.Textbox(placeholder="Seo", 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)