Update main.py

main.py

@@ -49,7 +49,8 @@ def api_client() -> HfApi:
 def _today_utc_str() -> str:
     return datetime.now(timezone.utc).strftime("%Y-%m-%d")
 
-def _count_submissions_today(username: str) -> int:
+def _count_submissions_today(username: str, phase_codename: str | None = None) -> int:
+    """Count today's submissions for a user, optionally filtered by phase."""
     try:
         files = api_client().list_repo_files(
             repo_id=DB_REPO_ID, repo_type=DB_REPO_TYPE, token=SUBMISSIONS_TOKEN
@@ -65,17 +66,24 @@ def _count_submissions_today(username: str) -> int:
                     filename=f, token=SUBMISSIONS_TOKEN
                 )
                 meta = json.load(open(p))
-                if meta.get("username", "").lower() == username.lower():
-                    ts = meta.get("timestamp", 0)
-                    sub_date = datetime.fromtimestamp(ts, tz=timezone.utc).strftime("%Y-%m-%d")
-                    if sub_date == today:
-                        count += 1
+                if meta.get("username", "").lower() != username.lower():
+                    continue
+                if phase_codename and meta.get("phase_codename") != phase_codename:
+                    continue
+                ts = meta.get("timestamp", 0)
+                sub_date = datetime.fromtimestamp(ts, tz=timezone.utc).strftime("%Y-%m-%d")
+                if sub_date == today:
+                    count += 1
             except Exception:
                 continue
         return count
     except Exception:
         return 0
 
+def _get_cap_for_phase(phase_codename: str) -> int:
+    """Return the daily submission cap for a given phase."""
+    return 1 if phase_codename == "test-challenge2024" else DAILY_SUBMISSION_CAP
+
 def _validate_submission_json(obj: Any) -> Tuple[bool, str]:
     if not isinstance(obj, list):
         return False, "Submission must be a JSON list of annotations."
@@ -241,7 +249,7 @@ def load_leaderboard():
         return pd.DataFrame(), f"❌ Could not load leaderboard: {e}"
 
     if not df.empty and "phase_codename" in df.columns:
-        df = df[df["phase_codename"] == "test-standard2024"]
+        df = df[df["phase_codename"] == "test-challenge2024"]
 
     if df.empty:
         return pd.DataFrame(), "ℹ️ No scored Standard phase submissions yet. Be the first!"
@@ -279,10 +287,15 @@ def handle_submit(file, team, model_name, phase_label, challenge_type, profile:
     if not model_name.strip():
         return "❌ Please enter a Model Name.", ""
 
-    # Daily cap check
-    subs_today = _count_submissions_today(username)
-    if subs_today >= DAILY_SUBMISSION_CAP:
-        return f"⛔ You've reached your daily limit of {DAILY_SUBMISSION_CAP} submissions. Come back tomorrow!", ""
+    # Resolve phase codename first (needed for cap check)
+    phase_codename = next((p["codename"] for p in PHASES if p["label"] == phase_label), phase_label)
+    cap = _get_cap_for_phase(phase_codename)
+
+    # Daily cap check (phase-specific)
+    subs_today = _count_submissions_today(username, phase_codename)
+    if subs_today >= cap:
+        phase_label_str = "challenge" if phase_codename == "test-challenge2024" else "this"
+        return f"⛔ You've reached your daily limit of {cap} submission(s) for the {phase_label_str} phase. Come back tomorrow!", ""
 
     # Parse JSON
     try:
@@ -296,8 +309,6 @@ def handle_submit(file, team, model_name, phase_label, challenge_type, profile:
     if not ok:
         return f"❌ Invalid submission format: {msg}", ""
 
-    # Resolve phase codename
-    phase_codename = next((p["codename"] for p in PHASES if p["label"] == phase_label), phase_label)
     original_filename = os.path.basename(file)
 
     # Upload
@@ -315,9 +326,9 @@ def handle_submit(file, team, model_name, phase_label, challenge_type, profile:
     except Exception as e:
         return f"❌ Upload failed: {e}", ""
 
-    remaining = DAILY_SUBMISSION_CAP - subs_today - 1
+    remaining = cap - subs_today - 1
     return (
-        f"✅ Submission queued successfully! Visit **My Submissions** to see the results. You have {remaining}/{DAILY_SUBMISSION_CAP} submissions remaining today.",
+        f"✅ Submission queued successfully! Visit **My Submissions** to see the results. You have {remaining}/{cap} submissions remaining today for this phase.",
         submission_id,
     )
 
@@ -381,9 +392,13 @@ def load_my_submissions(phase_filter: str, profile: gr.OAuthProfile | None):
 def get_daily_cap_info(profile: gr.OAuthProfile | None):
     if profile is None:
         return ""
-    subs_today = _count_submissions_today(profile.username)
-    remaining = DAILY_SUBMISSION_CAP - subs_today
-    return f"**{remaining}/{DAILY_SUBMISSION_CAP}** submissions remaining today"
+    lines = []
+    for p in PHASES:
+        cap = _get_cap_for_phase(p["codename"])
+        used = _count_submissions_today(profile.username, p["codename"])
+        remaining = cap - used
+        lines.append(f"**{p['label'].split('(')[0].strip()}:** {remaining}/{cap} remaining")
+    return "  \n".join(lines)
 
 
 def get_user_greeting(profile: gr.OAuthProfile | None):
@@ -553,8 +568,8 @@ with demo.route("Object Localization", "/object-localization") as obj_loc:
 
         # Leaderboard
         with gr.TabItem("🏆 Leaderboard"):
-            gr.Markdown("### Standard Phase Rankings")
-            gr.Markdown(f"Ranked by **{DEFAULT_SORT_METRIC}** (descending). Standard phase only.")
+            gr.Markdown("### Challenge Phase Rankings")
+            gr.Markdown(f"Ranked by **{DEFAULT_SORT_METRIC}** (descending). Challenge phase only.")
             with gr.Accordion("📐 How is the Score Calculated?", open=False):
                 gr.Markdown(EVAL_DETAILS_MD)
             lb_msg = gr.Markdown("")
@@ -654,7 +669,6 @@ with demo.route("Visual Question Answering", "/vqa"):
     gr.Markdown("### 🔒 Coming Soon")
     gr.Markdown(
         "This challenge is currently under development. "
-        "Check back soon or follow [@VizWiz](https://vizwiz.org) for updates."
     )
 
 # ── ANSWER GROUNDING PAGE ─────────────────────────────────────────────
@@ -663,7 +677,6 @@ with demo.route("Answer Grounding", "/answer-grounding"):
     gr.Markdown("### 🔒 Coming Soon")
     gr.Markdown(
         "This challenge is currently under development. "
-        "Check back soon or follow [@VizWiz](https://vizwiz.org) for updates."
     )