add model_name box

app.py

@@ -49,6 +49,11 @@ with gr.Blocks(title="Math Leaderboard", head=head_style) as demo:
 
             headers = ['Rank'] + check_box['essential'] + checkbox_group.value
             with gr.Row():
+                model_name = gr.Textbox(
+                    value='Input the Model Name (fuzzy)', 
+                    label='Model Name', 
+                    interactive=True,
+                    visible=True)
                 model_size = gr.CheckboxGroup(
                     choices=MODEL_SIZE,
                     value=MODEL_SIZE,
@@ -61,6 +66,7 @@ with gr.Blocks(title="Math Leaderboard", head=head_style) as demo:
                     label='Model Type',
                     interactive=True
                 )
+
             data_component = gr.components.DataFrame(
                 value=table[headers],
                 type='pandas',
@@ -68,7 +74,7 @@ with gr.Blocks(title="Math Leaderboard", head=head_style) as demo:
                 interactive=False,
                 visible=True)
 
-            def filter_df(fields, model_size, model_type):
+            def filter_df(fields, model_name, model_size, model_type):
                 results = load_results()['results']
                 headers = ['Rank'] + check_box['essential'] + fields
 
@@ -82,6 +88,15 @@ with gr.Blocks(title="Math Leaderboard", head=head_style) as demo:
                     df = df[df['flag']]
                     df.pop('flag')
                 df['Rank'] = list(range(1, len(df) + 1))
+                default_val = 'Input the Model Name (fuzzy)'
+                if model_name != default_val:
+                    print(model_name)
+                    model_name = model_name.lower()
+                    method_names = [x.split('</a>')[0].split('>')[-1].lower() for x in df['Method']]
+                    flag = [model_name in name for name in method_names]
+                    df['TEMP_FLAG'] = flag
+                    df = df[df['TEMP_FLAG'] == True] 
+                    df.pop('TEMP_FLAG')
 
                 comp = gr.components.DataFrame(
                     value=df[headers],
@@ -92,7 +107,8 @@ with gr.Blocks(title="Math Leaderboard", head=head_style) as demo:
                 return comp
 
             for cbox in [checkbox_group, model_size, model_type]:
-                cbox.change(fn=filter_df, inputs=[checkbox_group, model_size, model_type], outputs=data_component)
+                cbox.change(fn=filter_df, inputs=[checkbox_group, model_name, model_size, model_type], outputs=data_component)
+            model_name.submit(fn=filter_df, inputs=[checkbox_group, model_name, model_size, model_type], outputs=data_component)
 
         for i, dataset in enumerate(DATASETS):
             tab_name_map = {
@@ -118,6 +134,11 @@ with gr.Blocks(title="Math Leaderboard", head=head_style) as demo:
                 s.table['Rank'] = list(range(1, len(s.table) + 1))
 
                 with gr.Row():
+                    s.model_name = gr.Textbox(
+                        value='Input the Model Name (fuzzy)', 
+                        label='Model Name', 
+                        interactive=True,
+                        visible=True)
                     s.model_size = gr.CheckboxGroup(
                         choices=MODEL_SIZE,
                         value=MODEL_SIZE,
@@ -138,7 +159,7 @@ with gr.Blocks(title="Math Leaderboard", head=head_style) as demo:
                     visible=True)
                 s.dataset = gr.Textbox(value=dataset, label=dataset, visible=False)
 
-                def filter_df_l2(dataset_name, fields, model_size, model_type):
+                def filter_df_l2(dataset_name, fields, model_name, model_size, model_type):
                     results = load_results()['results']
                     s = structs[DATASETS.index(dataset_name)]
                     headers = ['Rank'] + s.check_box['essential'] + fields
@@ -151,6 +172,15 @@ with gr.Blocks(title="Math Leaderboard", head=head_style) as demo:
                         df = df[df['flag']]
                         df.pop('flag')
                     df['Rank'] = list(range(1, len(df) + 1))
+                    default_val = 'Input the Model Name (fuzzy)'
+                    if model_name != default_val:
+                        print(model_name)
+                        model_name = model_name.lower()
+                        method_names = [x.split('</a>')[0].split('>')[-1].lower() for x in df['Method']]
+                        flag = [model_name in name for name in method_names]
+                        df['TEMP_FLAG'] = flag
+                        df = df[df['TEMP_FLAG'] == True] 
+                        df.pop('TEMP_FLAG')
 
                     comp = gr.components.DataFrame(
                         value=df[headers],
@@ -163,8 +193,12 @@ with gr.Blocks(title="Math Leaderboard", head=head_style) as demo:
                 for cbox in [s.checkbox_group, s.model_size, s.model_type]:
                     cbox.change(
                         fn=filter_df_l2,
-                        inputs=[s.dataset, s.checkbox_group, s.model_size, s.model_type],
+                        inputs=[s.dataset, s.checkbox_group, s.model_name, s.model_size, s.model_type],
                         outputs=s.data_component)
+                s.model_name.submit(
+                    fn=filter_df_l2, 
+                    inputs=[s.dataset, s.checkbox_group, s.model_name, s.model_size, s.model_type],
+                    outputs=s.data_component)
 
     with gr.Row():
         with gr.Accordion('Citation', open=False):