Expend area and clean up

app.py

@@ -39,7 +39,7 @@ GRAPH_HISTORY_DAYS = 30
 MAX_GRAPH_ENTRIES = 10
 
 wandb_api = wandb.Api()
-demo = gr.Blocks(css=".typewriter {font-family: 'JMH Typewriter', sans-serif;}")
+demo = gr.Blocks(css=".typewriter {font-family: 'JMH Typewriter', sans-serif;}", fill_height=True, fill_width=True)
 runs: dict[int, list[Run]] = {}
 
 
@@ -47,7 +47,7 @@ runs: dict[int, list[Run]] = {}
 class LeaderboardEntry:
     uid: int
     winner: bool
-    model: str
+    repository: str
     score: float
     similarity: float
     hotkey: str
@@ -194,7 +194,7 @@ def create_leaderboard(runs: list[Run]) -> list[tuple]:
                 entries[uid] = LeaderboardEntry(
                     uid=uid,
                     winner="winner" in value,
-                    model=run.summary["submissions"][str(uid)]["repository"],
+                    repository=run.summary["submissions"][str(uid)]["repository"],
                     score=calculate_score(baseline_generation_time, generation_time, similarity),
                     similarity=similarity,
                     baseline_generation_time=baseline_generation_time,
@@ -211,7 +211,7 @@ def create_leaderboard(runs: list[Run]) -> list[tuple]:
     return [(
         entry.uid,
         f"<span style='color: {'springgreen' if entry.winner else 'red'}'>{entry.winner}</span>",
-        entry.model,
+        entry.repository,
         round(entry.score, 3),
         f"{entry.generation_time:.3f}s",
         f"{entry.similarity:.3f}",
@@ -251,25 +251,20 @@ def fetch_wandb_data():
     runs = dict(sorted(runs.items(), key=lambda item: item[0]))
 
 
-def get_choices() -> list[str]:
-    choices = []
+def get_choices() -> list[tuple[str, int]]:
+    choices: list[tuple[str, int]] = []
     for uid, run in runs.items():
         hotkey = run[0].config["hotkey"]
+        benchmarks = dict(run[0].summary.get("benchmarks", {}))
+        finished = any("winner" in value for value in benchmarks.values())
+        progress_text = "Finished" if finished else "In Progress"
         if hotkey in HOTKEYS_TO_KNOWN_VALIDATORS:
-            choices.append(f"{uid} - {HOTKEYS_TO_KNOWN_VALIDATORS[hotkey]}")
+            choices.append((f"{uid} - {HOTKEYS_TO_KNOWN_VALIDATORS[hotkey]} ({progress_text})", uid))
         else:
-            choices.append(str(f"{uid} - {hotkey}"))
+            choices.append((str(f"{uid} - {hotkey}"), uid))
     return choices
 
 
-def get_default_choice() -> str:
-    return f"{SOURCE_VALIDATOR_UID} - {HOTKEYS_TO_KNOWN_VALIDATORS[runs[SOURCE_VALIDATOR_UID][0].config['hotkey']]}"
-
-
-def parse_uid(choice: str) -> int:
-    return int(choice[:choice.index(" ")])
-
-
 def refresh():
     fetch_wandb_data()
     demo.clear()
@@ -283,6 +278,7 @@ def refresh():
             interactive=False,
             show_fullscreen_button=False,
             show_share_button=False,
+            container=False,
         )
 
         gr.Markdown(
@@ -297,24 +293,24 @@ def refresh():
         with gr.Accordion(f"Contest #1 Submission Leader: New Dream SDXL on NVIDIA RTX 4090s (Last updated: {now.strftime('%Y-%m-%d %I:%M:%S %p')} EST)"):
             dropdown = gr.Dropdown(
                 get_choices(),
-                value=get_default_choice(),
+                value=SOURCE_VALIDATOR_UID,
                 interactive=True,
                 label="Source Validator"
             )
 
-            graph = gr.Plot()
 
             leaderboard = gr.components.Dataframe(
-                create_leaderboard(runs[parse_uid(dropdown.value)]),
+                create_leaderboard(runs[dropdown.value]),
                 headers=["Uid", "Winner", "Model", "Score", "Gen Time", "Similarity", "Size", "VRAM Usage", "Power Usage", "Hotkey"],
                 datatype=["number", "markdown", "markdown", "number", "markdown", "number", "markdown", "markdown", "markdown", "markdown"],
                 elem_id="leaderboard-table",
             )
 
-            demo.load(lambda uid: create_graph(runs[parse_uid(uid)]), [dropdown], [graph])
+            graph = gr.Plot()
+            demo.load(lambda uid: create_graph(runs[uid]), [dropdown], [graph])
 
-            dropdown.change(lambda uid: create_graph(runs[parse_uid(uid)]), [dropdown], [graph])
-            dropdown.change(lambda uid: create_leaderboard(runs[parse_uid(uid)]), [dropdown], [leaderboard])
+            dropdown.change(lambda uid: create_graph(runs[uid]), [dropdown], [graph])
+            dropdown.change(lambda uid: create_leaderboard(runs[uid]), [dropdown], [leaderboard])
 
 
 if __name__ == "__main__":