update

indiebot_arena/ui/leaderboard.py

@@ -10,6 +10,7 @@ DESCRIPTION = "### 🏆️ リーダーボード"
 base_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", ".."))
 docs_path = os.path.join(base_dir, "docs", "leaderboard_header.md")
 
+
 def leaderboard_content(dao, language):
   arena_service = ArenaService(dao)
 
@@ -27,6 +28,26 @@ def leaderboard_content(dao, language):
       data = [["No data available", "", "", "", ""]]
     df = pd.DataFrame(data, columns=["Model Name", "Elo Score", "File Size (GB)", "Description", "Last Updated"])
     df.insert(0, "Rank", range(1, len(df) + 1))
+
+    def add_link(row):
+      raw_name = row["Model Name"]
+      if raw_name!="Unknown":
+        return f'<a href="https://huggingface.co/{raw_name}" target="_blank">{raw_name}</a>'
+      else:
+        return raw_name
+
+    df["Model Name"] = df.apply(add_link, axis=1)
+
+    def add_emoji(row):
+      rank = row["Rank"]
+      linked_name = row["Model Name"]
+      emoji = {1: "🥇", 2: "🥈", 3: "🥉"}.get(rank, "")
+      if emoji:
+        return f'{emoji} {linked_name}'
+      else:
+        return linked_name
+
+    df["Model Name"] = df.apply(add_emoji, axis=1)
     return df
 
   initial_weight_class = "U-5GB"
@@ -38,7 +59,8 @@ def leaderboard_content(dao, language):
     weight_class_radio = gr.Radio(choices=["U-5GB", "U-10GB"], label="階級", value=initial_weight_class)
     leaderboard_table = gr.Dataframe(
       headers=["Rank", "Model Name", "Elo Score", "File Size (GB)", "Description", "Last Updated"],
-      interactive=False
+      interactive=False,
+      datatype="markdown"
     )
     refresh_btn = gr.Button("更新", variant="primary")
     refresh_btn.click(fn=fetch_leaderboard_data, inputs=weight_class_radio, outputs=leaderboard_table)