Added link

app.py

@@ -45,7 +45,12 @@ download_meta()
 
 def build_demo():
     demo = gr.Blocks(title="Chatbot Arena Leaderboard", css=custom_css)
-    leaderboard_df = build_leadearboard_df()
+    try:
+        leaderboard_df = build_leadearboard_df()
+    except FileNotFoundError:
+        change_judgement(1)
+        return
+
     with demo:
         gr.HTML(TITLE)
         gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
@@ -79,6 +84,7 @@ def build_demo():
                 with gr.Column():
                     model_name_textbox = gr.Textbox(label="Model name")
                     submitter_username = gr.Textbox(label="Username")  # can we get this info from hf??
+                    model_link_web = gr.Textbox(label="Link to Model")  # can we get this info from hf??
 
                     def upload_file(file):
                         file_name = file.name.split("/")[-1] if "/" in file.name else file.name
@@ -94,6 +100,7 @@ def build_demo():
                                 "model_name": model_name_textbox,
                                 "username": submitter_username,
                                 "file_name": file_name,
+                                "link": model_link_web if model_link_web else "",
                             }
 
                             current_info.append(submit_info)
@@ -121,16 +128,7 @@ def build_demo():
                             repo_type="dataset",
                             token=HF_TOKEN_PRIVATE,
                         )
-                        with open(PERSISTENT_FILE_CHECK_PATH, "w", encoding="utf-8") as file:
-                            file.write("1")
-                        API.upload_file(
-                            path_or_fileobj=PERSISTENT_FILE_CHECK,
-                            path_in_repo=PERSISTENT_FILE_CHECK,
-                            repo_id=METAINFO_DATASET,
-                            repo_type="dataset",
-                            token=HF_TOKEN_PRIVATE,
-                        )
-                        os.environ[RESET_JUDGEMENT_ENV] = "1"
+                        change_judgement(1)
                         return file.name
 
                     if model_name_textbox and submitter_username:
@@ -152,14 +150,26 @@ def update_board():
     # the state is checked by the the file PERSISTENT_FILE_CHECK
     # very bad solution
     # but a fast one to code
-    need_reset = os.environ.get(RESET_JUDGEMENT_ENV)
+    need_reset = os.environ[RESET_JUDGEMENT_ENV] == "1"
     logging.info("Updating the judgement: %s", need_reset)
     if need_reset != "1":
         return
-    os.environ[RESET_JUDGEMENT_ENV] = "0"
     API.restart_space(repo_id=REPO_ID, token=H4_TOKEN)
 
 
+def change_judgement(need_recalc_state=0):
+    with open(PERSISTENT_FILE_CHECK_PATH, "w", encoding="utf-8") as file:
+        file.write(str(need_recalc_state))
+    API.upload_file(
+        path_or_fileobj=PERSISTENT_FILE_CHECK_PATH,
+        path_in_repo=PERSISTENT_FILE_CHECK,
+        repo_id=METAINFO_DATASET,
+        repo_type="dataset",
+        token=HF_TOKEN_PRIVATE,
+    )
+    os.environ[RESET_JUDGEMENT_ENV] = str(need_recalc_state)
+
+
 if __name__ == "__main__":
     os.environ[RESET_JUDGEMENT_ENV] = "0"
 
@@ -186,15 +196,7 @@ if __name__ == "__main__":
         # may lead to multiple attempts to get judements
         # if code above throws error
         # move to before judgement to saafeguard from that
-        with open(PERSISTENT_FILE_CHECK_PATH, "w", encoding="utf-8") as file:
-            file.write("0")
-        API.upload_file(
-            path_or_fileobj=PERSISTENT_FILE_CHECK_PATH,
-            path_in_repo=PERSISTENT_FILE_CHECK,
-            repo_id=METAINFO_DATASET,
-            repo_type="dataset",
-            token=HF_TOKEN_PRIVATE,
-        )
+        change_judgement(0)
 
     scheduler = BackgroundScheduler()
     scheduler.add_job(update_board, "interval", minutes=10)

src/leaderboard/build_leaderboard.py

@@ -64,8 +64,8 @@ def build_leadearboard_df():
         battle_info = pd.DataFrame.from_records(json.load(eval_file))[["model", "score", "lower", "upper"]]
     with open(f"{DATA_PATH}/{SUBMITS_META_FILE}", "r", encoding="utf-8") as submit_meta_file:
         submit_info = pd.DataFrame.from_records(
-            json.load(submit_meta_file), columns=["model_name", "username", "file_name"]
-        )[["model_name", "username", "file_name"]]
+            json.load(submit_meta_file), columns=["model_name", "username", "file_name", "link"]
+        )[["model_name", "username", "file_name", "link"]]
 
     # Merge the DataFrames
     merged_df = pd.merge(battle_info, submit_info, how="left", left_on="model", right_on="model_name")