Update app.py

app.py

@@ -1,25 +1,19 @@
 import gradio as gr
-import os
 import random
 from pathlib import Path
+import time
+import threading
 
-####################################
-# Constants (static data version)
-####################################
-
-
-SPACE_ID = os.getenv('SPACE_ID')
-MAX_SAMPLE_TXT_LENGTH = 300
-MIN_SAMPLE_TXT_LENGTH = 10
+LAST_UPDATED = "Nov 22th 2024"
 
 ####################################
 # Datos estáticos del leaderboard
 ####################################
 leaderboard_data = [
-    {'name': 'StyleTTS 2', 'STOI': 0.998, 'PESQ': 3.921,             'WER': 0.162, 'UTMOS':3.47},
-    {'name': 'Matxa-TTS', 'STOI': 0.996, 'PESQ': 3.539,              'WER': 0.179, 'UTMOS':3.50},
-    {'name': 'Matxa-TTS-multiaccent', 'STOI': 0.996, 'PESQ': 3.415,  'WER': 0.242, 'UTMOS':2.98},
-    {'name': 'StableTTS', 'STOI': 0.997, 'PESQ': 3.643,              'WER': 0.164, 'UTMOS':2.62},
+    {'name': 'StyleTTS 2', 'STOI': 0.998, 'PESQ': 3.921, 'WER': 0.162, 'UTMOS': 3.47},
+    {'name': 'Matxa-TTS', 'STOI': 0.996, 'PESQ': 3.539, 'WER': 0.179, 'UTMOS': 3.50},
+    {'name': 'Matxa-TTS-multiaccent', 'STOI': 0.996, 'PESQ': 3.415, 'WER': 0.242, 'UTMOS': 2.98},
+    {'name': 'StableTTS', 'STOI': 0.997, 'PESQ': 3.643, 'WER': 0.164, 'UTMOS': 2.62},
 ]
 
 ####################################
@@ -32,16 +26,12 @@ def get_leaderboard():
     """
     return sorted(leaderboard_data, key=lambda x: x['PESQ'], reverse=True)
 
-
-
-def update_leaderboard(reveal):
+def update_leaderboard():
     """
-    Actualiza la tabla del leaderboard con base en si se deben mostrar
-    resultados preliminares o no.
+    Actualiza la tabla del leaderboard.
     """
     filtered_leaderboard = get_leaderboard()
     
-    
     # Ordenar por 'score' y asignar el rank dinámicamente
     sorted_leaderboard = sorted(filtered_leaderboard, key=lambda x: x['UTMOS'], reverse=True)
     
@@ -49,7 +39,14 @@ def update_leaderboard(reveal):
     for rank, model in enumerate(sorted_leaderboard):
         model['rank'] = rank + 1  # rank es la posición en la lista (1-indexed)
     
-    return [[model['rank'], model['name'], model['UTMOS'], model['WER'], model['STOI'], model['PESQ'], model['SI-SDR']] for model in sorted_leaderboard]
+    return [[model['rank'], model['name'], model['UTMOS'], model['WER'], model['STOI'], model['PESQ']] for model in sorted_leaderboard]
+
+# Función para actualizar la tabla periódicamente
+def auto_update_table(leaderboard_table):
+    while True:
+        # Actualizar cada 10 segundos
+        leaderboard_table.update(value=update_leaderboard())
+        time.sleep(10)  # Esperar 10 segundos antes de actualizar de nuevo
 
 ####################################
 # Interfaz con Gradio
@@ -62,14 +59,12 @@ theme = gr.themes.Base(
 with gr.Blocks(theme=theme) as demo:
     gr.Markdown("# 🏆 Leaderboard\nVote to help the community determine the best Catalan TTS models.\n")
     
-    reveal_checkbox = gr.Checkbox(label="Reveal preliminary results", value=False)
-    
     # Inicializa la tabla sin datos, solo con encabezados
-    leaderboard_table = gr.DataFrame(headers=["Rank", "Model", "UTMOS", "WER", "STOI", "PESQ", "SI-SDR"], 
-                                     datatype=["str", "str", "str", "str", "str"], value=[])
-
-    # Al cambiar el valor del checkbox, actualizamos la tabla
-    reveal_checkbox.change(fn=update_leaderboard, inputs=[reveal_checkbox], outputs=[leaderboard_table])
+    leaderboard_table = gr.DataFrame(headers=["Rank", "Model", "UTMOS", "WER", "STOI", "PESQ"], 
+                                     datatype=["str", "str", "str", "str", "str", "str"], value=[])
+    
+    # Iniciar un hilo para actualizar la tabla automáticamente
+    threading.Thread(target=auto_update_table, args=(leaderboard_table,), daemon=True).start()
 
 # Lanzar la aplicación
 demo.queue(api_open=False, default_concurrency_limit=40).launch(show_api=False)