Update app.py

app.py

@@ -4,8 +4,8 @@ import time
 from threading import Thread
 
 # Initialize clients
-client1 = Client("orionai/training-data-collection_2")
-client2 = Client("orionai/training-data-collection_3")
+client1 = Client("orionai/training-data-collection-2")
+client2 = Client("orionai/training-data-collection-3")
 client3 = Client("orionai/training-data-collection")
 
 state = {
@@ -52,18 +52,23 @@ Thread(target=monitor_growth, daemon=True).start()
 def get_dashboard_metrics():
     return state["prev_total_tokens"], state["total_growth_speed"]
 
+def update_dashboard():
+    tokens, speed = get_dashboard_metrics()
+    return tokens, speed
+
 # Create Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# Token Count Dashboard")
+    
     total_tokens = gr.Number(label="Total Token Count", value=0)
     growth_speed = gr.Number(label="Total Growth Speed", value=0)
     
-    def update_dashboard():
-        tokens, speed = get_dashboard_metrics()
+    def refresh_metrics():
+        tokens, speed = update_dashboard()
         total_tokens.update(value=tokens)
         growth_speed.update(value=speed)
     
-    # Automatically update the metrics every 5 seconds
-    gr.Timer(5, update_dashboard)
-
+    update_button = gr.Button("Update Stats")
+    update_button.click(refresh_metrics)
+    
 demo.launch()