feat: check if user passed hackathon (#3)

- feat: check if user passed hackathon (06735809450dbaec195f09bab2aaaed56e34c24a)

app.py

@@ -19,23 +19,56 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 DATASET_REPO_URL = "https://huggingface.co/datasets/wseo/huggingface-krew-hackathon23"
 CERTIFIED_USERS_FILENAME = "usernames.csv"
 
+ORGANIZATION = "pseudolab"
+
+
+def has_contributions(repo_type, hf_username, organization):
+    """
+    Check if a user has contributions in the specified repository type.
+    :param repo_type: A repo type supported by the Hub
+    :param hf_username: HF Hub username
+    :param organization: HF Hub organization
+    """
+    repo_list = {
+        'model': api.list_models,
+        'dataset': api.list_datasets,
+        'space': api.list_spaces
+    }
+
+    for repo in repo_list[repo_type](author=organization):
+        commits = api.list_repo_commits(repo.id, repo_type=repo_type)
+        if any(hf_username in commit.authors for commit in commits):
+            return True
+    return False
+
+
+def get_hub_footprint(hf_username, organization):
+    """
+    Check the types of contributions a user has made.
+    :param hf_username: HF Hub username
+    :param organization: HF Hub organization
+    """
+    has_models = has_contributions('model', hf_username, organization)
+    has_datasets = has_contributions('dataset', hf_username, organization)
+    has_spaces = has_contributions('space', hf_username, organization)
+
+    return (has_models, has_datasets, has_spaces)
+
 
 def check_if_passed(hf_username):
     """
-    Check if given user passed enough assignments
-    :param hf_username: Hugging Face Hub username
+    Check if given user contributed to hackathon
+    :param hf_username: HF Hub username
     """
     
     passed = False  
     certificate_type = ""
 
-    # TODO: Get all the repos that the user contributed to in the organization using Git refs
-
-    # TODO: If the user contributed to all repo types (model, dataset, space) then assign excellence
-    if True:
+    # If the user contributed to models, datasets and spaces then assign excellence
+    if all(get_hub_footprint(hf_username, ORGANIZATION)):
       passed = True
       certificate_type = "excellence"
-    elif False:
+    elif any(get_hub_footprint(hf_username, ORGANIZATION)):
       passed = True
       certificate_type = "completion"
 
@@ -71,7 +104,7 @@ def generate_certificate(certificate_template, first_name, last_name, hf_usernam
     #d.line(((863, 0), (863, 1400)), "gray")
 
     # Date of certification
-    d.text((863, 336), f"psh23-{username}", fill=(117,117,117), anchor="mm", font=username_font)
+    d.text((863, 336), f"HKH23-{username}", fill=(117,117,117), anchor="mm", font=username_font)
 
     pdf = im.convert('RGB')
     pdf.save('certificate.pdf')