ElLokoAkrata commited on
Commit
df7b421
1 Parent(s): 0fe8471

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -5
app.py CHANGED
@@ -1,19 +1,29 @@
1
  import gradio as gr
2
  from PIL import Image
3
  import json
 
4
 
5
- # Inicializa puntuaciones Elo y las guarda en un archivo
 
 
 
 
6
  def load_elo_scores():
7
  try:
8
- with open("elo_scores.json", "r") as f:
9
  return json.load(f)
10
  except FileNotFoundError:
11
- # Si el archivo no existe, inicializamos las puntuaciones en 1000
12
  return {"DALL-E 3": 1000, "Midjourney 6.1": 1000, "Flux": 1000}
13
 
 
14
  def save_elo_scores(elo_scores):
15
- with open("elo_scores.json", "w") as f:
16
- json.dump(elo_scores, f)
 
 
 
 
17
 
18
  # Función para calcular la actualización de Elo
19
  def elo_update(winner_elo, loser_elo, k=32):
@@ -55,7 +65,9 @@ def update_elo(first, second, third):
55
  elo_scores["Flux"], elo_scores["DALL-E 3"] = elo_update(elo_scores["Flux"], elo_scores["DALL-E 3"])
56
  elo_scores["Flux"], elo_scores["Midjourney 6.1"] = elo_update(elo_scores["Flux"], elo_scores["Midjourney 6.1"])
57
 
 
58
  save_elo_scores(elo_scores)
 
59
  return elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"], elo_scores["Flux"]
60
 
61
  # Cargar las puntuaciones Elo
 
1
  import gradio as gr
2
  from PIL import Image
3
  import json
4
+ import os
5
 
6
+ # Obtener la ruta absoluta para guardar el archivo en el directorio actual
7
+ def get_file_path(filename):
8
+ return os.path.join(os.getcwd(), filename)
9
+
10
+ # Cargar puntuaciones Elo desde un archivo JSON
11
  def load_elo_scores():
12
  try:
13
+ with open(get_file_path("elo_scores.json"), "r") as f:
14
  return json.load(f)
15
  except FileNotFoundError:
16
+ # Si el archivo no existe, inicializamos las puntuaciones
17
  return {"DALL-E 3": 1000, "Midjourney 6.1": 1000, "Flux": 1000}
18
 
19
+ # Guardar puntuaciones Elo en un archivo JSON
20
  def save_elo_scores(elo_scores):
21
+ try:
22
+ with open(get_file_path("elo_scores.json"), "w") as f:
23
+ json.dump(elo_scores, f)
24
+ print("Puntuaciones guardadas correctamente en elo_scores.json")
25
+ except Exception as e:
26
+ print(f"Error al guardar el archivo: {e}")
27
 
28
  # Función para calcular la actualización de Elo
29
  def elo_update(winner_elo, loser_elo, k=32):
 
65
  elo_scores["Flux"], elo_scores["DALL-E 3"] = elo_update(elo_scores["Flux"], elo_scores["DALL-E 3"])
66
  elo_scores["Flux"], elo_scores["Midjourney 6.1"] = elo_update(elo_scores["Flux"], elo_scores["Midjourney 6.1"])
67
 
68
+ # Guardar las puntuaciones actualizadas en el archivo
69
  save_elo_scores(elo_scores)
70
+
71
  return elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"], elo_scores["Flux"]
72
 
73
  # Cargar las puntuaciones Elo