Completing submission rules

src/submission/submit.py

@@ -1,6 +1,6 @@
 import json
 import os
-from datetime import datetime, timezone
+from datetime import datetime, timedelta, timezone
 import gradio as gr
 from src.display.formatting import styled_error, styled_message, styled_warning
 from src.envs import API, EVAL_REQUESTS_PATH, TOKEN, QUEUE_REPO
@@ -44,6 +44,19 @@ def add_new_eval(
     if not is_model_on_hub(model_name=model, token=TOKEN, test_tokenizer=True): #revision=revision
         return styled_error("Model does not exist on HF Hub. Please select a valid model name.")
     
+
+    ##check for org banning
+    progress(0.2, desc=f"Checking for banned orgs")
+    banned_orgs = [{
+        'org_name':'TEMPLATE',
+        'banning_reason':'Submitting contaminated models'
+    }]
+
+    if user_name in [banned_org['org_name'] for banned_org in banned_orgs]:
+        return styled_error(
+            f"Your org  \"{user_name}\" is banned from submitting models on ABL. If you think this is a mistake then please contact [email protected]"
+        )
+
     """
     if model_type is None or model_type == "":
         return styled_error("Please select a model type.")
@@ -69,10 +82,11 @@ def add_new_eval(
     except Exception:
         return styled_error("Could not get your model information. Please fill it up properly.")
 
+    progress(0.3, desc=f"Checking model size")
     model_size = get_model_size(model_info=model_info)#, precision=precision
 
     if model_size>15:
-        return styled_error("Unfortunately we do not accept models above 15B parameters from the community due to limited GPU availability.")
+        return styled_error("Due to limited GPU resources, we are currently unable to accept community-submitted models exceeding 15 billion parameters")
 
     # Were the model card and license filled?
     try:
@@ -83,6 +97,32 @@ def add_new_eval(
     modelcard_OK, error_msg = check_model_card(model)
     if not modelcard_OK:
         return styled_error(error_msg)
+    
+
+
+    ##check if org have submitted in the last 30 days
+    progress(0.6, desc=f"Checking last submission date")
+    previous_user_submissions = USERS_TO_SUBMISSION_DATES.get(user_name)
+
+    if previous_user_submissions:
+
+        previous_user_submission_dates = [datetime.strptime(date.replace("T"," ").split(" ")[0], "%Y-%m-%d") for date in previous_user_submissions]
+        previous_user_submission_dates.sort(reverse=True)
+        most_recent_submission = previous_user_submission_dates[0]
+
+        time_since_last_submission = datetime.now() - most_recent_submission
+        if time_since_last_submission < timedelta(days=30):
+            return styled_warning(
+                f"Your org \"{user_name}\" have already submitted a model in the last 30 days. Please wait before submitting another model. For exceptions please contact [email protected]"
+            )
+            
+
+
+    progress(0.8, desc=f"Checking same model submissions")
+
+    # Check for duplicate submission
+    if f"{model}" in REQUESTED_MODELS: #_{revision}_{precision}
+        return styled_warning("This model has been already submitted.")
 
     # Seems good, creating the eval
     print("Preparing a new eval")
@@ -103,10 +143,7 @@ def add_new_eval(
         #"private": False,
     }
 
-    progress(0.5, desc=f"Checking previous submissions")
-    # Check for duplicate submission
-    if f"{model}" in REQUESTED_MODELS: #_{revision}_{precision}
-        return styled_warning("This model has been already submitted.")
+   
 
     print("Creating eval file")
     OUT_DIR = f"{EVAL_REQUESTS_PATH}/{user_name}"