Update app.py

app.py

@@ -48,13 +48,13 @@ 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"])
 
-    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'])}"
+    return elo_scores["DALL-E 3"], elo_scores["Midjourney 6.1"], elo_scores["Flux"]
 
 # Interfaz Gradio
 def vote(first, second, third):
-    updated_scores = update_elo(first, second, third)
+    dalle3_score, midjourney_score, flux_score = update_elo(first, second, third)
     dalle3_image, midjourney_image, flux_image = load_images()
-    return dalle3_image, midjourney_image, flux_image, updated_scores
+    return dalle3_image, midjourney_image, flux_image, f"DALL-E 3: {round(dalle3_score)}", f"Midjourney 6.1: {round(midjourney_score)}", f"Flux: {round(flux_score)}"
 
 # Cargar imágenes al inicio
 dalle3_image, midjourney_image, flux_image = load_images()
@@ -72,10 +72,15 @@ with gr.Blocks() as demo:
         second_choice = gr.Dropdown(["DALL-E 3", "Midjourney 6.1", "Flux"], label="Segunda Favorita")
         third_choice = gr.Dropdown(["DALL-E 3", "Midjourney 6.1", "Flux"], label="Tercera Favorita")
     
-    elo_scores_text = gr.Textbox(label="Puntuación Elo")
+    with gr.Row():
+        dalle3_score_text = gr.Textbox(value=f"DALL-E 3: {elo_scores['DALL-E 3']}", label="Puntuación DALL-E 3", interactive=False)
+        midjourney_score_text = gr.Textbox(value=f"Midjourney 6.1: {elo_scores['Midjourney 6.1']}", label="Puntuación Midjourney 6.1", interactive=False)
+        flux_score_text = gr.Textbox(value=f"Flux: {elo_scores['Flux']}", label="Puntuación Flux", interactive=False)
 
     vote_button = gr.Button("Votar y Actualizar")
 
-    vote_button.click(vote, inputs=[first_choice, second_choice, third_choice], outputs=[dalle3_img, midjourney_img, flux_img, elo_scores_text])
+    vote_button.click(vote, inputs=[first_choice, second_choice, third_choice], 
+                      outputs=[dalle3_img, midjourney_img, flux_img, 
+                               dalle3_score_text, midjourney_score_text, flux_score_text])
 
 demo.launch()