Added the button for restart the space

app.py

@@ -16,8 +16,10 @@ from content import (
     MORE_DETAILS_MARKDOWN,
     ABOUT_MARKDOWN,
 )
-from server import LeaderboardServer, xmlAndMarkdownEscape, xmlQuoteAttr
+from server import LeaderboardServer, xmlAndMarkdownEscape, xmlQuoteAttr, api
 
+HF_SPACE_TOKEN = os.environ["HF_SPACE_TOKEN"]
+HF_SPACE_ID = os.environ["HF_SPACE_ID"]
 # For testing purpose
 HF_DISABLE_SUBMIT = bool(int(os.environ.get("HF_DISABLE_SUBMIT", "0")))
 from server import HF_FAKE_TOURNAMENT
@@ -294,14 +296,30 @@ def results_dataset_integrity_check():
 def update_results_dataset_integrity_solving_progress():
     progress = leaderboard_server.tournament_results_integrity_solving_progress * 100
     
-    if progress < 100:
-        return gr.update(
+    
+    if leaderboard_server.tournament_results_corrupted:
+        results_dataset_integrity_solving_progress = gr.update(
             value=f"Tournament results is preparing, please wait… {progress:.0f}% done"
         )
+        
+        results_dataset_integrity_restart_btn = gr.update(
+            interactive=False,
+            visible=False,
+        )
     else:
-        return gr.update(
-            value="Tournament results is prepared, please refresh this page in your browser (e.g. by pressing F5)."
+        results_dataset_integrity_solving_progress = gr.update(
+            value="Tournament results is prepared, please restart this space with the button showed below."
         )
+        
+        results_dataset_integrity_restart_btn = gr.update(
+            interactive=True,
+            visible=True,
+        )
+    
+    return (
+        results_dataset_integrity_solving_progress,
+        results_dataset_integrity_restart_btn,
+    )
 
 def on_tournament_results_corrupted():
     if leaderboard_server.tournament_results_corrupted and not leaderboard_server.tournament_results_integrity_solving:
@@ -326,6 +344,9 @@ def on_tournament_results_corrupted():
         results_dataset_integrity_solving_progress,
     )
 
+def restart_space():
+    api.restart_space(repo_id=HF_SPACE_ID, token=HF_SPACE_TOKEN)
+
 def on_application_load():
     with leaderboard_server.var_lock.ro:
         leaderboard = gr.update(
@@ -393,13 +414,22 @@ def gradio_app():
                 visible=False,
             )
             
+            results_dataset_integrity_restart_btn = gr.Button(
+                value='Restart this space',
+                interactive=False,
+                visible=False,
+            )
+            
             results_dataset_integrity_solving_progress_timer = gr.Timer(
                 value=60,  # seconds
             )
             
             results_dataset_integrity_solving_progress_timer.tick(
                 fn=update_results_dataset_integrity_solving_progress,
-                outputs=results_dataset_integrity_solving_progress
+                outputs=[
+                    results_dataset_integrity_solving_progress,
+                    results_dataset_integrity_restart_btn,
+                ]
             )
             
             results_dataset_integrity_check_btn.click(
@@ -415,6 +445,13 @@ def gradio_app():
                 fn=results_dataset_integrity_check
             )
             
+            results_dataset_integrity_restart_btn.click(
+                fn=lambda: gr.update(interactive=False),
+                outputs=results_dataset_integrity_restart_btn
+            ).then(
+                fn=restart_space
+            )
+            
             main.load(
                 fn=on_tournament_results_corrupted,
                 outputs=[
@@ -423,7 +460,10 @@ def gradio_app():
                 ]
             ).then(
                 fn=update_results_dataset_integrity_solving_progress,
-                outputs=results_dataset_integrity_solving_progress
+                outputs=[
+                    results_dataset_integrity_solving_progress,
+                    results_dataset_integrity_restart_btn,
+                ]
             )
             
             return main

server.py

@@ -30,9 +30,10 @@ VISIBLE_METRICS = SUPPORTED_METRICS + ["macro_f1"]
 
 api = HfApi()
 
+HF_TOKEN = os.environ["HF_TOKEN"]
+HF_RESULTS_DATASET = os.environ["HF_RESULTS_DATASET"]  # <HF_RESULTS_DATASET> ::= <owner> "/" <dataset name>; e.g. CZLC/LLM_benchmark_data
+# For testing purpose
 HF_FAKE_TOURNAMENT = bool(int(os.environ.get("HF_FAKE_TOURNAMENT", "0")))
-HF_RESULTS_DATASET = os.environ.get("HF_RESULTS_DATASET")  # <HF_RESULTS_DATASET> ::= <owner> "/" <dataset name>; e.g. CZLC/LLM_benchmark_data
-HF_TOKEN = os.environ.get("HF_TOKEN")
 
 TASKS_METADATA_PATH = "./tasks_metadata.json"