Update app.py

app.py

@@ -1,27 +1,30 @@
-# prompt: instead of using JSON_FILE in local drive, use the HuggingFace dataset "NLP-UNED/dipromats2024-t2_leaderboard-data"
 import gradio as gr
 import json
 import os
 import random
 import datasets #load_dataset, save_to_disk, load_from_disk
 
+from google.colab import userdata
+
+HF_TOKEN = userdata.get('HF-DIPROMATS2024-T2-LEADERBOARD-TOKEN')
+
 # Use the Hugging Face dataset
 DATASET_NAME = "NLP-UNED/dipromats2024-t2_leaderboard-data"
-HF_TOKEN = os.getenv('HF_TOKEN_UNEDNLP_WRITE')
+SPLIT = 'results'
 
 try:
     dataset = datasets.load_dataset(DATASET_NAME)
-    data = dataset['train']
 
 except Exception as e:
     print(f"Error loading dataset: {e}")
-    data = []
+    dataset = datasets.Dataset.from_dict({"email": [], "team_name": [], "run_id": [], "description": [], "lenient_f1": [], "strict_f1": [], "average_f1": []})
+    dataset = datasets.DatasetDict({SPLIT: dataset})
 
 # Función para convertir el dataset en tabla
 def data_to_table():
-    global data
+    global dataset
     table_data = []
-    for item in data:
+    for item in dataset[SPLIT]:
         table_data.append([item.get("team_name", ""), item.get("run_id", ""),
                            item.get("lenient_f1", ""), item.get("strict_f1", ""), item.get("average_f1", "")])
     return table_data
@@ -29,8 +32,8 @@ def data_to_table():
 
 # Función para subir los resultados al leaderboard
 def update_leaderboard(email, team_input, run_id, description, lenient_f1, strict_f1, average_f1):
-    global data
-    data.add_item({
+    global datataset
+    new_data = dataset[SPLIT].add_item({
         "email": email,
         "team_name": team_input,
         "run_id": run_id,
@@ -39,7 +42,8 @@ def update_leaderboard(email, team_input, run_id, description, lenient_f1, stric
         "strict_f1": strict_f1,
         "average_f1": average_f1
     })
-    data.push_to_hub(DATASET_NAME, token=HF_TOKEN)
+    dataset[SPLIT] = new_data
+    new_data.push_to_hub(DATASET_NAME, split=SPLIT, token=HF_TOKEN)
     return data_to_table(), gr.Tabs(selected=0), gr.Button(visible=True), gr.Button(visible=False), "", "", "", "", None, None, None, None
 
 
@@ -140,4 +144,4 @@ with gr.Blocks() as leaderboard:
     submit_button.click(update_leaderboard,
                         inputs=[email_input, team_input, run_id, description_input, lenient_f1, strict_f1, average_f1],
                         outputs=[leaderboard_table, tabs, evaluate_button, submit_button, team_input, run_id, description_input, email_input, file_input,lenient_f1, strict_f1, average_f1])
-leaderboard.launch(debug=True)
+leaderboard.launch()