Update app.py

app.py

@@ -153,9 +153,11 @@ def update_table(
     size_query: list,
     show_deleted: bool,
     query: str,
-    recent: bool
+    recent7: bool,
+    recent14: bool,
+    recent21: bool
 ):
-    filtered_df = filter_models(hidden_df, type_query, size_query, precision_query, show_deleted, recent)
+    filtered_df = filter_models(hidden_df, type_query, size_query, precision_query, show_deleted, recent7, recent14, recent21)
     filtered_df = filter_queries(query, filtered_df)
     df = select_columns(filtered_df, columns)
     return df
@@ -197,7 +199,7 @@ def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame:
 
 
 def filter_models(
-    df: pd.DataFrame, type_query: list, size_query: list, precision_query: list, show_deleted: bool, recent: bool
+    df: pd.DataFrame, type_query: list, size_query: list, precision_query: list, show_deleted: bool, recent7: bool, recent14: bool, recent21: bool
 ) -> pd.DataFrame:
     # Show all models
     if show_deleted:
@@ -206,8 +208,12 @@ def filter_models(
         filtered_df = df[df[AutoEvalColumn.still_on_hub.name] == True]
 
     # blz
-    if recent:
-        filtered_df = df[df[AutoEvalColumn.recent.name] == True]
+    if recent7:
+        filtered_df = df[df[AutoEvalColumn.recent7.name] == True]
+    if recent14:
+        filtered_df = df[df[AutoEvalColumn.recent14.name] == True]
+    if recent21:
+        filtered_df = df[df[AutoEvalColumn.recent21.name] == True]
     
     type_emoji = [t[0] for t in type_query]
     filtered_df = filtered_df.loc[df[AutoEvalColumn.model_type_symbol.name].isin(type_emoji)]
@@ -258,10 +264,15 @@ with demo:
                         )
 
                     with gr.Row():
-                        filter_recent = gr.Checkbox(
-                            value=False, label="Recent", interactive=True
+                        filter_recent7 = gr.Checkbox(
+                            value=False, label="Recent (7 days)", interactive=True
+                        )
+                        filter_recent14 = gr.Checkbox(
+                            value=False, label="Recent (14 days)", interactive=True
+                        )
+                        filter_recent21 = gr.Checkbox(
+                            value=False, label="Recent (21 days)", interactive=True
                         )
-                        
                 with gr.Column(min_width=320):
                     #with gr.Box(elem_id="box-filter"):
                     filter_columns_type = gr.CheckboxGroup(
@@ -317,11 +328,13 @@ with demo:
                     filter_columns_size,
                     deleted_models_visibility,
                     search_bar,
-                    filter_recent,
+                    filter_recent7,
+                    filter_recent14,
+                    filter_recent21,
                 ],
                 leaderboard_table,
             )
-            for selector in [shown_columns, filter_columns_type, filter_columns_precision, filter_columns_size, deleted_models_visibility, filter_recent]:
+            for selector in [shown_columns, filter_columns_type, filter_columns_precision, filter_columns_size, deleted_models_visibility, filter_recent7, filter_recent14, filter_recent21]:
                 selector.change(
                     update_table,
                     [
@@ -332,7 +345,9 @@ with demo:
                         filter_columns_size,
                         deleted_models_visibility,
                         search_bar,
-                        filter_recent,
+                        filter_recent7,
+                        filter_recent14,
+                        filter_recent21,
                     ],
                     leaderboard_table,
                     queue=True,