Rename Document Understanding section to Document

app/backend/app_init_func.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import os
 
-from app.backend.constant import LEADERBOARD_ICON_MAP
+from app.backend.constant import LEADERBOARD_ICON_MAP, get_display_name
 
 LEADERBOARD_MAP = {}
 LI_CSS = []
@@ -34,7 +34,7 @@ def init_leaderboard():
             text-indent: 2rem;
         }}
             """)
-        page_name = leaderboard_section if name.lower() == leaderboard.lower() else name.capitalize()
+        page_name = leaderboard_section if name.lower() == leaderboard.lower() else get_display_name(name)
         page_sections.append(f"""
         [[pages]]
         path = "app/ui/pages/{name}.py"

app/backend/constant.py

@@ -67,6 +67,15 @@ class Similarity(Enum):
     EUCLIDEAN = "euclidean"
 
 
+DISPLAY_NAME_MAP = {
+    "document_understanding": "Document",
+}
+
+
+def get_display_name(group_name):
+    return DISPLAY_NAME_MAP.get(group_name, group_name.capitalize())
+
+
 LEADERBOARD_ICON_MAP = {
     "Text Leaderboard": "📚",
     "Multimodal Leaderboard": "🖼️",
@@ -81,7 +90,7 @@ LEADERBOARD_ICON_MAP = {
     "Finance": "💰",
     "French": "🇫🇷",
     "Japanese": "🇯🇵",
-    "Document_understanding": "📄",
+    "Document": "📄",
 }
 
 USERNAME = "embedding-benchmark"

app/backend/data_page.py

@@ -15,7 +15,7 @@ from utils.st_copy_to_clipboard import st_copy_to_clipboard
 from streamlit_theme import st_theme
 
 from app.backend.app_init_func import LEADERBOARD_MAP
-from app.backend.constant import LEADERBOARD_ICON_MAP, BASE_URL
+from app.backend.constant import LEADERBOARD_ICON_MAP, BASE_URL, get_display_name
 from app.backend.json_util import compress_msgpack, decompress_msgpack
 
 COLUMNS = ['model_name', 'vendor',
@@ -395,9 +395,10 @@ def main_page(group_name, grid_state):
 
     #  logo
     # st.markdown('<img src="https://www.voyageai.com/logo.svg" class="center-image" width="200">', unsafe_allow_html=True)
-    title = f'<h2 class="center-text">{LEADERBOARD_ICON_MAP.get(group_name.capitalize(), "")} {group_name.capitalize()}</h2>'
+    display_name = get_display_name(group_name)
+    title = f'<h2 class="center-text">{LEADERBOARD_ICON_MAP.get(display_name, "")} {display_name}</h2>'
     if is_section(group_name):
-        title = f'<h2 class="center-text">{LEADERBOARD_ICON_MAP.get(group_name.capitalize() + " Leaderboard", "")} {group_name.capitalize() + " Leaderboard"}</h2>'
+        title = f'<h2 class="center-text">{LEADERBOARD_ICON_MAP.get(display_name + " Leaderboard", "")} {display_name + " Leaderboard"}</h2>'
     # title
     st.markdown(title, unsafe_allow_html=True)
 
@@ -406,7 +407,7 @@ def main_page(group_name, grid_state):
     df = data_engine.jsons_to_df().copy()
 
     csv = convert_df_to_csv(df)
-    file_name = f"{group_name.capitalize()} Leaderboard" if is_section(group_name) else group_name.capitalize()
+    file_name = f"{display_name} Leaderboard" if is_section(group_name) else display_name
 
     grid = table_area(group_name, grid_state, data_engine, df)
 

app/ui/pages_sections.toml

@@ -0,0 +1,66 @@
+
+        [[pages]]
+        path = "app/ui/pages/text.py"
+        name = "Text Leaderboard"
+        icon = "📚"
+        
+
+        [[pages]]
+        path = "app/ui/pages/legal.py"
+        name = "Legal"
+        icon = "📜"
+        
+
+        [[pages]]
+        path = "app/ui/pages/english.py"
+        name = "English"
+        icon = "🇬🇧"
+        
+
+        [[pages]]
+        path = "app/ui/pages/code.py"
+        name = "Code"
+        icon = "💻"
+        
+
+        [[pages]]
+        path = "app/ui/pages/healthcare.py"
+        name = "Healthcare"
+        icon = "🏥"
+        
+
+        [[pages]]
+        path = "app/ui/pages/finance.py"
+        name = "Finance"
+        icon = "💰"
+        
+
+        [[pages]]
+        path = "app/ui/pages/french.py"
+        name = "French"
+        icon = "🇫🇷"
+        
+
+        [[pages]]
+        path = "app/ui/pages/german.py"
+        name = "German"
+        icon = "🇩🇪"
+        
+
+        [[pages]]
+        path = "app/ui/pages/japanese.py"
+        name = "Japanese"
+        icon = "🇯🇵"
+        
+
+        [[pages]]
+        path = "app/ui/pages/multimodal.py"
+        name = "Multimodal Leaderboard"
+        icon = "🖼️"
+        
+
+        [[pages]]
+        path = "app/ui/pages/document_understanding.py"
+        name = "Document"
+        icon = "📄"
+