Update app.py

app.py

@@ -7,7 +7,6 @@ from gradio.themes.utils import colors, fonts, sizes
 from gradio.themes.utils.colors import Color
 from typing import Iterable
 from communex.client import CommuneClient
-from communex.balance import from_nano
 
 FONT = """<link href="https://fonts.googleapis.com/css2?family=Roboto+Mono&display=swap" rel="stylesheet">"""
 
@@ -16,11 +15,11 @@ TITLE = """<h1 align="center" id="space-title" class="typewriter">Synthia Leader
 IMAGE = """<a href="https://www.communeai.org/commune-logo.svg" target="_blank"><img src="https://www.communeai.org/commune-logo.svg" alt="Synthia Logo" style="margin: auto; width: 20%; border: 0;" /></a>"""
 
 HEADER = """<h2 align="center" class="typewriter">Welcome to the Synthia Leaderboard!</h2>
-<p align="center">This leaderboard showcases the top-performing models in the Synthia competition. The models are ranked based on their daily rewards and perplexity scores.</p>"""
+<p align="center">This leaderboard showcases the top-performing models in the Synthia Subnet. The models are ranked based on their daily rewards.</p>"""
 
 EVALUATION_HEADER = """<h3 align="center">Evaluation Details</h3>"""
 
-EVALUATION_DETAILS = """<p align="center"><b>Name</b> represents the model name. <b>Rewards / Day</b> indicates the expected daily rewards for each model. <b>Perplexity</b> represents the model's performance on the evaluation data (lower is better). <b>UID</b> is the unique identifier of the model submitter. <b>Block</b> represents the block number when the model was submitted.</p>"""
+EVALUATION_DETAILS = """<p align="center"><b>Name</b> represents the model name. <b>Rewards / Day</b> indicates the expected daily rewards for each model in $COMAI. <b>UID</b> is the unique identifier of the model submitter. <b>$USD Value</b> is the estimated dollar value of the daily rewards.</p>"""
 netuid = 24
 node_url = "wss://commune.api.onfinality.io/public-ws"
 
@@ -40,24 +39,50 @@ def get_com_price() -> Any:
             print(f"Error fetching COM price: {e}")
             if i == retries - 1:
                 raise
-            asyncio.sleep(retries)
+            _ = asyncio.sleep(retries)
     raise RuntimeError("Failed to fetch COM price")
 
 
+def make_query(client: CommuneClient) -> tuple[dict[int, int], dict[int, str]]:
+    request_dict: dict[Any, Any] = {
+        "SubspaceModule": [
+            ("Name", [netuid]),
+            ("Emission", []),
+        ],
+    }
+    emission_dict: dict[int, int] = {}
+    name_dict: dict[int, str] = {}
+    result = client.query_batch_map(request_dict)
+    emission = result["Emission"]
+    netuid_emission = emission[netuid]
+    names = result["Name"]
+    highest_uid = max(names.keys())
+    for uid in range(highest_uid + 1):
+        emission_dict[uid] = netuid_emission[uid]
+        name_dict[uid] = names[uid]
+
+    return emission_dict, name_dict
+
+
 async def get_leaderboard_data():
     try:
         com_price = get_com_price()
         blocks_in_day = 10_800
         client = CommuneClient(node_url)
-        emission = client.query("Emission", params=[netuid])
+        emission_dict, name_dict = make_query(client)
         print("got tha emission")
         scores = {}
-        for uid, emi in enumerate(emission):
-            scores[uid] = ((emi / 10**11) * blocks_in_day) * com_price
+        for uid, emi in emission_dict.items():
+            # we are doing ** 11 because emission are for modules have 2 extra units,
+            # (otherwise it would be more intuive to do ** 9, hence from_nano)
+            scores[uid] = (emi / 10**11) * blocks_in_day
         sorted_scores = sorted(scores.items(), key=lambda x: x[1], reverse=True)
         leaderboard_data = []
         for rank, (uid, score) in enumerate(sorted_scores, start=1):
-            leaderboard_data.append((rank, uid, f"${score:.2f}"))
+            name = name_dict[uid]
+            units = score
+            usd_value = score * com_price
+            leaderboard_data.append((rank, uid, name, units, f"${usd_value:.2f}"))
         return leaderboard_data
     except Exception as e:
         print(f"Error fetching leaderboard data: {e}")
@@ -145,7 +170,7 @@ with gr.Blocks(theme=seafoam, analytics_enabled=True) as demo:
     gr.HTML(EVALUATION_DETAILS)
 
     leaderboard_table = gr.components.Dataframe(
-        headers=["Rank 🏆", "User 😄", "Score 💰"],
+        headers=["Rank 🏆", "UID", "Name", "Rewards / Day", "$USD Value"],
         datatype=["str", "str", "str"],
         interactive=False,
         visible=True,