Update app.py

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 from PIL import Image
-from elo import Elo
 
-# Inicializa el sistema de puntuación Elo
-elo_system = Elo(k=32)
+# Función para calcular la actualización de Elo
+def elo_update(winner_elo, loser_elo, k=32):
+    expected_score_winner = 1 / (1 + 10 ** ((loser_elo - winner_elo) / 400))
+    new_winner_elo = winner_elo + k * (1 - expected_score_winner)
+    new_loser_elo = loser_elo + k * (0 - (1 - expected_score_winner))
+    return new_winner_elo, new_loser_elo
 
 # Inicializa puntuaciones Elo
 elo_scores = {
@@ -24,17 +27,29 @@ def load_images():
 def update_elo(first, second, third):
     global elo_scores
 
+    # Primera elección
     if first == "DALL-E 3":
-        elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"] = elo_system.rate_match(elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"])
-        elo_scores["DALL-E 3"], elo_scores["Flux"] = elo_system.rate_match(elo_scores["DALL-E 3"], elo_scores["Flux"])
+        elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"] = elo_update(elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"])
+        elo_scores["DALL-E 3"], elo_scores["Flux"] = elo_update(elo_scores["DALL-E 3"], elo_scores["Flux"])
     elif first == "Midjourney 6.1":
-        elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"] = elo_system.rate_match(elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"])
-        elo_scores["Midjourney 6.1"], elo_scores["Flux"] = elo_system.rate_match(elo_scores["Midjourney 6.1"], elo_scores["Flux"])
+        elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"] = elo_update(elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"])
+        elo_scores["Midjourney 6.1"], elo_scores["Flux"] = elo_update(elo_scores["Midjourney 6.1"], elo_scores["Flux"])
     elif first == "Flux":
-        elo_scores["Flux"], elo_scores["DALL-E 3"] = elo_system.rate_match(elo_scores["Flux"], elo_scores["DALL-E 3"])
-        elo_scores["Flux"], elo_scores["Midjourney 6.1"] = elo_system.rate_match(elo_scores["Flux"], elo_scores["Midjourney 6.1"])
+        elo_scores["Flux"], elo_scores["DALL-E 3"] = elo_update(elo_scores["Flux"], elo_scores["DALL-E 3"])
+        elo_scores["Flux"], elo_scores["Midjourney 6.1"] = elo_update(elo_scores["Flux"], elo_scores["Midjourney 6.1"])
 
-    return f"DALL-E 3: {elo_scores['DALL-E 3']}, Midjourney 6.1: {elo_scores['Midjourney 6.1']}, Flux: {elo_scores['Flux']}"
+    # Segunda elección
+    if second == "DALL-E 3":
+        elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"] = elo_update(elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"])
+        elo_scores["DALL-E 3"], elo_scores["Flux"] = elo_update(elo_scores["DALL-E 3"], elo_scores["Flux"])
+    elif second == "Midjourney 6.1":
+        elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"] = elo_update(elo_scores["Midjourney 6.1"], elo_scores["DALL-E 3"])
+        elo_scores["Midjourney 6.1"], elo_scores["Flux"] = elo_update(elo_scores["Midjourney 6.1"], elo_scores["Flux"])
+    elif second == "Flux":
+        elo_scores["Flux"], elo_scores["DALL-E 3"] = elo_update(elo_scores["Flux"], elo_scores["DALL-E 3"])
+        elo_scores["Flux"], elo_scores["Midjourney 6.1"] = elo_update(elo_scores["Flux"], elo_scores["Midjourney 6.1"])
+
+    return f"DALL-E 3: {round(elo_scores['DALL-E 3'])}, Midjourney 6.1: {round(elo_scores['Midjourney 6.1'])}, Flux: {round(elo_scores['Flux'])}"
 
 # Interfaz Gradio
 def vote(first, second, third):