:lipstick: Improve refresh button and control layout

backend/api.py

@@ -1,6 +1,7 @@
 import asyncio
 from typing import Dict, Any
 from .submission import process_submission
+from .rate_limiter import check_submission_limit, record_submission, get_submission_status
 
 
 # Sample data - replace with actual data loading
@@ -23,8 +24,14 @@ async def submit_model(
     pretraining: str = "",
     organization: str = ""
 ) -> Dict[str, Any]:
-    """API endpoint for model submission."""
+    """API endpoint for model submission with rate limiting."""
     
+    # Check rate limit
+    if not check_submission_limit(hf_space_tag):
+        status = get_submission_status(hf_space_tag)
+        raise Exception(f"Daily submission limit exceeded. Used {status['submissions_today']}/{status['daily_limit']} submissions today. Try again tomorrow.")
+    
+    # Process submission
     record = await process_submission(
         model_name=model_name,
         hf_space_tag=hf_space_tag,
@@ -38,6 +45,9 @@ async def submit_model(
         true_labels=SAMPLE_LABELS
     )
     
+    # Record successful submission for rate limiting
+    record_submission(hf_space_tag)
+    
     return record
 
 

backend/rate_limiter.py

@@ -0,0 +1,99 @@
+import json
+import os
+from datetime import datetime, date
+from typing import Dict, Any
+from config.settings import DAILY_SUBMISSION_LIMIT, DEBUG_MODE
+
+
+RATE_LIMIT_FILE = "submission_tracking.json"
+
+
+def load_submission_tracking() -> Dict[str, Any]:
+    """Load submission tracking data from file."""
+    if not os.path.exists(RATE_LIMIT_FILE):
+        return {}
+    
+    with open(RATE_LIMIT_FILE, 'r') as f:
+        return json.load(f)
+
+
+def save_submission_tracking(data: Dict[str, Any]) -> None:
+    """Save submission tracking data to file."""
+    with open(RATE_LIMIT_FILE, 'w') as f:
+        json.dump(data, f, indent=2)
+
+
+def check_submission_limit(hf_space_tag: str) -> bool:
+    """
+    Check if the HF space can submit (hasn't exceeded daily limit).
+    
+    Returns:
+        True if submission is allowed, False if rate limited
+    """
+    
+    # Debug mode = unlimited submissions
+    if DEBUG_MODE:
+        return True
+    
+    today = str(date.today())
+    data = load_submission_tracking()
+    
+    # Initialize space tracking if not exists
+    if hf_space_tag not in data:
+        data[hf_space_tag] = {}
+    
+    # Get today's submission count for this space
+    today_count = data[hf_space_tag].get(today, 0)
+    
+    # Check if limit exceeded
+    return today_count < DAILY_SUBMISSION_LIMIT
+
+
+def record_submission(hf_space_tag: str) -> None:
+    """Record a successful submission for rate limiting."""
+    
+    today = str(date.today())
+    data = load_submission_tracking()
+    
+    # Initialize space tracking if not exists
+    if hf_space_tag not in data:
+        data[hf_space_tag] = {}
+    
+    # Increment today's count
+    data[hf_space_tag][today] = data[hf_space_tag].get(today, 0) + 1
+    
+    # Clean up old dates (keep last 7 days for reference)
+    cleanup_old_entries(data)
+    
+    save_submission_tracking(data)
+
+
+def cleanup_old_entries(data: Dict[str, Any]) -> None:
+    """Remove entries older than 7 days to keep file small."""
+    
+    from datetime import timedelta
+    cutoff_date = str(date.today() - timedelta(days=7))
+    
+    for space_tag in data:
+        old_dates = [d for d in data[space_tag] if d < cutoff_date]
+        for old_date in old_dates:
+            del data[space_tag][old_date]
+
+
+def get_submission_status(hf_space_tag: str) -> Dict[str, Any]:
+    """Get current submission status for a space."""
+    
+    today = str(date.today())
+    data = load_submission_tracking()
+    
+    today_count = data.get(hf_space_tag, {}).get(today, 0)
+    remaining = max(0, DAILY_SUBMISSION_LIMIT - today_count)
+    
+    return {
+        "space": hf_space_tag,
+        "submissions_today": today_count,
+        "daily_limit": DAILY_SUBMISSION_LIMIT,
+        "remaining": remaining,
+        "debug_mode": DEBUG_MODE,
+        "can_submit": check_submission_limit(hf_space_tag)
+    }

config/settings.py

@@ -33,9 +33,12 @@ APP_DESCRIPTION = "Leaderboard for molecular toxicity prediction on the Tox21 da
 
 # Evaluation Settings
 DEFAULT_METRIC = "roc_auc"
-SUBMISSION_RATE_LIMIT = 5  # Max submissions per day per user
 EVALUATION_TIMEOUT = 3600  # 1 hour timeout for evaluations
 
+# Rate Limiting
+DEBUG_MODE = os.environ.get("DEBUG_SUBMISSIONS", "false").lower() == "true"
+DAILY_SUBMISSION_LIMIT = 999 if DEBUG_MODE else 5  # Per HF space per day
+
 # Display Settings
 MAX_MODELS_DISPLAYED = 100
 DEFAULT_SORT_BY = "average_score"

frontend/layout.py

@@ -16,28 +16,31 @@ def create_leaderboard_tab(refresh_callback: Callable = None) -> gr.TabItem:
     
     with gr.TabItem("🏅 Leaderboard", elem_id="leaderboard-tab", id=0) as tab:
         
-        # Header section with refresh button
-        with gr.Row():
-            gr.HTML(LeaderboardContent.get_header_html())
-            refresh_btn = gr.Button("🔄 Refresh", variant="secondary", size="sm")
+        # Header section
+        header_html = gr.HTML(LeaderboardContent.get_header_html())
         
-        # Load initial data
-        result_data = refresh_leaderboard().reset_index(drop=True)
-        result_data.columns = result_data.columns.map(str)
-
-        leaderboard_table = Leaderboard(
-            value=result_data,
-            search_columns=["Model", "Model Description", "Publication"],        # or e.g. ["Model", "Model Description", "Publication"]
-            select_columns=[],        # or e.g. ["Publication"]
-            filter_columns=[],        # or a list of valid column names/filters
-            hide_columns=[],          # keep explicit to be safe
-            elem_id="leaderboard-table",
-            height=480,       # only controls vertical
-            min_width=160,    # doesn’t prevent horizontal scroll
-            wrap=True,
-            column_widths=[200, 300, 150, 100, 120, 100] + [80] * 12,
-        )
+        # Leaderboard with integrated refresh button
+        with gr.Row():
+            with gr.Column(scale=10):
+                # Load initial data
+                result_data = refresh_leaderboard().reset_index(drop=True)
+                result_data.columns = result_data.columns.map(str)
 
+                leaderboard_table = Leaderboard(
+                    value=result_data,
+                    search_columns=["Model", "Model Description", "Publication"],
+                    select_columns=[],
+                    filter_columns=[],
+                    hide_columns=[],
+                    elem_id="leaderboard-table",
+                    height=480,
+                    min_width=160,
+                    wrap=True,
+                    column_widths=[200, 300, 150, 100, 120, 100] + [80] * 12,
+                )
+            
+            with gr.Column(scale=1, min_width=100):
+                refresh_btn = gr.Button("🔄 Refresh", variant="secondary", size="sm")
         
         # Connect refresh button
         def refresh_data():

submission_tracking.json

@@ -0,0 +1,5 @@
+{
+  "test/space": {
+    "2025-08-30": 5
+  }
+}