Update app.py

app.py

@@ -2,7 +2,7 @@ import requests
 from bs4 import BeautifulSoup
 import pandas as pd
 import gradio as gr
-import io
+import plotly.express as px
 
 BASE_URL = "https://scale.com/leaderboard"
 
@@ -39,42 +39,39 @@ def scrape_leaderboard(leaderboard):
         data.append([rank, model, score, confidence])
     
     df = pd.DataFrame(data, columns=['Rank', 'Model', 'Score', '95% Confidence'])
+    df['Score'] = pd.to_numeric(df['Score'])  # Convert Score to numeric
     return df
 
+def create_chart(df):
+    fig = px.bar(df, x='Model', y='Score', title='Model Scores Comparison',
+                 labels={'Score': 'Overall Score', 'Model': 'Model Name'},
+                 color='Score', color_continuous_scale='viridis')
+    fig.update_layout(xaxis_tickangle=-45, xaxis_title=None)
+    return fig
+
 def update_leaderboard(leaderboard):
     try:
         df = scrape_leaderboard(leaderboard)
-        return df, df.to_html(index=False)
+        chart = create_chart(df)
+        return df, df.to_html(index=False), chart
     except Exception as e:
-        return None, f"An error occurred: {str(e)}"
-
-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
+        return None, f"An error occurred: {str(e)}", None
 
 # Create Gradio interface
 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")
+    dropdown = gr.Dropdown(choices=list(LEADERBOARDS.keys()), label="Select Leaderboard", value="Coding")
     
     table_output = gr.HTML()
-    df_state = gr.State()
+    chart_output = gr.Plot()
     
     def on_load():
-        df, html = update_leaderboard("Coding")
-        return df, html
+        df, html, chart = update_leaderboard("Coding")
+        return html, chart
     
-    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")])
+    dropdown.change(update_leaderboard, inputs=[dropdown], outputs=[table_output, table_output, chart_output])
     
-    iface.load(on_load, outputs=[df_state, table_output])
+    iface.load(on_load, outputs=[table_output, chart_output])
 
 # Launch the app
 iface.launch()