[Update]Add filter models

app.py

@@ -104,6 +104,7 @@ def update_table(
     # filtered_df = filtered_df[filtered_df['Open?'].isin([map_open[o] for o in open_query])]
     filtered_df=select_columns(filtered_df,model1_column)
     filtered_df = filter_queries(query, filtered_df)
+    filtered_df = filter_queries_model(open_query, filtered_df)
     # filtered_df = filtered_df[[map_columns[k] for k in columns]]
     # deduplication
     # df = df.drop_duplicates(subset=["Model"])
@@ -116,7 +117,7 @@ def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
     return df[(df['Unlearned_Methods'].str.contains(query, case=False))]
 
 
-def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame:
+def filter_queries(query: list, filtered_df: pd.DataFrame) -> pd.DataFrame:
     final_df = []
     if query != "":
         queries = [q.strip() for q in query.split(";")]
@@ -131,6 +132,25 @@ def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame:
 
     return filtered_df
 
+def search_table_model(df: pd.DataFrame, query: str) -> pd.DataFrame:
+    return df[(df['Models'].str.contains(query, case=False))]
+
+
+def filter_queries_model(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame:
+    final_df = []
+    # if query != "":
+        # queries = [q.strip() for q in query.split(";")]
+    for _q in query:
+        _q = _q.strip()
+        if _q != "":
+            temp_filtered_df = search_table_model(filtered_df, _q)
+            if len(temp_filtered_df) > 0:
+                final_df.append(temp_filtered_df)
+    if len(final_df) > 0:
+        filtered_df = pd.concat(final_df)
+
+    return filtered_df
+
 def select_columns(df: pd.DataFrame, columns_1: list) -> pd.DataFrame:
     always_here_cols = ['Unlearned_Methods','Source', 'Diffusion_Models']