Update app.py

app.py

@@ -74,18 +74,53 @@ def make_results_tab(model_info, results):
     return df 
 
 
+def calculate_column_widths(df):
+    column_widths = []
+    for column in df.columns:
+        header_length = len(str(column))
+        max_content_length = df[column].astype(str).map(len).max()
+        width = max(header_length * 10, max_content_length * 8) + 20
+        width = max(160, min(400, width))
+        column_widths.append(width)
+    return column_widths
+
+
 def show_results_tab(df):
     with gr.Row():
         model_name = gr.Textbox(value='Input the Model Name (fuzzy)', label='Model Name')
 
+
+    def filter_df(model_name):
+
+        df = generate_table(results)
+
+        df['flag'] = [model_size_flag(x, model_size) for x in df['Param (B)']]
+        df = df[df['flag']]
+        df.pop('flag')
+        if len(df):
+            df['flag'] = [model_type_flag(df.iloc[i], model_type) for i in range(len(df))]
+            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:
+            method_names = [x.split('</a>')[0].split('>')[-1].lower() for x in df['Method']]
+            flag = [model_name.lower() in name for name in method_names]
+            df['TEMP'] = flag
+            df = df[df['TEMP'] == True] 
+            df.pop('TEMP')
+
+        comp = gr.components.DataFrame(value=df[headers], datatype=[type_map[x] for x in headers])
+    return comp
+
     with gr.Column():
         table = gr.DataFrame(
             value=df,
             interactive=False,
             wrap=False,
-            #column_widths=calculate_column_widths(initial_data),
+            column_widths=calculate_column_widths(df),
     )
-    
+
 
 def create_interface():
     model_info, results = findfile()
@@ -98,7 +133,7 @@ def create_interface():
                 df = make_results_tab(model_info, results)
                 show_results_tab(df)
 
-            with gr.TabItem('Predictions', elem_id='notmain', id=0):
+            with gr.TabItem('Predictions', elem_id='notmain', id=1):
                 # dataset_tab(results, structs[i], dataset)
                 pass
 
@@ -112,4 +147,3 @@ if __name__ == '__main__':
     demo.queue()
     demo.launch(server_name='0.0.0.0')
 
-