add main metrics

app.py

@@ -1,20 +1,28 @@
+from dashboard_utils.main_metrics import get_main_metrics
 import streamlit as st
 import wandb
 from streamlit_observable import observable
 
 from dashboard_utils.bubbles import get_new_bubble_data
 
-st.title("Training transformers together dashboard")
-st.write("test")
 wandb.login(anonymous="must")
 
-serialized_data, profiles = get_new_bubble_data()
+st.title("Training transformers together dashboard")
+st.header("Training Loss")
 
+steps, losses, alive_peers = get_main_metrics()
 
+st.line_chart(data={"steps": steps, "loss":losses})
+
+st.header("Collaborative training participants")
+st.header("Snapshot")
+serialized_data, profiles = get_new_bubble_data()
 observers = observable(
     "Participants",
     notebook="d/9ae236a507f54046",  # "@huggingface/participants-bubbles-chart",
     targets=["c_noaws"],
-    # observe=["selectedCounties"]
     redefine={"serializedData": serialized_data, "profileSimple": profiles},
 )
+
+st.header("Overtime")
+st.line_chart(data={"steps": steps, "alive participants":alive_peers})

dashboard_utils/main_metrics.py

@@ -0,0 +1,19 @@
+import wandb
+
+WANDB_REPO = "learning-at-home/Main_metrics"
+
+def get_main_metrics():
+    api = wandb.Api()
+    runs = api.runs(WANDB_REPO)
+    run = runs[0]
+    history = run.scan_history(keys=["step", "loss", "alive peers"])
+
+    steps = []
+    losses = []
+    alive_peers = []
+    for row in history:
+        steps.append(row["step"])
+        losses.append(row["loss"])
+        alive_peers.append(row["alive peers"])
+
+    return steps, losses, alive_peers