added some info for submits

app.py

@@ -87,15 +87,16 @@ def build_demo():
                             current_info = json.loads(submit_meta_file)
 
                             # for now just do not save same name model
-                            if model_name_textbox in current_info:
+                            if any(filter(lambda x: x["model_name"] == model_name_textbox, current_info)):
                                 return False
 
                             submit_info = {
+                                "model_name": model_name_textbox,
                                 "username": submitter_username,
                                 "file_name": file_name,
                             }
 
-                            current_info[model_name_textbox] = submit_info
+                            current_info.append(submit_info)
 
                         with open(f"{DATA_PATH}/{SUBMITS_META_FILE}", "w", encoding="utf-8") as submit_meta_file:
                             submit_meta_file.write(json.dumps(current_info))

src/leaderboard/build_leaderboard.py

@@ -61,7 +61,17 @@ def download_meta():
 def build_leadearboard_df():
     # Retrieve the leaderboard DataFrame
     with open(f"{DATA_PATH}/leaderboard.json", "r", encoding="utf-8") as eval_file:
-        battle_info = pd.DataFrame.from_records(json.load(eval_file))
+        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))
-    return battle_info.copy()
+        submit_info = pd.DataFrame.from_records(
+            json.load(submit_meta_file), columns=["model_name", "username", "file_name"]
+        )[["model_name", "username", "file_name"]]
+
+    # Merge the DataFrames
+    merged_df = pd.merge(battle_info, submit_info, how="left", left_on="model", right_on="model_name")
+    # Fill missing usernames with None
+    merged_df["username"] = merged_df["username"].fillna("undefined")
+    # Drop the model_name column if no longer needed
+    merged_df = merged_df.drop(columns=["model_name", "file_name"])
+
+    return merged_df.copy()