added restart

src/submission/submit.py

@@ -8,6 +8,8 @@ import tempfile
 from datetime import datetime
 from typing import Dict, List, Tuple
 import shutil
+import threading
+import time
 
 from huggingface_hub import HfApi
 from datasets import load_dataset
@@ -99,6 +101,24 @@ def submit_leaderboard_to_hub(entries: List[Dict], version="v0") -> Tuple[bool,
         return False, f"Error updating leaderboard: {e}"
 
 
+def restart_space_after_delay(delay_seconds: int = 2) -> None:
+    """
+    Restart the Hugging Face Space after a delay.
+    """
+    def _restart_space():
+        time.sleep(delay_seconds)
+        try:
+            api = HfApi(token=TOKEN)
+            api.restart_space(repo_id=REPO_ID)
+        except Exception as e:
+            print(f"Error restarting space: {e}")
+
+    # Start the restart in a separate thread
+    thread = threading.Thread(target=_restart_space)
+    thread.daemon = True
+    thread.start()
+
+
 def process_submission(file_path: str, metadata: Dict, version="v0") -> str:
     """
     Process a submission to the GuardBench leaderboard.
@@ -212,7 +232,10 @@ def process_submission(file_path: str, metadata: Dict, version="v0") -> str:
             success, message = submit_leaderboard_to_hub(all_entries, version)
             if not success:
                 return styled_error(message)
-            return styled_message(f"Submission successful! Model evaluated and leaderboard updated.")
+
+            restart_space_after_delay(5)
+
+            return styled_message("Submission successful! Model evaluated and leaderboard updated.")
 
         except Exception as eval_error:
             return styled_error(f"Error during evaluation: {eval_error}")