Refactor search_leaderboard to use retrieval_df and update UI tab labels for clarity

app.py

@@ -40,15 +40,15 @@ CITATION_BUTTON_TEXT = """
 }
 """
 
-df = None
+retrieval_df = None
 original_columns_order = None
 
 def search_leaderboard(model_name, columns_to_show):
     if len(model_name.strip()) == 0:
-        return df.loc[:, columns_to_show]
+        return retrieval_df.loc[:, columns_to_show]
     
     threshold = 95  # You can adjust this value to make the search more or less strict
-    filtered_df = df.copy()
+    filtered_df = retrieval_df.copy()
 
     def calculate_similarity(row):
         similarity = fuzz.partial_ratio(model_name.lower(), row["Model"].lower())
@@ -70,20 +70,20 @@ def search_leaderboard(model_name, columns_to_show):
 
 
 def main():
-    global df, original_columns_order
-    df = load_retrieval_results()
-    df[["Model"]] = df[["Model"]].map(lambda x: f'<a href="https://huggingface.co/{x}" target="_blank">{x}</a>')
-    df.drop(columns=["Revision", "Precision", "Task"], inplace=True)
-    df.sort_values("Web Search Dataset (Overall Score)", ascending=False, inplace=True)
+    global retrieval_df, original_columns_order
+    retrieval_df = load_retrieval_results()
+    retrieval_df[["Model"]] = retrieval_df[["Model"]].map(lambda x: f'<a href="https://huggingface.co/{x}" target="_blank">{x}</a>')
+    retrieval_df.drop(columns=["Revision", "Precision", "Task"], inplace=True)
+    retrieval_df.sort_values("Web Search Dataset (Overall Score)", ascending=False, inplace=True)
 
     columns_to_show = ["Model", "Web Search Dataset (Overall Score)", "Model Size (in Millions)", "Embedding Dimension", "Max Tokens", "Num Likes"]
     with gr.Blocks() as demo:
         gr.HTML(HEADER)
         
         with gr.Tabs():
-            with gr.Tab("Retrieval"):
+            with gr.Tab("🕵️‍♂️ Retrieval"):
                 with gr.Tabs():
-                    with gr.Tab("Leaderboard"):
+                    with gr.Tab("👑 Leaderboard"):
                         with gr.Row():
                             search_box_retrieval = gr.Textbox(
                                 placeholder="Search for models...", 
@@ -92,13 +92,13 @@ def main():
                             )
                             columns_to_show_input = gr.CheckboxGroup(
                                 label="Columns to Show",
-                                choices=df.columns.tolist(),
+                                choices=retrieval_df.columns.tolist(),
                                 value=columns_to_show,
                                 scale=4
                             )
 
                         retrieval_leaderboard = gr.Dataframe(
-                            value=df[columns_to_show],
+                            value=retrieval_df[columns_to_show],
                             datatype="markdown",
                             wrap=True,
                             show_fullscreen_button=True,
@@ -112,17 +112,20 @@ def main():
                             outputs=retrieval_leaderboard
                         )
                         columns_to_show_input.select(
-                            lambda columns: df.loc[:, columns],
+                            lambda columns: retrieval_df.loc[:, columns],
                             inputs=columns_to_show_input,
                             outputs=retrieval_leaderboard
                         )
                         
-                    with gr.Tab("Submit Retriever"):
+                    with gr.Tab("🏵️ Submit Retriever"):
                         submit_gradio_module("Retriever")
 
-            with gr.Tab("Reranking"):
+                    with gr.Tab("ℹ️ About"):
+                        gr.Markdown(ABOUT_SECTION)
+
+            with gr.Tab("📊 Reranking"):
                 with gr.Tabs():
-                    with gr.Tab("Leaderboard"):
+                    with gr.Tab("👑 Leaderboard"):
                         search_box_reranker = gr.Textbox(
                                 placeholder="Search for models...", 
                                 label="Search", 
@@ -130,7 +133,7 @@ def main():
                             )
                         
                         reranker_leaderboard = gr.Dataframe(
-                            value=df[columns_to_show],
+                            value=retrieval_df[columns_to_show],
                             datatype="markdown",
                             wrap=True,
                             show_fullscreen_button=True,
@@ -143,11 +146,13 @@ def main():
                             outputs=reranker_leaderboard
                         )
 
-                    with gr.Tab("Submit Reranker"):
+                    with gr.Tab("🏵️ Submit Reranker"):
                         submit_gradio_module("Reranker")
 
+                    with gr.Tab("ℹ️ About"):
+                        gr.Markdown(ABOUT_SECTION)
                     
-            # with gr.Tab("LLM Context Answering"):
+            # with gr.Tab("🧠 LLM Context Answering"):
             #     with gr.Tabs():
             #         with gr.Tab("Leaderboard"):
             #             pass