Update app.py

app.py

@@ -18,6 +18,7 @@ import sys
 from internal_stats import get_fun_stats
 import threading
 import time
+from collections import Counter
 
 
 # Initialize logging for errors only
@@ -53,13 +54,29 @@ def call_ollama_api(model, prompt):
         return f"Error: Unable to get response from the model."
 
 # Generate responses using two randomly selected models
+def get_battle_counts():
+    leaderboard = get_current_leaderboard()
+    battle_counts = Counter()
+    for model, data in leaderboard.items():
+        battle_counts[model] = data['wins'] + data['losses']
+    return battle_counts
+
 def generate_responses(prompt):
     available_models = get_available_models()
     if len(available_models) < 2:
         return "Error: Not enough models available", "Error: Not enough models available", None, None
     
-    selected_models = random.sample(available_models, 2)
-    model_a, model_b = selected_models
+    battle_counts = get_battle_counts()
+    
+    # Sort models by battle count (ascending)
+    sorted_models = sorted(available_models, key=lambda m: battle_counts.get(m, 0))
+    
+    # Select the first model (least battles)
+    model_a = sorted_models[0]
+    
+    # For the second model, use weighted random selection
+    weights = [1 / (battle_counts.get(m, 1) + 1) for m in sorted_models[1:]]
+    model_b = random.choices(sorted_models[1:], weights=weights, k=1)[0]
     
     model_a_response = call_ollama_api(model_a, prompt)
     model_b_response = call_ollama_api(model_b, prompt)
@@ -213,7 +230,6 @@ def get_leaderboard_chart():
     )
 
     chart_data = fig.to_json()
-    print(f"Chart size: {sys.getsizeof(chart_data)} bytes")
     return fig
 
 def new_battle():