Update app.py

app.py

@@ -155,17 +155,20 @@ async def chat_with_bots(user_input, state, character_name, character_descriptio
     )
     return bot1_response.replace("<|im_end|>",""), bot2_response.replace("<|im_end|>","")
 def update_ratings(state, winner_index, collection, category):
-    elo_ratings = get_user_elo_ratings(collection)
-    winner_adapter = state['last_bots'][winner_index]
-    loser_adapter = state['last_bots'][1 - winner_index]
-    
-    winner = next(entry['original_model'] for entry in chatbots_data if entry['adapter'] == winner_adapter)
-    loser = next(entry['original_model'] for entry in chatbots_data if entry['adapter'] == loser_adapter)
-    
-    elo_ratings = update_elo_ratings(elo_ratings, winner_adapter, loser_adapter, category)
-    update_elo_rating(collection, elo_ratings, winner_adapter, loser_adapter, category)
-    return [('Winner: ', winner), ('Loser: ', loser)]
-    
+    try:
+        elo_ratings = get_user_elo_ratings(collection)
+        winner_adapter = state['last_bots'][winner_index]
+        loser_adapter = state['last_bots'][1 - winner_index]
+        
+        winner = next(entry['original_model'] for entry in chatbots_data if entry['adapter'] == winner_adapter)
+        loser = next(entry['original_model'] for entry in chatbots_data if entry['adapter'] == loser_adapter)
+        
+        elo_ratings = update_elo_ratings(elo_ratings, winner_adapter, loser_adapter, category)
+        update_elo_rating(collection, elo_ratings, winner_adapter, loser_adapter, category)
+        return [('Winner: ', winner), ('Loser: ', loser)]
+    except Exception as e:
+        print(f"Error updating ratings: {str(e)}")
+        return [('Error', 'Failed to update ratings.'), ('Error', 'Failed to update ratings.')]
 def vote_up_model(state, chatbot, chatbot2, character_name, character_description, user_name):
     user_input = format_prompt(state, 0, character_name, character_description, user_name)
     collection = init_database()