Add logic to upvote a user model

src/display/utils.py

@@ -1,14 +1,25 @@
 from dataclasses import dataclass, make_dataclass
 from enum import Enum
+import requests 
 import json
 import logging
 from datetime import datetime
 import pandas as pd
 
-
 # Configure logging
 logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 
+
+def is_user_in_org(username: str, org: str) -> bool:
+    url = f"https://huggingface.co/api/users/{username}/overview"
+    response = requests.get(url)
+    if response.status_code == 200:
+        user_info = response.json()
+        return any(org_info['name'] == org for org_info in user_info.get('orgs', []))
+    else:
+        print(f"Failed to fetch user info for {username}. Status code: {response.status_code}")
+        return False
+        
 # Convert ISO 8601 dates to datetime objects for comparison
 def parse_iso8601_datetime(date_str):
     if date_str.endswith('Z'):

src/submission/submit.py

@@ -14,6 +14,8 @@ from src.envs import (
     QUEUE_REPO,
     RATE_LIMIT_PERIOD,
     RATE_LIMIT_QUOTA,
+    VOTES_REPO,
+    VOTES_PATH,
 )
 from src.leaderboard.filter_models import DO_NOT_SUBMIT_MODELS
 from src.submission.check_validity import (
@@ -24,9 +26,14 @@ from src.submission.check_validity import (
     user_submission_permission,
 )
 
+from src.voting.vote_system import VoteManager
+from src.display.utils import is_user_in_org
+
 REQUESTED_MODELS = None
 USERS_TO_SUBMISSION_DATES = None
 
+vote_manager = VoteManager(VOTES_PATH, EVAL_REQUESTS_PATH, VOTES_REPO)
+
 @dataclass
 class ModelSizeChecker:
     model: str
@@ -62,7 +69,7 @@ def add_new_eval(
     use_chat_template: bool,
     profile: gr.OAuthProfile | None
 ):  
-    # Login require
+    # Login required
     if profile is None:
         return styled_error("Hub Login Required")
 
@@ -87,10 +94,6 @@ def add_new_eval(
     if model_type is None or model_type == "":
         return styled_error("Please select a model type.")
 
-    # Is user submitting own model?
-    # Check that username in the org. 
-    # if org_or_user != profile.username:
-
     # Is the user rate limited?
     if org_or_user != "":
         user_can_submit, error_msg = user_submission_permission(
@@ -196,6 +199,20 @@ def add_new_eval(
     # Remove the local file
     os.remove(out_path)
 
+
+    # Check if the user is submitting their own model or a model from their organization
+    if org_or_user == username or is_user_in_org(username, org_or_user):
+        # Automatically add a vote for the user's own model or their organization's model
+        vote_manager.add_vote(
+            selected_model=model,
+            pending_models_df=None,
+            profile=profile
+        )
+        print(f"Automatically added a vote for {model} submitted by {username}")
+
+    # Upload votes to the repository
+    vote_manager.upload_votes()
+
     return styled_message(
         "Your request has been submitted to the evaluation queue!\nPlease wait for up to an hour for the model to show in the PENDING list."
-    )
+    )