Create app.py

app.py

@@ -0,0 +1,195 @@
+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 excellence
+    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)