Update app.py - leaderboard

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import pandas as pd
 import os
 import shutil
+import json
 
 # Function to load leaderboard data from a CSV file
 def load_leaderboard_data(csv_file_path):
@@ -13,43 +14,43 @@ def load_leaderboard_data(csv_file_path):
         return pd.DataFrame()  # Return an empty DataFrame in case of error
 
 # Function to process uploaded JSON file
-def process_json_file(json_file):
+def process_json_file(file):
+    if file is None:
+        return None, "Please upload a JSON file."
     try:
-        # Read the JSON file
-        data = pd.read_json(json_file.name)
-        # Here you can process the data as needed
-        # For demonstration, we'll just return the data as a dictionary
-        return data.to_dict()
+        with open(file.name, 'r') as f:
+            data = json.load(f)
+        return data, None
     except Exception as e:
-        return {"error": str(e)}
-
-# Load the leaderboard data
-leaderboard1 = load_leaderboard_data("leaderboard_swe.csv")
-leaderboard2 = load_leaderboard_data("leaderboard_gaia.csv")
+        return None, f"Error reading JSON file: {str(e)}"
 
 # Function to save the uploaded JSON file
-def save_json_file(file_path):
-    if not file_path:
+def save_json_file(file):
+    if file is None:
         return "No file uploaded."
 
     # Define the directory to save uploaded files
     save_dir = "uploaded_jsons"
     os.makedirs(save_dir, exist_ok=True)
 
-    # Extract the original filename
-    original_filename = os.path.basename(file_path)
+    # Get the original filename
+    original_filename = os.path.basename(file.name)
 
     # Define the path to save the file
     save_path = os.path.join(save_dir, original_filename)
 
-    # Move the uploaded file to the save directory
-    shutil.move(file_path, save_path)
+    # Copy the uploaded file to the save directory
+    shutil.copy2(file.name, save_path)
 
     return f"File saved to {save_path}"
 
+# Load the leaderboard data
+leaderboard1 = load_leaderboard_data("leaderboard_swe.csv")
+leaderboard2 = load_leaderboard_data("leaderboard_gaia.csv")
+
 # Create the Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# 🥇 Leaderboards")
+    gr.Markdown("# 🥇 TRAIL: Trace Reasoning and Agentic Issue Localization Leaderboard")
     with gr.Row():
         with gr.Column():
             gr.Markdown("## TRAIL-SWE Leaderboard")
@@ -58,21 +59,31 @@ with gr.Blocks() as demo:
             gr.Markdown("## TRAIL-GAIA Leaderboard")
             gr.Dataframe(leaderboard2)
     
-    """
-    gr.Markdown("# Submit Here")
-    with gr.Row():
-        json_input = gr.File(label="Upload JSON File", type="filepath")
-        json_output = gr.JSON(label="Processed Output")
-    submit_button = gr.Button("Submit")
-    submit_button.click(process_json_file, inputs=json_input, outputs=json_output)
-    """
     with gr.Blocks() as submit_page:
         gr.Markdown("## Submit Your JSON File Here")
-        file_input = gr.File(label="Upload JSON File", type="filepath", file_types=['.json'])
+        file_input = gr.File(label="Upload JSON File", file_types=['.json'])
+        json_preview = gr.JSON(label="JSON Preview")
         submit_button = gr.Button("Submit", interactive=True)
-        output = gr.Textbox("") # Successfully submitted! Thank you for your contribution!
-        submit_button.click(fn=save_json_file, inputs=file_input, outputs=output)
-
+        output = gr.Textbox(label="Status")
+        
+        def handle_submission(file):
+            if file is None:
+                return None, "Please upload a JSON file."
+            try:
+                # Process and preview the JSON
+                with open(file.name, 'r') as f:
+                    data = json.load(f)
+                # Save the file
+                save_result = save_json_file(file)
+                return data, save_result
+            except Exception as e:
+                return None, f"Error: {str(e)}"
+        
+        submit_button.click(
+            fn=handle_submission,
+            inputs=[file_input],
+            outputs=[json_preview, output]
+        )
 
 if __name__ == "__main__":
     demo.launch()