add rate limit

src/submission/submit_csv.py

@@ -1,14 +1,14 @@
 import json
 import os
 import uuid
-from datetime import datetime, timezone
+from datetime import datetime, timedelta, timezone
 
 from src.envs import API, SUBMISSIONS_PATH, SUBMISSIONS_REPO
 from src.evaluation.compute_metrics import compute_metrics
 from src.evaluation.load_labels import load_true_labels
 from src.submission.validate_csv import validate_csv
 from src.teams.auth import validate_token
-from src.teams.storage import hash_token
+from src.teams.storage import get_team_by_name, hash_token, update_last_valid_submission
 
 
 def get_team_best_scores(team_name: str) -> dict | None:
@@ -105,6 +105,29 @@ def should_update_scores(new_scores: dict, best_scores: dict | None) -> bool:
     return False
 
 
+def check_rate_limit(team_name: str) -> tuple[bool, str]:
+    team_data = get_team_by_name(team_name)
+    if not team_data:
+        return True, ""
+
+    last_submission = team_data.get("last_valid_submission")
+    if not last_submission:
+        return True, ""
+
+    try:
+        last_time = datetime.strptime(last_submission, "%Y-%m-%dT%H:%M:%SZ").replace(tzinfo=timezone.utc)
+        now = datetime.now(timezone.utc)
+        time_diff = now - last_time
+
+        if time_diff < timedelta(minutes=1):
+            remaining = 60 - time_diff.total_seconds()
+            return False, f"Rate limit exceeded. Please wait {int(remaining)} seconds before submitting again."
+
+        return True, ""
+    except Exception:
+        return True, ""
+
+
 def submit_csv(token: str, csv_content: str) -> tuple[bool, str]:
     team = validate_token(token)
     if not team:
@@ -121,8 +144,14 @@ def submit_csv(token: str, csv_content: str) -> tuple[bool, str]:
     if not is_valid:
         return False, f"CSV validation failed: {error_msg}"
 
+    can_submit, rate_limit_msg = check_rate_limit(team_name)
+    if not can_submit:
+        return False, rate_limit_msg
+
     scores = compute_metrics(predictions_df, true_labels)
 
+    update_last_valid_submission(team_name)
+
     best_scores = get_team_best_scores(team_name)
 
     if should_update_scores(scores, best_scores):

src/teams/storage.py

@@ -84,3 +84,44 @@ def get_all_teams() -> list[dict]:
             continue
 
     return teams
+
+
+def update_last_valid_submission(team_name: str) -> None:
+    ensure_teams_dir()
+    team_file = os.path.join(TEAMS_PATH, f"{team_name}.json")
+
+    if not os.path.exists(team_file):
+        return
+
+    with open(team_file, "r") as f:
+        team_data = json.load(f)
+
+    team_data["last_valid_submission"] = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
+
+    with open(team_file, "w") as f:
+        json.dump(team_data, f)
+
+    try:
+        API.upload_file(
+            path_or_fileobj=team_file,
+            path_in_repo=f"{team_name}.json",
+            repo_id=TEAMS_REPO,
+            repo_type="dataset",
+            commit_message=f"Update last submission for team: {team_name}",
+        )
+    except Exception as e:
+        print(f"Warning: Could not upload team update to hub: {e}")
+
+
+def get_team_by_name(team_name: str) -> dict | None:
+    ensure_teams_dir()
+    team_file = os.path.join(TEAMS_PATH, f"{team_name}.json")
+
+    if not os.path.exists(team_file):
+        return None
+
+    try:
+        with open(team_file, "r") as f:
+            return json.load(f)
+    except Exception:
+        return None