Update app.py

app.py

@@ -2,7 +2,9 @@ import gradio as gr
 import pandas as pd
 from apscheduler.schedulers.background import BackgroundScheduler
 # Removed Hugging Face Hub imports as they are not needed for the simplified leaderboard
-# Assuming these are correctly defined in your src.about module
+
+# --- Make sure these imports work relative to your file structure ---
+# Option 1: If src is a directory in the same folder as your script:
 from src.about import (
     CITATION_BUTTON_LABEL,
     CITATION_BUTTON_TEXT,
@@ -11,18 +13,28 @@ from src.about import (
     LLM_BENCHMARKS_TEXT,
     TITLE,
 )
-# Assuming this is correctly defined in your src.display.css_html_js module
 from src.display.css_html_js import custom_css
-# Assuming this is correctly defined in your src.envs module
 from src.envs import REPO_ID # Keep if needed for restart_space or other functions
-# Assuming this is correctly defined in your src.submission.submit module
 from src.submission.submit import add_new_eval # Keep if using the submit tab
 
+# Option 2: If you don't have these files, define placeholders (REMOVE THIS if using Option 1)
+# print("Warning: Using placeholder values for src module imports.")
+# CITATION_BUTTON_LABEL="Citation"
+# CITATION_BUTTON_TEXT="Please cite us if you use this benchmark..."
+# EVALUATION_QUEUE_TEXT="Current evaluation queue:"
+# INTRODUCTION_TEXT="Welcome to the MLE-Dojo Benchmark Leaderboard."
+# LLM_BENCHMARKS_TEXT="Information about the benchmarks..."
+# TITLE="<h1>🏆 MLE-Dojo Benchmark Leaderboard</h1>"
+# custom_css=""
+# REPO_ID="your/space-id" # Replace with actual ID if needed
+# def add_new_eval(*args): return "Submission placeholder."
+# --- End Placeholder Definitions ---
+
 
 # --- Elo Leaderboard Configuration ---
 # Enhanced data with Rank (placeholder), Organizer, License, and URL
-# NOTE: URLs are placeholders, replace them with actual model/project pages.
-# License information is based on common knowledge - verify for accuracy.
+# !!! IMPORTANT: Replace placeholder URLs with actual model/project pages. !!!
+# Verify organizer and license information for accuracy.
 data = [
     {'model_name': 'gpt-4o-mini', 'url': 'https://openai.com/index/hello-gpt-4o/', 'organizer': 'OpenAI', 'license': 'Proprietary', 'MLE-Lite_Elo': 753, 'Tabular_Elo': 839, 'NLP_Elo': 758, 'CV_Elo': 754, 'Overall': 778},
     {'model_name': 'gpt-4o', 'url': 'https://openai.com/index/hello-gpt-4o/', 'organizer': 'OpenAI', 'license': 'Proprietary', 'MLE-Lite_Elo': 830, 'Tabular_Elo': 861, 'NLP_Elo': 903, 'CV_Elo': 761, 'Overall': 841},
@@ -63,7 +75,7 @@ def update_leaderboard(category):
         score_column = category_to_column[DEFAULT_CATEGORY]
         if score_column not in master_df.columns: # Check fallback column too
              # Return empty df with correct columns if still invalid
-             # *** FIX APPLIED HERE: Use lowercase keys consistent with DataFrame ***
+             # Use lowercase keys here consistent with master_df for the empty case
             return pd.DataFrame({
                 "Rank": [],
                 "Model": [],
@@ -94,12 +106,10 @@ def update_leaderboard(category):
     # Rename the score column to 'Elo Score' for consistent display
     df.rename(columns={score_column: 'Elo Score'}, inplace=True)
 
-    # --- FIX APPLIED HERE ---
     # Select and reorder columns for final display using the ACTUAL column names in df
     # Use lowercase 'organizer' and 'license' here because they haven't been renamed.
     final_columns = ["Rank", "Model", "organizer", "license", "Elo Score"]
     df = df[final_columns]
-    # -----------------------
 
     # Note: The DataFrame returned now has columns:
     # 'Rank', 'Model', 'organizer', 'license', 'Elo Score'
@@ -118,27 +128,28 @@ EVAL_TYPES = ["str", "str", "str", "str"] # Define for the dataframe types
 
 # --- Keep restart function if relevant ---
 def restart_space():
+    # Make sure REPO_ID is correctly defined/imported if this function is used
     print(f"Attempting to restart space: {REPO_ID}")
-    # Replace with your actual space restart mechanism if needed
+    # Replace with your actual space restart mechanism if needed (e.g., HfApi().restart_space(REPO_ID))
 
 # --- Gradio App Definition ---
-# Add some potential CSS for better table appearance (can be added to custom_css file or here)
-# Example CSS (add to your custom_css or inline):
+# Add custom CSS rules here or ensure custom_css is imported correctly
+# Example CSS rules you might want in your custom_css:
 # table { width: 100%; border-collapse: collapse; }
-# th, td { padding: 8px 12px; border: 1px solid #ddd; text-align: left; }
-# th { background-color: #f2f2f2; font-weight: bold; } /* Added bold */
+# th, td { padding: 8px 12px; border: 1px solid #ddd; text-align: left; white-space: normal; } /* Allow wrapping */
+# th { background-color: #f2f2f2; font-weight: bold; }
 # tr:nth-child(even) { background-color: #f9f9f9; }
 # tr:hover { background-color: #e9e9e9; }
-# td a { color: #007bff; text-decoration: none; } /* Style links in table */
+# td a { color: #007bff; text-decoration: none; }
 # td a:hover { text-decoration: underline; }
 
 # Use a theme for better default styling
 demo = gr.Blocks(css=custom_css, theme=gr.themes.Soft())
 
 with demo:
-    # Use the TITLE variable imported from src.about
+    # Use the TITLE variable imported or defined above
     gr.HTML(TITLE)
-    # Use the INTRODUCTION_TEXT variable imported from src.about
+    # Use the INTRODUCTION_TEXT variable imported or defined above
     gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
@@ -157,14 +168,14 @@ with demo:
                     # Headers for DISPLAY remain capitalized
                     headers=["Rank", "Model", "Organizer", "License", "Elo Score"],
                     # Datatype maps to the final df columns: Rank, Model, organizer, license, Elo Score
-                    # 'html' is used for the 'Model' column containing the <a> tag
                     datatype=["number", "html", "str", "str", "number"],
                     interactive=False,
-                    height=400 + (len(master_df) * 25), # Adjusted height slightly
+                    # --- FIX APPLIED: Removed unsupported 'height' argument ---
+                    # row_count determines the number of rows to display
                     row_count=(len(master_df), "fixed"),
                     col_count=(5, "fixed"),
                     wrap=True, # Allow text wrapping
-                    elem_id="leaderboard-table" # CSS hook
+                    elem_id="leaderboard-table" # CSS hook for custom styling
                 )
                 # Link the radio button change to the update function
                 category_selector.change(
@@ -174,15 +185,15 @@ with demo:
                 )
 
         with gr.TabItem("📝 About", elem_id="llm-benchmark-tab-about", id=1):
-             # Use the LLM_BENCHMARKS_TEXT variable imported from src.about
+             # Use the LLM_BENCHMARKS_TEXT variable imported or defined above
             gr.Markdown(LLM_BENCHMARKS_TEXT, elem_classes="markdown-text")
 
         # --- Submit Tab (Commented out as in original request) ---
-        # Make sure EVALUATION_QUEUE_TEXT and add_new_eval are imported if uncommented
+        # Make sure EVALUATION_QUEUE_TEXT and add_new_eval are imported/defined if uncommented
         # with gr.TabItem("🚀 Submit here! ", elem_id="llm-benchmark-tab-submit", id=2):
         #     with gr.Column():
         #          with gr.Row():
-        #              gr.Markdown(EVALUATION_QUEUE_TEXT, elem_classes="markdown-text") # Requires import
+        #              gr.Markdown(EVALUATION_QUEUE_TEXT, elem_classes="markdown-text") # Requires import/definition
         #          with gr.Column():
         #              with gr.Accordion(f"✅ Finished Evaluations ({len(finished_eval_queue_df)})", open=False):
         #                   finished_eval_table = gr.components.Dataframe(
@@ -200,19 +211,18 @@ with demo:
         #          gr.Markdown("# ✉️✨ Submit your model here!", elem_classes="markdown-text")
         #     with gr.Row():
         #          with gr.Column():
-        #              model_name_textbox = gr.Textbox(label="Model name (on Hugging Face Hub)") # Clarified label
+        #              model_name_textbox = gr.Textbox(label="Model name (on Hugging Face Hub)")
         #              revision_name_textbox = gr.Textbox(label="Revision / Commit Hash", placeholder="main")
-        #              # Example choices, adjust as needed for your actual submission logic
-        #              model_type = gr.Dropdown(choices=["Type A", "Type B", "Type C"], label="Model type", multiselect=False, value=None, interactive=True)
+        #              model_type = gr.Dropdown(choices=["Type A", "Type B", "Type C"], label="Model type", multiselect=False, value=None, interactive=True) # Example choices
         #          with gr.Column():
         #              precision = gr.Dropdown(choices=["float16", "bfloat16", "float32", "int8", "auto"], label="Precision", multiselect=False, value="auto", interactive=True)
         #              weight_type = gr.Dropdown(choices=["Original", "Adapter", "Delta"], label="Weights type", multiselect=False, value="Original", interactive=True)
         #              base_model_name_textbox = gr.Textbox(label="Base model (for delta or adapter weights)")
         #     submit_button = gr.Button("Submit Eval")
-        #     submission_result = gr.Markdown() # To display success/failure message
-        #     # Ensure add_new_eval is correctly imported and handles these inputs
+        #     submission_result = gr.Markdown()
+        #     # Ensure add_new_eval is correctly imported/defined and handles these inputs
         #     submit_button.click(
-        #          add_new_eval, # Requires import
+        #          add_new_eval, # Requires import/definition
         #          [ model_name_textbox, base_model_name_textbox, revision_name_textbox, precision, weight_type, model_type, ],
         #          submission_result,
         #      )
@@ -220,11 +230,11 @@ with demo:
 
     # --- Citation Row (at the bottom, outside Tabs) ---
     with gr.Accordion("📙 Citation", open=False):
-         # Use the CITATION_BUTTON_TEXT and CITATION_BUTTON_LABEL variables imported from src.about
+         # Use the CITATION_BUTTON_TEXT and CITATION_BUTTON_LABEL variables imported or defined above
          citation_button = gr.Textbox(
              value=CITATION_BUTTON_TEXT,
              label=CITATION_BUTTON_LABEL,
-             lines=10, # Reduced lines slightly
+             lines=10,
              elem_id="citation-button",
              show_copy_button=True,
          )
@@ -238,5 +248,5 @@ with demo:
 # Ensures the app launches only when the script is run directly
 if __name__ == "__main__":
     # Ensure you have installed necessary libraries: pip install gradio pandas apscheduler
-    # Make sure your src module files (about.py etc.) are in the same directory or accessible in PYTHONPATH
+    # Make sure your src module files (about.py etc.) are accessible OR use the placeholder definitions above.
     demo.launch()