Update app.py

app.py

@@ -2,17 +2,15 @@ import requests
 from bs4 import BeautifulSoup
 import pandas as pd
 import gradio as gr
+import io
 
 BASE_URL = "https://scale.com/leaderboard"
 
 LEADERBOARDS = {
-    "Main Leaderboard": "",
-    "Adversarial Robustness": "/adversarial_robustness",
     "Coding": "/coding",
+    "Adversarial Robustness": "/adversarial_robustness",
     "Instruction Following": "/instruction_following",
-    "Math": "/math",
-    "Spanish": "/spanish",
-    "Methodology": "/methodology"
+    "Math": "/math"
 }
 
 def scrape_leaderboard(leaderboard):
@@ -46,18 +44,66 @@ def scrape_leaderboard(leaderboard):
 def update_leaderboard(leaderboard):
     try:
         df = scrape_leaderboard(leaderboard)
-        return df.to_html(index=False)
+        return df, create_interactive_table(df)
     except Exception as e:
-        return f"An error occurred: {str(e)}"
+        return None, f"An error occurred: {str(e)}"
+
+def create_interactive_table(df):
+    html = f"""
+    <script src="https://cdn.jsdelivr.net/npm/ag-grid-community/dist/ag-grid-community.min.js"></script>
+    <div id="myGrid" style="height: 500px; width: 100%;" class="ag-theme-alpine"></div>
+    <script>
+        var gridOptions = {{
+            columnDefs: [
+                {{field: "Rank", sortable: true, filter: true}},
+                {{field: "Model", sortable: true, filter: true}},
+                {{field: "Score", sortable: true, filter: true}},
+                {{field: "95% Confidence", sortable: true, filter: true}}
+            ],
+            rowData: {df.to_dict(orient='records')},
+            defaultColDef: {{
+                flex: 1,
+                minWidth: 100,
+                resizable: true,
+            }},
+            domLayout: 'autoHeight'
+        }};
+        
+        document.addEventListener('DOMContentLoaded', function() {{
+            var gridDiv = document.querySelector('#myGrid');
+            new agGrid.Grid(gridDiv, gridOptions);
+        }});
+    </script>
+    """
+    return html
+
+def export_to_excel(df):
+    if df is not None:
+        output = io.BytesIO()
+        with pd.ExcelWriter(output, engine='openpyxl') as writer:
+            df.to_excel(writer, index=False, sheet_name='Leaderboard')
+        output.seek(0)
+        return output
+    return None
 
 # Create Gradio interface
-iface = gr.Interface(
-    fn=update_leaderboard,
-    inputs=gr.Dropdown(choices=list(LEADERBOARDS.keys()), label="Select Leaderboard"),
-    outputs=gr.HTML(label="Leaderboard Data"),
-    title="Scale AI Leaderboard Viewer",
-    description="Select a leaderboard to view the latest data from Scale.com"
-)
+with gr.Blocks() as iface:
+    gr.Markdown("# Scale AI Leaderboard Viewer")
+    with gr.Row():
+        dropdown = gr.Dropdown(choices=list(LEADERBOARDS.keys()), label="Select Leaderboard", value="Coding")
+        export_button = gr.Button("Export to Excel")
+    
+    table_output = gr.HTML()
+    df_state = gr.State()
+    
+    def on_load():
+        df, html = update_leaderboard("Coding")
+        return df, html
+    
+    dropdown.change(update_leaderboard, inputs=[dropdown], outputs=[df_state, table_output])
+    export_button.click(export_to_excel, inputs=[df_state], outputs=[gr.File(label="Download Excel")])
+    
+    iface.load(on_load, outputs=[df_state, table_output])
 
 # Launch the app
 iface.launch()