update leader info fix

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 import os
 import time
 import datetime
-from utils import fetch_competition_summary, fetch_models_evaluation, get_all_competition_summary, update_leader_info, highlight_score_column
+from utils import fetch_competition_summary, fetch_models_evaluation, get_all_competition_summary, highlight_score_column
 # from dotenv import load_dotenv
 from cfg import * 
 # load_dotenv()
@@ -28,7 +28,28 @@ else:
 
 wandb_api = wandb.Api()
 
-
+@st.cache_data
+def update_leader_info(leader_info, competition, best_model):
+    if leader_info.get(competition) is None:
+        leader_info[competition] = {
+            "Username": best_model["Username"],
+            "Hotkey": best_model["Hotkey"],
+            "Date": time.strftime("%Y-%m-%d"),
+            "Days on Top": 1
+        }
+    else:
+        if leader_info[competition]["Hotkey"] == best_model["Hotkey"]:
+            # count the number of days on top
+            start_date = datetime.datetime.strptime(leader_info[competition]["Date"], "%Y-%m-%d")
+            current_date = datetime.datetime.now()
+            days_on_top = (current_date - start_date).days
+            leader_info[competition]["Days on Top"] = days_on_top + 1
+        else:
+            leader_info[competition]["Username"] = best_model["Username"]
+            leader_info[competition]["Hotkey"] = best_model["Hotkey"]
+            leader_info[competition]["Date"] = time.strftime("%Y-%m-%d")
+            leader_info[competition]["Days on Top"] = 1
+    return leader_info[competition]
 
 @st.cache_data()
 def load_competition_data(last_update_time=None):

utils.py

@@ -1,31 +1,11 @@
 import streamlit as st
 import pandas as pd
 import wandb
+import time
 from datetime import datetime
 
 
 
-def update_leader_info(leader_info, competition, best_model):
-    current_date = datetime.now().strftime("%Y-%m-%d")
-    if leader_info.get(competition) is None:
-        leader_info[competition] = {
-            "Username": best_model["Username"],
-            "Hotkey": best_model["Hotkey"],
-            "Date": current_date,
-            "Days on Top": 1
-        }
-    else:
-        if leader_info[competition]["UID"] == best_model["ID"]:
-            leader_info[competition]["Days on Top"] += 1
-        else:
-            leader_info[competition] = {
-                "Username": best_model["Username"],
-                "Hotkey": best_model["Hotkey"],
-                "Date": current_date,
-                "Days on Top": 1
-            }
-    return leader_info[competition]
-
 
 def get_all_competition_summary(api, projects):