Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 from gradio_client import Client
+import time
+from collections import deque
 
 # Initialize clients
 client1 = Client("orionai/training-data-collection_2")
@@ -14,6 +16,8 @@ state = {
     "total_growth_speed": 0
 }
 
+token_counts = deque(maxlen=60)  # Store up to 60 seconds of token counts
+
 def get_token_count(client):
     try:
         result = client.predict(api_name="/update_token_display")
@@ -27,19 +31,22 @@ def update_dashboard():
         curr_count1 = get_token_count(client1)
         curr_count2 = get_token_count(client2)
         curr_count3 = get_token_count(client3)
-        growth_speed1 = curr_count1 - state["prev_count1"]
-        growth_speed2 = curr_count2 - state["prev_count2"]
-        growth_speed3 = curr_count3 - state["prev_count3"]
+        
         total_tokens = curr_count1 + curr_count2 + curr_count3
-        total_growth_speed = growth_speed1 + growth_speed2 + growth_speed3
-
+        token_counts.append(total_tokens)
+        
+        if len(token_counts) > 1:
+            growth_speed = (token_counts[-1] - token_counts[0]) / len(token_counts)
+        else:
+            growth_speed = 0
+        
         state["prev_count1"] = curr_count1
         state["prev_count2"] = curr_count2
         state["prev_count3"] = curr_count3
         state["prev_total_tokens"] = total_tokens
-        state["total_growth_speed"] = total_growth_speed
+        state["total_growth_speed"] = growth_speed
 
-        return total_tokens, total_growth_speed
+        return total_tokens, growth_speed
     except Exception as e:
         print(f"Error in update dashboard: {e}")
         return 0, 0
@@ -53,7 +60,7 @@ with gr.Blocks() as demo:
     gr.Markdown("# Token Count Dashboard")
     
     total_tokens = gr.Number(label="Total Token Count")
-    growth_speed = gr.Number(label="Total Growth Speed")
+    growth_speed = gr.Number(label="Average Growth Speed (per second over the last minute)")
     
     update_button = gr.Button("Update Stats")
     update_button.click(fn=refresh_metrics, inputs=[], outputs=[total_tokens, growth_speed])