Update app.py

app.py

@@ -19,7 +19,6 @@ state = {
 }
 
 state_lock = Lock()
-stop_threads = False
 
 @retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry=retry_if_exception_type(httpx.ReadTimeout))
 def get_token_count(client):
@@ -27,7 +26,7 @@ def get_token_count(client):
     return int(result)
 
 def monitor_growth():
-    while not stop_threads:
+    while True:
         try:
             curr_count1 = get_token_count(client1)
             curr_count2 = get_token_count(client2)
@@ -63,19 +62,21 @@ token_count_placeholder = st.empty()
 growth_speed_placeholder = st.empty()
 
 def update_ui():
-    while not stop_threads:
+    while True:
         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)
+        st.experimental_rerun()
 
+# Start the update UI thread
 update_thread = Thread(target=update_ui, daemon=True)
 update_thread.start()
 
+# Keep the app running
 try:
     while True:
         time.sleep(1)
 except KeyboardInterrupt:
-    stop_threads = True
     monitor_thread.join()
     update_thread.join()