Update app.py

app.py

@@ -3,7 +3,6 @@ import requests
 import os
 import json
 import random
-import threading
 from elo import update_elo_ratings  # Custom function for ELO ratings
 
 # Load the chatbot URLs and their respective model names from a JSON file
@@ -14,27 +13,24 @@ with open('chatbot_urls.json', 'r') as file:
 user_data = threading.local()
 
 # Initialize or get user-specific ELO ratings
-def get_user_elo_ratings():
-    if not hasattr(user_data, 'elo_ratings'):
-        user_data.elo_ratings = read_elo_ratings()
-    return user_data.elo_ratings
+def get_user_elo_ratings(state):
+    if 'elo_ratings' not in state:
+        state['elo_ratings'] = read_elo_ratings()
+    return state['elo_ratings']
 
 # Read ELO ratings from file (thread-safe)
 def read_elo_ratings():
-    elo_ratings = {}
-    with threading.Lock():
-        try:
-            with open('elo_ratings.json', 'r') as file:
-                elo_ratings = json.load(file)
-        except FileNotFoundError:
-            elo_ratings = {model: 1200 for model in chatbots.keys()}
+    try:
+        with open('elo_ratings.json', 'r') as file:
+            elo_ratings = json.load(file)
+    except FileNotFoundError:
+        elo_ratings = {model: 1200 for model in chatbots.keys()}
     return elo_ratings
 
 # Write ELO ratings to file (thread-safe)
 def write_elo_ratings(elo_ratings):
-    with threading.Lock():
-        with open('elo_ratings.json', 'w') as file:
-            json.dump(elo_ratings, file, indent=4)
+    with open('elo_ratings.json', 'w') as file:
+        json.dump(elo_ratings, file, indent=4)
 
 def get_bot_response(url, prompt):
     payload = {
@@ -101,6 +97,7 @@ def user_ask(state, chatbot1, chatbot2, textbox, upvote_btn_a, upvote_btn_b):
     upvote_btn_a.interactive = True
     upvote_btn_b.interactive = True
     print(state, chatbot1, chatbot2, "", upvote_btn_a, upvote_btn_b)
+    state['elo_ratings'] = get_user_elo_ratings(state)
     return state, chatbot1, chatbot2, "", upvote_btn_a, upvote_btn_b
 
 # ... [Rest of your existing functions] ...