Update app.py

app.py

@@ -4,9 +4,8 @@ import time
 from typing import Any
 from communex.client import CommuneClient
 from communex.balance import from_nano
-import json
 
-netuid = 0
+netuid = 1
 node_url = "wss://commune.api.onfinality.io/public-ws"
 
 def get_com_price() -> Any:
@@ -21,37 +20,26 @@ def get_com_price() -> Any:
             time.sleep(retries)
     raise RuntimeError()
 
-def get_score():
+def get_leaderboard_data():
     com_price = get_com_price()
     blocks_in_day = 10_800
     client = CommuneClient(node_url)
     emission = client.query("Emission", params=[netuid])
-    distribution_dict: dict[int, float] = {}
+    scores = {}
     for uid, emi in enumerate(emission):
-        distribution_dict[uid] = ((emi / 10**11) * blocks_in_day) * com_price
-    return distribution_dict
-
-def load_leaderboard():
-    with open("leaderboard.json", "r") as f:
-        leaderboard_data = json.load(f)
-    return leaderboard_data
-
-def save_leaderboard():
-    scores = get_score()
+        scores[uid] = ((emi / 10**11) * blocks_in_day) * com_price
+    
     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": rank, "uid": uid, "score": score})
-    with open("leaderboard.json", "w") as f:
-        json.dump(leaderboard_data, f)
+    
+    return leaderboard_data
 
 with gr.Blocks() as demo:
     gr.Markdown("# Commune Leaderboard")
     
-    leaderboard_data = load_leaderboard()
-    
     leaderboard_table = gr.components.DataFrame(
-        value=leaderboard_data,
         headers=["rank", "uid", "score"],
         datatype=["number", "number", "number"],
         interactive=False,
@@ -61,11 +49,14 @@ with gr.Blocks() as demo:
     refresh_button = gr.Button("Refresh Leaderboard")
     
     def refresh_leaderboard():
-        save_leaderboard()
-        updated_data = load_leaderboard()
-        return gr.components.DataFrame.update(value=updated_data)
+        leaderboard_data = get_leaderboard_data()
+        return gr.components.DataFrame.update(value=leaderboard_data)
     
     refresh_button.click(fn=refresh_leaderboard, outputs=leaderboard_table)
+    
+    # Initial load of leaderboard data
+    leaderboard_data = get_leaderboard_data()
+    leaderboard_table.value = leaderboard_data
 
 if __name__ == "__main__":
     demo.launch()