added search bar + plot

app.py

@@ -6,6 +6,7 @@ from init import is_model_on_hub, upload_file, load_all_info_from_dataset_hub
 from utils_display import AutoEvalColumn, fields, make_clickable_model, styled_error, styled_message
 from datetime import datetime, timezone
 from utils_display import make_best_bold
+import plotly.graph_objects as go
 
 LAST_UPDATED = "Sep 11th 2024"
 
@@ -51,8 +52,10 @@ def request_model(model_text, chbcoco2017):
     # ... (keep the existing request_model function as is)
     pass
 
-def update_table(column_selection):
+def update_table(column_selection, search:str):
     original_df = pd.read_csv("data.csv")
+
+    original_df = original_df[original_df['model'].str.contains(search, case=False, na=False)]
     
     if column_selection == "All Columns":
         new_df = original_df
@@ -74,6 +77,43 @@ def update_table(column_selection):
 
     return new_df
 
+def generate_plot():
+    df = pd.read_csv("data.csv")
+    fig = go.Figure()
+
+    fig.add_trace(go.Scatter(
+        x=df['Average WER ⬇️ '],
+        y=df['RTFx ⬆️ '],
+        mode='markers',
+        text=df['model'],
+        hovertemplate=
+        '<b>%{text}</b><br>' +
+        'Average WER: %{x:.2f}<br>' +
+        'RTFx: %{y:.2f}<br>' +
+        '<extra></extra>',
+        marker=dict(
+            size=10,
+            # color=df['Average WER ⬇️ '],
+            colorscale='Viridis',
+            # colorbar=dict(title='Average WER'),
+            # showscale=True
+        )
+    ))
+
+    # Update the layout
+    fig.update_layout(
+        title='ASR Model Performance: Average WER vs RTFx',
+        xaxis_title='Average WER (lower is better)',
+        yaxis_title='RTFx (higher is better)',
+        #yaxis_type='log',
+        hovermode='closest'
+    )
+
+    # Show the plot
+    return fig
+
+
+
 with gr.Blocks() as demo:
     gr.HTML(BANNER, elem_id="banner")
     gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
@@ -94,8 +134,11 @@ with gr.Blocks() as demo:
                     label="Categories",
                     value="All Columns"
                     )
+                
+                search_bar = gr.Textbox(label="Search models", placeholder="Enter model name...")
 
-        column_radio.change(update_table, inputs=[column_radio], outputs=[leaderboard_table])
+        column_radio.change(update_table, inputs=[column_radio, search_bar], outputs=[leaderboard_table])
+        search_bar.submit(update_table, inputs=[column_radio, search_bar], outputs=[leaderboard_table])
 
         with gr.TabItem("📈 Metrics", elem_id="od-benchmark-tab-table", id=1):
             gr.Markdown(METRICS_TAB_TEXT, elem_classes="markdown-text")
@@ -114,6 +157,10 @@ with gr.Blocks() as demo:
                     btn_submitt.click(request_model, 
                                       [model_name_textbox, chb_coco2017], 
                                       mdw_submission_result)
+        
+        with gr.TabItem("📊 Plots", elem_id="od-benchmark-tab-table", id=3):
+            
+            plot = gr.Plot(generate_plot)
 
     gr.Markdown(f"Last updated on **{LAST_UPDATED}**", elem_classes="markdown-text")
     

requirements.txt

@@ -59,3 +59,4 @@ tzdata==2023.3
 urllib3==2.0.4
 xxhash==3.2.0
 yarl==1.9.2
+plotly==5.6.0

utils_display.py

@@ -43,6 +43,6 @@ def make_best_bold(column, column_name):
         best_value = column.max()
     else:
         best_value = column.min()
-    return column.apply(lambda x: f'#### <span style="color: red;">{float(x):.2f}</span>' if x == best_value else f"{x:.2f}")
+    return column.apply(lambda x: f'#### <span style="color: blue;">{float(x):.2f}</span>' if x == best_value else f"{x:.2f}")