Change the model search behavior

app.py

@@ -141,27 +141,16 @@ def filter_models(
     return filtered_df
 
 
-def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
-    return df[df[AutoEvalColumn.dummy.name].str.contains(query, case=False)]
-
-
-def filter_queries(query: str, filtered_df: pd.DataFrame) -> pd.DataFrame:
-    """Added by Abishek"""
-    if not query:
-        return filtered_df
-
-    final_df = []
-    queries = [q.strip() for q in query.split(";")]
-    for _q in queries:
-        _q = _q.strip()
-        if _q != "":
-            temp_filtered_df = search_table(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)
-        filtered_df = filtered_df.drop_duplicates(subset=AutoEvalColumn.row_id.name)
-    return filtered_df
+def search_model_by_name(df: pd.DataFrame, model_name: str) -> pd.DataFrame:
+    return df[df[AutoEvalColumn.dummy.name].str.contains(model_name, case=False)]
+
+
+def search_models_by_multiple_names(df: pd.DataFrame, search_text: str) -> pd.DataFrame:
+    if not search_text:
+        return df
+    model_names = [name.strip() for name in search_text.split(";")]
+    dfs = [search_model_by_name(df, name) for name in model_names if name]
+    return pd.concat(dfs).drop_duplicates(subset=AutoEvalColumn.row_id.name)
 
 
 def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
@@ -217,8 +206,8 @@ def update_table(
     )
     print(f"filtered_df shape after filter_models: {filtered_df.shape}")
 
-    filtered_df = filter_queries(query, filtered_df)
-    print(f"filtered_df shape after filter_queries: {filtered_df.shape}")
+    filtered_df = search_models_by_multiple_names(filtered_df, query)
+    print(f"filtered_df shape after search_models_by_multiple_names: {filtered_df.shape}")
 
     print(
         f"Filter applied: query={query}, columns={columns}, type_query={type_query}, precision_query={precision_query}"