Update app.py

app.py

@@ -1,11 +1,39 @@
 import gradio as gr
+import pandas as pd
 
 from about import ABOUT_TEXT, INTRO_TEXT, SUBMISSION_TEXT, TITLE
 from utils import load_results_dataframe, submit_zip
 
 
+EMPTY_COLUMNS = [
+    "rank",
+    "model_name",
+    "team_name",
+    "avg_vrmse",
+    "density_vrmse",
+    "pressure_vrmse",
+    "velocity_x_vrmse",
+    "velocity_y_vrmse",
+    "submitted_at",
+    "status",
+]
+
+
 def refresh_leaderboard():
-    return load_results_dataframe()
+    try:
+        df = load_results_dataframe()
+        if df is None or not isinstance(df, pd.DataFrame):
+            df = pd.DataFrame(columns=EMPTY_COLUMNS)
+        return df, "Leaderboard loaded."
+    except Exception as exc:
+        return pd.DataFrame(columns=EMPTY_COLUMNS), f"Error loading leaderboard: {exc}"
+
+
+def handle_submit(zip_file):
+    try:
+        return submit_zip(zip_file)
+    except Exception as exc:
+        return f"Submission failed: {exc}"
 
 
 with gr.Blocks(title="The Well Leaderboard MVP") as demo:
@@ -13,36 +41,35 @@ with gr.Blocks(title="The Well Leaderboard MVP") as demo:
     gr.Markdown(INTRO_TEXT)
 
     with gr.Tab("Leaderboard"):
+        leaderboard_status = gr.Markdown("Loading leaderboard...")
         leaderboard = gr.Dataframe(
-            value=load_results_dataframe(),
+            headers=EMPTY_COLUMNS,
+            value=pd.DataFrame(columns=EMPTY_COLUMNS),
             label="Ranked results",
             wrap=True,
+            interactive=False,
         )
-
         refresh_button = gr.Button("Refresh leaderboard")
         refresh_button.click(
             fn=refresh_leaderboard,
-            outputs=leaderboard,
-            api_name="refresh_leaderboard",
+            outputs=[leaderboard, leaderboard_status],
+            api_name=False,
         )
 
     with gr.Tab("Submit"):
         gr.Markdown(SUBMISSION_TEXT)
-
         zip_file = gr.File(
             label="Submission zip",
             file_count="single",
             file_types=[".zip"],
         )
-
         submit_button = gr.Button("Submit")
         submission_status = gr.Markdown()
-
         submit_button.click(
-            fn=submit_zip,
+            fn=handle_submit,
             inputs=zip_file,
             outputs=submission_status,
-            api_name="submit_zip",
+            api_name=False,
         )
 
     with gr.Tab("About"):
@@ -50,9 +77,8 @@ with gr.Blocks(title="The Well Leaderboard MVP") as demo:
 
     demo.load(
         fn=refresh_leaderboard,
-        outputs=leaderboard,
+        outputs=[leaderboard, leaderboard_status],
+        api_name=False,
     )
 
-
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()