Update app.py

app.py

@@ -1,5 +1,19 @@
 import gradio as gr
 from PIL import Image
+import json
+
+# Inicializa puntuaciones Elo y las guarda en un archivo
+def load_elo_scores():
+    try:
+        with open("elo_scores.json", "r") as f:
+            return json.load(f)
+    except FileNotFoundError:
+        # Si el archivo no existe, inicializamos las puntuaciones en 1000
+        return {"DALL-E 3": 1000, "Midjourney 6.1": 1000, "Flux": 1000}
+
+def save_elo_scores(elo_scores):
+    with open("elo_scores.json", "w") as f:
+        json.dump(elo_scores, f)
 
 # Función para calcular la actualización de Elo
 def elo_update(winner_elo, loser_elo, k=32):
@@ -8,13 +22,6 @@ def elo_update(winner_elo, loser_elo, k=32):
     new_loser_elo = loser_elo + k * (0 - (1 - expected_score_winner))
     return new_winner_elo, new_loser_elo
 
-# Inicializa puntuaciones Elo
-elo_scores = {
-    "DALL-E 3": 1000,
-    "Midjourney 6.1": 1000,
-    "Flux": 1000,
-}
-
 # Cargar imágenes predefinidas
 def load_images():
     dalle3_image = Image.open("images/dalle3_image.png")
@@ -48,8 +55,12 @@ def update_elo(first, second, third):
         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"])
 
+    save_elo_scores(elo_scores)
     return elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"], elo_scores["Flux"]
 
+# Cargar las puntuaciones Elo
+elo_scores = load_elo_scores()
+
 # Interfaz Gradio
 def vote(first, second, third):
     dalle3_score, midjourney_score, flux_score = update_elo(first, second, third)
@@ -84,3 +95,4 @@ with gr.Blocks() as demo:
                                dalle3_score_text, midjourney_score_text, flux_score_text])
 
 demo.launch()
+