Refactored app.py

app.py

@@ -48,7 +48,7 @@ from src.tools.collections import update_collections
 from src.tools.plots import create_metric_plot_obj, create_plot_df, create_scores_df
 
 # Start ephemeral Spaces on PRs (see config in README.md)
-# enable_space_ci()
+enable_space_ci()
 
 
 def restart_space():
@@ -115,10 +115,14 @@ def init_space(full_init: bool = True):
     return leaderboard_df, original_df, plot_df, finished_eval_queue_df, running_eval_queue_df, pending_eval_queue_df
 
 
-skip_init = os.getenv("SKIP_INIT", "False") == "True"
+# Convert the environment variable "LEADERBOARD_FULL_INIT" to a boolean value, defaulting to True if the variable is not set.
+# This controls whether a full initialization should be performed.
+do_full_init = os.getenv("LEADERBOARD_FULL_INIT", "True") == "True"
 
+# Calls the init_space function with the `full_init` parameter determined by the `do_full_init` variable.
+# This initializes various DataFrames used throughout the application, with the level of initialization detail controlled by the `do_full_init` flag.
 leaderboard_df, original_df, plot_df, finished_eval_queue_df, running_eval_queue_df, pending_eval_queue_df = (
-    init_space(full_init=(not skip_init))
+    init_space(full_init=do_full_init)
 )
 
 
@@ -152,7 +156,7 @@ def load_query(request: gr.Request):  # triggered only once at startup => read q
     )  # return one for the "search_bar", one for a hidden component that triggers a reload only if value has changed
 
 
-def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
+def search_model(df: pd.DataFrame, query: str) -> pd.DataFrame:
     return df[(df[AutoEvalColumn.dummy.name].str.contains(query, case=False, na=False))]
 
 
@@ -171,27 +175,26 @@ def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
 
 
 def filter_queries(query: str, df: pd.DataFrame):
-    """Added by Abishek + Updated by alozowski"""
     tmp_result_df = []
 
     # Empty query return the same df
     if query == "":
         return df
 
-    all_queries = [q.strip() for q in query.split(";")]
-    license_queries = []
+    # all_queries = [q.strip() for q in query.split(";")]
+    # license_queries = []
+    all_queries = [q.strip() for q in query.split(";") if q.strip() != ""]
+    model_queries = [q for q in all_queries if not q.startswith("licence")]
+    license_queries_raw = [q for q in all_queries if q.startswith("license")]
+    license_queries = [
+        q.replace("license:", "").strip() for q in license_queries_raw if q.replace("license:", "").strip() != ""
+    ]
 
     # Handling model name search
-    for _q in all_queries:
-        if _q.startswith("license:"):
-            # Skipping license query here
-            license_queries.append(_q)
-            continue
-
-        if _q != "":
-            tmp_df = search_table(df, _q)
-            if len(tmp_df) > 0:
-                tmp_result_df.append(tmp_df)
+    for query in model_queries:
+        tmp_df = search_model(df, query)
+        if len(tmp_df) > 0:
+            tmp_result_df.append(tmp_df)
 
     if not tmp_result_df and not license_queries:
         # Nothing is found, no license_queries -> return empty df
@@ -206,13 +209,12 @@ def filter_queries(query: str, df: pd.DataFrame):
     if not license_queries:
         return df
 
+    # Handling license search
     tmp_result_df = []
-    for _q in license_queries:
-        _q = _q.replace("license:", "").strip()
-        if _q != "":
-            tmp_df = search_license(df, _q)
-            if len(tmp_df) > 0:
-                tmp_result_df.append(tmp_df)
+    for query in license_queries:
+        tmp_df = search_license(df, query)
+        if len(tmp_df) > 0:
+            tmp_result_df.append(tmp_df)
 
     if not tmp_result_df:
         # Nothing is found, return empty df

pyproject.toml

@@ -16,7 +16,7 @@ line-length = 119
 name = "open-llm-leaderboard"
 version = "0.1.0"
 description = ""
-authors = ["Alina Lozovskaia <[email protected]>"]
+authors = []
 readme = "README.md"
 
 [tool.poetry.dependencies]