Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import time
-from threading import Thread
+from threading import Thread, Lock
 from gradio_client import Client
 from tenacity import retry, stop_after_attempt, wait_fixed, retry_if_exception_type
 import httpx
@@ -18,6 +18,8 @@ state = {
     "total_growth_speed": 0
 }
 
+state_lock = Lock()
+
 @retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry=retry_if_exception_type(httpx.ReadTimeout))
 def get_token_count(client):
     result = client.predict(api_name="/update_token_display")
@@ -29,21 +31,25 @@ def monitor_growth():
             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
-
-            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
-
-            time.sleep(1)
+            
+            with state_lock:
+                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
+
+                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
+
+            time.sleep(5)  # Increased sleep duration
         except httpx.ReadTimeout:
             print("Timeout occurred while fetching token count.")
+        except Exception as e:
+            print(f"An error occurred: {e}")
 
 # Start the monitor thread
 Thread(target=monitor_growth, daemon=True).start()
@@ -56,8 +62,9 @@ growth_speed_placeholder = st.empty()
 
 def update_ui():
     while True:
-        token_count_placeholder.metric("Total Token Count", state["prev_total_tokens"])
-        growth_speed_placeholder.metric("Total Growth Speed", state["total_growth_speed"])
+        with state_lock:
+            token_count_placeholder.metric("Total Token Count", state["prev_total_tokens"])
+            growth_speed_placeholder.metric("Total Growth Speed", state["total_growth_speed"])
         time.sleep(1)
 
 Thread(target=update_ui, daemon=True).start()