Colorize

app.py

@@ -45,13 +45,19 @@ class State:
 
 # enum
 class Status(Enum):
- IN_PROGRESS = "In Progress"
- DONE = "Done"
- BROKEN = "Broken"
+ IN_PROGRESS = ("In Progress", "orange")
+ DONE = ("Done", "springgreen")
+ BROKEN = ("Broken", "red")
 
- def get_alt_time_text(self):
+ def get_alt_time_text(self) -> str:
  return "∞" if self == Status.BROKEN else "N/A"
 
+ def name(self):
+ return self.value[0]
+
+ def color(self):
+ return self.value[1]
+
 
 wandb_api = wandb.Api()
 demo = gr.Blocks(css=".typewriter {font-family: 'JMH Typewriter', sans-serif;}")
@@ -123,6 +129,15 @@ def fetch_wandb_data():
  data = dict(sorted(data.items()))
 
 
+def get_latest_version() -> str:
+ latest_version = "0.0.0"
+ for source_validator_uid, state in data.items():
+ version = state.version
+ if version > latest_version:
+ latest_version = version
+ return latest_version
+
+
 def refresh():
  fetch_wandb_data()
  demo.clear()
@@ -131,6 +146,7 @@ def refresh():
  with demo:
  with gr.Accordion(f"Validator States (Last updated: {now.strftime('%Y-%m-%d %I:%M:%S %p')} EST)"):
  elements: list[tuple] = []
+ latest_version = get_latest_version()
 
  for source_validator_uid, state in data.items():
  status = Status.IN_PROGRESS
@@ -144,26 +160,26 @@ def refresh():
  eta_date = now + time_left
  eta_time = eta_date.strftime("%Y-%m-%d %I:%M:%S %p") if eta > 0 else status.get_alt_time_text()
 
+ average_time_text = f"{timedelta(seconds=int(state.average_benchmark_time))}" if state.average_benchmark_time else status.get_alt_time_text(),
+
  elements.append((
  source_validator_uid,
  HOTKEYS_TO_KNOWN_VALIDATORS.get(state.hotkey, state.hotkey),
- state.version,
- status.value,
+ f"<span style='color: {'springgreen' if state.version == latest_version else 'red'}'>{state.version}</span>",
+ f"<span style='color: {status.color()}'>{status.name()}</span>",
  state.benchmarks + state.invalid,
  state.submissions,
  state.invalid,
- f"{timedelta(seconds=int(state.average_benchmark_time))}" if state.average_benchmark_time else status.get_alt_time_text(),
- eta_time,
- time_left if eta > 0 else status.get_alt_time_text(),
+ f"<span style='color: {'orange' if state.average_benchmark_time > 600 else 'springgreen'}'>{average_time_text[0]}</span>",
+ f"<span style='color: {'orange' if eta > 43200 else 'springgreen'}'>{eta_time}</span>",
+ f"<span style='color: {'orange' if eta > 43200 else 'springgreen'}'>{time_left if eta > 0 else status.get_alt_time_text()}</span>"
  ))
 
  gr.components.Dataframe(
  elements,
  headers=["UID", "Name", "Version", "Status", "Tested", "Submissions", "Invalid", "Avg. Benchmark Time", "ETA (Eastern Time)", "ETA Remaining"],
- datatype=["number", "markdown", "markdown", "markdown", "number", "number", "number", "number", "markdown", "markdown"],
+ datatype=["number", "markdown", "markdown", "markdown", "number", "number", "number", "markdown", "markdown", "markdown"],
  elem_id="state-table",
- interactive=False,
- visible=True,
  )