adding labels

app.py

@@ -389,7 +389,7 @@ def show_leaderboard(task):
         st.dataframe(real_tmp, column_config=column_config)
 
 
-def make_roc(results):
+def make_roc(results,show_text = False):
     results["FA"] = 1.0 - results["real_accuracy"]
 
     chart = (
@@ -405,7 +405,26 @@ def make_roc(results):
             detail=["submission_id","auc","balanced_accuracy"]
         )
         .properties(width=400, height=400, title="Detection vs False Alarm vs Inference Time")
+        
     )
+    if show_text:
+        text = (
+            alt.Chart(results)
+            .mark_text(
+                align="right",
+                fontSize=14,
+                dx=-5,  # shift text to right of point
+                dy=-5,  # shift text slightly up
+            )
+        .encode(
+                x=alt.X("FA:Q", title="🧑‍🎤 False Positive Rate", scale=alt.Scale(domain=[0, 1])),
+                y=alt.Y("generated_accuracy:Q", title="👤 True Positive Rate", scale=alt.Scale(domain=[0, 1])),
+                color="team:N",  # Color by categorical field
+                text = "team",
+            )
+        )
+
+        chart = chart + text
 
     diag_line = (
         alt.Chart(pd.DataFrame(dict(tpr=[0, 1], fpr=[0, 1])))
@@ -413,10 +432,10 @@ def make_roc(results):
         .encode(x="fpr", y="tpr")
     )
 
-    return chart + diag_line
+    return (chart + diag_line)
 
 
-def make_acc(results):
+def make_acc(results,show_text = False):
     results = results.loc[results["total_time"] >= 0]
 
     chart = (
@@ -433,6 +452,32 @@ def make_acc(results):
         )
         .properties(width=400, height=400, title="Inference Time vs Balanced Accuracy")
     )
+
+    if show_text:
+        text = (
+            alt.Chart(results)
+            .mark_text(
+                align="right",
+                dx=-5,  # shift text to right of point
+                dy=-5,  # shift text slightly up
+                fontSize=14,
+            )
+        .encode(
+               x=alt.X("total_time:Q", title="🕒 Inference Time (sec)", scale=alt.Scale(type = "log",domain = [100,100000])),
+                y=alt.Y(
+                "balanced_accuracy:Q",
+                title="Balanced Accuracy",
+                scale=alt.Scale(domain=[0.4, 1]),
+            ),
+            color="team:N",  # Color by categorical field # Size by quantitative field
+                text = "team",
+            )
+        )
+
+        chart = chart + text
+
+
+
     diag_line = (
         alt.Chart(pd.DataFrame(dict(t=[100, 100000], y=[0.5, 0.5])))
         .mark_line(color="lightgray", strokeDash=[8, 4])
@@ -460,6 +505,8 @@ def show_charts(task):
     results = load_results(task, best_only=True)
     temp = results[f"{split}_score"].reset_index()
     teams = get_unique_teams(temp["team"])
+
+    best_only = True
    
     if split == "private":
 
@@ -486,8 +533,8 @@ def show_charts(task):
 
         # with st.spinner("making plots...", show_time=True):
         
-    roc_scatter = make_roc(temp)
-    acc_vs_time = make_acc(temp)
+    roc_scatter = make_roc(temp, show_text=best_only)
+    acc_vs_time = make_acc(temp, show_text=best_only)
 
     if split == "private" and hf_token is not None:
         if full_curves:
@@ -506,8 +553,6 @@ def make_plots_for_task(task):
 
     with t2:
         show_charts(task)
-        
-
 
 
 updated = get_updated_time()

utils.py

@@ -190,11 +190,13 @@ if __name__ == "__main__":
                           "safe-challenge/video-challenge-task-1-config",
                           "safe-challenge/video-challenge-task-2-config"]
     
-    # download_competition_data(competition_names=spaces)
+    download_competition_data(competition_names=spaces)
 
-
-    add_custom_submission(path_to_cache="competition_cache/safe-challenge/video-challenge-task-1-config",
-                          path_to_subfile="competition_cache/custom/Scores-DSRI-brian.txt")
+    try:
+        add_custom_submission(path_to_cache="competition_cache/safe-challenge/video-challenge-task-1-config",
+                            path_to_subfile="competition_cache/custom/Scores-DSRI-brian.txt")
+    except Exception as e:
+        print("no custom subs found.. skipping")
 
 
     ## Loop