vectordb-bench 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

vectordb_bench/frontend/components/run_test/caseSelector.py

@@ -1,4 +1,6 @@
-from vectordb_bench.frontend.const import *
+from vectordb_bench.frontend.const.styles import *
+from vectordb_bench.backend.cases import CaseType
+from vectordb_bench.frontend.const.dbCaseConfigs import *
 
 
 def caseSelector(st, activedDbList):
@@ -12,34 +14,35 @@ def caseSelector(st, activedDbList):
         unsafe_allow_html=True,
     )
 
-    caseIsActived = {case["name"]: False for case in CASE_LIST}
-    allCaseConfigs = {db: {case["name"]: {} for case in CASE_LIST} for db in DB_LIST}
-    for case in CASE_LIST:
-        caseItemContainer = st.container()
-        caseIsActived[case["name"]] = caseItem(
-            caseItemContainer, allCaseConfigs, case, activedDbList
-        )
-        if case.get("divider"):
+    caseIsActived = {case: False for case in CASE_LIST}
+    allCaseConfigs = {db: {case: {} for case in CASE_LIST} for db in DB_LIST}
+    for caseOrDivider in CASE_LIST_WITH_DIVIDER:
+        if caseOrDivider == DIVIDER:
             caseItemContainer.markdown(
                 "<div style='border: 1px solid #cccccc60; margin-bottom: 24px;'></div>",
                 unsafe_allow_html=True,
             )
-    activedCaseList = [
-        case["name"] for case in CASE_LIST if caseIsActived[case["name"]]
-    ]
+        else:
+            case = caseOrDivider
+            caseItemContainer = st.container()
+            caseIsActived[case] = caseItem(
+                caseItemContainer, allCaseConfigs, case, activedDbList
+            )
+    activedCaseList = [case for case in CASE_LIST if caseIsActived[case]]
     return activedCaseList, allCaseConfigs
 
-def caseItem(st, allCaseConfigs, case, activedDbList):
-    selected = st.checkbox(case["name"].value)
+
+def caseItem(st, allCaseConfigs, case: CaseType, activedDbList):
+    selected = st.checkbox(case.case_name)
     st.markdown(
-        f"<div style='color: #1D2939; margin: -8px 0 20px {CHECKBOX_INDENT}px; font-size: 14px;'>{case['intro']}</div>",
+        f"<div style='color: #1D2939; margin: -8px 0 20px {CHECKBOX_INDENT}px; font-size: 14px;'>{case.case_description}</div>",
         unsafe_allow_html=True,
     )
 
     if selected:
         caseConfigSettingContainer = st.container()
         caseConfigSetting(
-            caseConfigSettingContainer, allCaseConfigs, case["name"], activedDbList
+            caseConfigSettingContainer, allCaseConfigs, case, activedDbList
         )
 
     return selected

vectordb_bench/frontend/components/run_test/dbConfigSetting.py

@@ -1,4 +1,5 @@
-from vectordb_bench.frontend.const import *
+from pydantic import ValidationError
+from vectordb_bench.frontend.const.styles import *
 from vectordb_bench.frontend.utils import inputIsPassword
 
 
@@ -10,12 +11,25 @@ def dbConfigSettings(st, activedDbList):
     expander = st.expander("Configurations for the selected databases", True)
 
     dbConfigs = {}
+    isAllValid = True
     for activeDb in activedDbList:
         dbConfigSettingItemContainer = expander.container()
         dbConfig = dbConfigSettingItem(dbConfigSettingItemContainer, activeDb)
-        dbConfigs[activeDb] = dbConfig
-        
-    return dbConfigs
+        try:
+            dbConfigs[activeDb] = activeDb.init_cls.config_cls()(**dbConfig)
+        except ValidationError as e:
+            isAllValid = False
+            errTexts = []
+            for err in e.raw_errors:
+                errLocs = err.loc_tuple()
+                errInfo = err.exc
+                errText = f"{', '.join(errLocs)} - {errInfo}"
+                errTexts.append(errText)
+
+            dbConfigSettingItemContainer.error(f"{'; '.join(errTexts)}")
+
+    return dbConfigs, isAllValid
+
 
 def dbConfigSettingItem(st, activeDb):
     st.markdown(
@@ -39,9 +53,8 @@ def dbConfigSettingItem(st, activeDb):
             value=value.get("default", ""),
             type="password" if inputIsPassword(key) else "default",
         )
-    return dbConfigClass(**dbConfig)
+    return dbConfig
 
 
 def moveDBLabelToLast(propertiesItems):
-    propertiesItems.sort(key=lambda x: 1 if x[0] == 'db_label' else 0)
-    
+    propertiesItems.sort(key=lambda x: 1 if x[0] == "db_label" else 0)

vectordb_bench/frontend/components/run_test/dbSelector.py

@@ -1,5 +1,5 @@
-
-from vectordb_bench.frontend.const import *
+from vectordb_bench.frontend.const.styles import *
+from vectordb_bench.frontend.const.dbCaseConfigs import DB_LIST
 
 
 def dbSelector(st):
@@ -12,10 +12,10 @@ def dbSelector(st):
         "<div style='color: #647489; margin-bottom: 24px; margin-top: -12px;'>Choose at least one case you want to run the test for. </div>",
         unsafe_allow_html=True,
     )
-    
+
     dbContainerColumns = st.columns(DB_SELECTOR_COLUMNS, gap="small")
     dbIsActived = {db: False for db in DB_LIST}
-    
+
     # style - image; column gap; checkbox font;
     st.markdown(
         """
@@ -32,5 +32,5 @@ def dbSelector(st):
         dbIsActived[db] = column.checkbox(db.name)
         column.image(DB_TO_ICON.get(db, ""))
     activedDbList = [db for db in DB_LIST if dbIsActived[db]]
-    
+
     return activedDbList

vectordb_bench/frontend/components/run_test/hideSidebar.py

@@ -1,10 +1,8 @@
 def hideSidebar(st):
     st.markdown(
-        "<style> div[data-testid='collapsedControl'] {display: none;} </style>",
-        unsafe_allow_html=True,
-    )
-
-    st.markdown(
-        "<style> .block-container { max-width: 1000px; } </style>",
+        """<style>
+            div[data-testid='collapsedControl'] {display: none;} 
+            .block-container { max-width: 1000px; }
+        </style>""",
         unsafe_allow_html=True,
     )

vectordb_bench/frontend/components/run_test/submitTask.py

@@ -1,9 +1,9 @@
 from datetime import datetime
-from vectordb_bench.frontend.const import *
+from vectordb_bench.frontend.const.styles import *
 from vectordb_bench.interface import benchMarkRunner
 
 
-def submitTask(st, tasks):
+def submitTask(st, tasks, isAllValid):
     st.markdown(
         "<div style='height: 24px;'></div>",
         unsafe_allow_html=True,
@@ -15,34 +15,36 @@ def submitTask(st, tasks):
     )
 
     taskLabel = taskLabelInput(st)
-    
+
     st.markdown(
         "<div style='height: 24px;'></div>",
         unsafe_allow_html=True,
     )
 
     controlPanelContainer = st.container()
-    controlPanel(controlPanelContainer, tasks, taskLabel)
+    controlPanel(controlPanelContainer, tasks, taskLabel, isAllValid)
 
 
 def taskLabelInput(st):
     defaultTaskLabel = datetime.now().strftime("%Y%m%d%H")
     columns = st.columns(TASK_LABEL_INPUT_COLUMNS)
-    taskLabel = columns[0].text_input("task_label", defaultTaskLabel, label_visibility="collapsed")
+    taskLabel = columns[0].text_input(
+        "task_label", defaultTaskLabel, label_visibility="collapsed"
+    )
     return taskLabel
 
 
-def controlPanel(st, tasks, taskLabel):
+def controlPanel(st, tasks, taskLabel, isAllValid):
     isRunning = benchMarkRunner.has_running()
     runHandler = lambda: benchMarkRunner.run(tasks, taskLabel)
     stopHandler = lambda: benchMarkRunner.stop_running()
-    
+
     if isRunning:
         currentTaskId = benchMarkRunner.get_current_task_id()
         tasksCount = benchMarkRunner.get_tasks_count()
         text = f":running: Running Task {currentTaskId} / {tasksCount}"
         st.progress(currentTaskId / tasksCount, text=text)
-        
+
         columns = st.columns(6)
         columns[0].button(
             "Run Your Test",
@@ -55,12 +57,16 @@ def controlPanel(st, tasks, taskLabel):
             on_click=stopHandler,
             type="primary",
         )
-        
+
     else:
         errorText = benchMarkRunner.latest_error or ""
         if len(errorText) > 0:
             st.error(errorText)
-        disabled = True if len(tasks) == 0 else False
+        disabled = True if len(tasks) == 0 or not isAllValid else False
+        if not isAllValid:
+            st.error("Make sure all config is valid.")
+        elif len(tasks) == 0:
+            st.warning("No tests to run.")
         st.button(
             "Run Your Test",
             disabled=disabled,

vectordb_bench/frontend/const/dbCaseConfigs.py

@@ -0,0 +1,291 @@
+from enum import IntEnum
+import typing
+from pydantic import BaseModel
+from vectordb_bench.backend.cases import CaseType
+from vectordb_bench.backend.clients import DB
+from vectordb_bench.backend.clients.api import IndexType
+
+from vectordb_bench.models import CaseConfigParamType
+
+MAX_STREAMLIT_INT = (1 << 53) - 1
+
+DB_LIST = [d for d in DB]
+
+DIVIDER = "DIVIDER"
+CASE_LIST_WITH_DIVIDER = [
+    CaseType.Performance100M,
+    CaseType.Performance10M,
+    CaseType.Performance1M,
+    DIVIDER,
+    CaseType.Performance10M1P,
+    CaseType.Performance1M1P,
+    DIVIDER,
+    CaseType.Performance10M99P,
+    CaseType.Performance1M99P,
+    DIVIDER,
+    CaseType.CapacityDim960,
+    CaseType.CapacityDim128,
+]
+
+CASE_LIST = [item for item in CASE_LIST_WITH_DIVIDER if isinstance(item, CaseType)]
+
+
+class InputType(IntEnum):
+    Text = 20001
+    Number = 20002
+    Option = 20003
+
+
+class CaseConfigInput(BaseModel):
+    label: CaseConfigParamType
+    inputType: InputType = InputType.Text
+    inputConfig: dict = {}
+    # todo type should be a function
+    isDisplayed: typing.Any = lambda x: True
+
+
+CaseConfigParamInput_IndexType = CaseConfigInput(
+    label=CaseConfigParamType.IndexType,
+    inputType=InputType.Option,
+    inputConfig={
+        "options": [
+            IndexType.HNSW.value,
+            IndexType.IVFFlat.value,
+            IndexType.DISKANN.value,
+            IndexType.Flat.value,
+            IndexType.AUTOINDEX.value,
+        ],
+    },
+)
+
+CaseConfigParamInput_M = CaseConfigInput(
+    label=CaseConfigParamType.M,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 4,
+        "max": 64,
+        "value": 30,
+    },
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
+    == IndexType.HNSW.value,
+)
+
+CaseConfigParamInput_EFConstruction_Milvus = CaseConfigInput(
+    label=CaseConfigParamType.EFConstruction,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 8,
+        "max": 512,
+        "value": 360,
+    },
+    isDisplayed=lambda config: config[CaseConfigParamType.IndexType]
+    == IndexType.HNSW.value,
+)
+
+CaseConfigParamInput_EFConstruction_Weaviate = CaseConfigInput(
+    label=CaseConfigParamType.EFConstruction,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 8,
+        "max": 512,
+        "value": 128,
+    },
+)
+
+CaseConfigParamInput_EFConstruction_ES = CaseConfigInput(
+    label=CaseConfigParamType.EFConstruction,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 8,
+        "max": 512,
+        "value": 360,
+    },
+)
+
+CaseConfigParamInput_M_ES = CaseConfigInput(
+    label=CaseConfigParamType.M,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 4,
+        "max": 64,
+        "value": 30,
+    },
+)
+
+CaseConfigParamInput_NumCandidates_ES = CaseConfigInput(
+    label=CaseConfigParamType.numCandidates,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 1,
+        "max": 10000,
+        "value": 100,
+    },
+)
+
+CaseConfigParamInput_EF_Milvus = CaseConfigInput(
+    label=CaseConfigParamType.EF,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 100,
+        "max": MAX_STREAMLIT_INT,
+        "value": 100,
+    },
+    isDisplayed=lambda config: config[CaseConfigParamType.IndexType]
+    == IndexType.HNSW.value,
+)
+
+CaseConfigParamInput_EF_Weaviate = CaseConfigInput(
+    label=CaseConfigParamType.EF,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": -1,
+        "max": MAX_STREAMLIT_INT,
+        "value": -1,
+    },
+)
+
+CaseConfigParamInput_MaxConnections = CaseConfigInput(
+    label=CaseConfigParamType.MaxConnections,
+    inputType=InputType.Number,
+    inputConfig={"min": 1, "max": MAX_STREAMLIT_INT, "value": 64},
+)
+
+CaseConfigParamInput_SearchList = CaseConfigInput(
+    label=CaseConfigParamType.SearchList,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 100,
+        "max": MAX_STREAMLIT_INT,
+        "value": 100,
+    },
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
+    == IndexType.DISKANN.value,
+)
+
+CaseConfigParamInput_Nlist = CaseConfigInput(
+    label=CaseConfigParamType.Nlist,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 1,
+        "max": 65536,
+        "value": 1000,
+    },
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
+    == IndexType.IVFFlat.value,
+)
+
+CaseConfigParamInput_Nprobe = CaseConfigInput(
+    label=CaseConfigParamType.Nprobe,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 1,
+        "max": 65536,
+        "value": 10,
+    },
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
+    == IndexType.IVFFlat.value,
+)
+
+CaseConfigParamInput_Lists = CaseConfigInput(
+    label=CaseConfigParamType.lists,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 1,
+        "max": 65536,
+        "value": 10,
+    },
+)
+
+CaseConfigParamInput_Probes = CaseConfigInput(
+    label=CaseConfigParamType.probes,
+    inputType=InputType.Number,
+    inputConfig={
+        "min": 1,
+        "max": 65536,
+        "value": 1,
+    },
+)
+
+
+MilvusLoadConfig = [
+    CaseConfigParamInput_IndexType,
+    CaseConfigParamInput_M,
+    CaseConfigParamInput_EFConstruction_Milvus,
+    CaseConfigParamInput_Nlist,
+]
+MilvusPerformanceConfig = [
+    CaseConfigParamInput_IndexType,
+    CaseConfigParamInput_M,
+    CaseConfigParamInput_EFConstruction_Milvus,
+    CaseConfigParamInput_EF_Milvus,
+    CaseConfigParamInput_SearchList,
+    CaseConfigParamInput_Nlist,
+    CaseConfigParamInput_Nprobe,
+]
+
+WeaviateLoadConfig = [
+    CaseConfigParamInput_MaxConnections,
+    CaseConfigParamInput_EFConstruction_Weaviate,
+]
+WeaviatePerformanceConfig = [
+    CaseConfigParamInput_MaxConnections,
+    CaseConfigParamInput_EFConstruction_Weaviate,
+    CaseConfigParamInput_EF_Weaviate,
+]
+
+ESLoadingConfig = [CaseConfigParamInput_EFConstruction_ES, CaseConfigParamInput_M_ES]
+ESPerformanceConfig = [
+    CaseConfigParamInput_EFConstruction_ES,
+    CaseConfigParamInput_M_ES,
+    CaseConfigParamInput_NumCandidates_ES,
+]
+
+PgVectorLoadingConfig = [CaseConfigParamInput_Lists]
+PgVectorPerformanceConfig = [CaseConfigParamInput_Lists, CaseConfigParamInput_Probes]
+
+CASE_CONFIG_MAP = {
+    DB.Milvus: {
+        CaseType.CapacityDim960: MilvusLoadConfig,
+        CaseType.CapacityDim128: MilvusLoadConfig,
+        CaseType.Performance100M: MilvusPerformanceConfig,
+        CaseType.Performance10M: MilvusPerformanceConfig,
+        CaseType.Performance1M: MilvusPerformanceConfig,
+        CaseType.Performance10M1P: MilvusPerformanceConfig,
+        CaseType.Performance1M1P: MilvusPerformanceConfig,
+        CaseType.Performance10M99P: MilvusPerformanceConfig,
+        CaseType.Performance1M99P: MilvusPerformanceConfig,
+    },
+    DB.WeaviateCloud: {
+        CaseType.CapacityDim960: WeaviateLoadConfig,
+        CaseType.CapacityDim128: WeaviateLoadConfig,
+        CaseType.Performance100M: WeaviatePerformanceConfig,
+        CaseType.Performance10M: WeaviatePerformanceConfig,
+        CaseType.Performance1M: WeaviatePerformanceConfig,
+        CaseType.Performance10M1P: WeaviatePerformanceConfig,
+        CaseType.Performance1M1P: WeaviatePerformanceConfig,
+        CaseType.Performance10M99P: WeaviatePerformanceConfig,
+        CaseType.Performance1M99P: WeaviatePerformanceConfig,
+    },
+    DB.ElasticCloud: {
+        CaseType.CapacityDim960: ESLoadingConfig,
+        CaseType.CapacityDim128: ESLoadingConfig,
+        CaseType.Performance100M: ESPerformanceConfig,
+        CaseType.Performance10M: ESPerformanceConfig,
+        CaseType.Performance1M: ESPerformanceConfig,
+        CaseType.Performance10M1P: ESPerformanceConfig,
+        CaseType.Performance1M1P: ESPerformanceConfig,
+        CaseType.Performance10M99P: ESPerformanceConfig,
+        CaseType.Performance1M99P: ESPerformanceConfig,
+    },
+    DB.PgVector: {
+        CaseType.CapacityDim960: PgVectorLoadingConfig,
+        CaseType.CapacityDim128: PgVectorLoadingConfig,
+        CaseType.Performance100M: PgVectorPerformanceConfig,
+        CaseType.Performance10M: PgVectorPerformanceConfig,
+        CaseType.Performance1M: PgVectorPerformanceConfig,
+        CaseType.Performance10M1P: PgVectorPerformanceConfig,
+        CaseType.Performance1M1P: PgVectorPerformanceConfig,
+        CaseType.Performance10M99P: PgVectorPerformanceConfig,
+        CaseType.Performance1M99P: PgVectorPerformanceConfig,
+    },
+}

vectordb_bench/frontend/const/dbPrices.py

@@ -0,0 +1,6 @@
+from vectordb_bench import config
+import ujson
+import pathlib
+
+with open(pathlib.Path(config.RESULTS_LOCAL_DIR, "dbPrices.json")) as f:
+    DB_DBLABEL_TO_PRICE = ujson.load(f)

vectordb_bench/frontend/const/styles.py

@@ -0,0 +1,58 @@
+from vectordb_bench.models import DB
+
+# style const
+DB_SELECTOR_COLUMNS = 6
+DB_CONFIG_SETTING_COLUMNS = 3
+CASE_CONFIG_SETTING_COLUMNS = 4
+CHECKBOX_INDENT = 30
+TASK_LABEL_INPUT_COLUMNS = 2
+CHECKBOX_MAX_COLUMNS = 4
+DB_CONFIG_INPUT_MAX_COLUMNS = 2
+CASE_CONFIG_INPUT_MAX_COLUMNS = 3
+DB_CONFIG_INPUT_WIDTH_RADIO = 2
+CASE_CONFIG_INPUT_WIDTH_RADIO = 0.98
+CASE_INTRO_RATIO = 3
+SIDEBAR_CONTROL_COLUMNS = 3
+LEGEND_RECT_WIDTH = 24
+LEGEND_RECT_HEIGHT = 16
+LEGEND_TEXT_FONT_SIZE = 14
+
+PATTERN_SHAPES = ["", "+", "\\", "x", ".", "|", "/", "-"]
+
+
+def getPatternShape(i):
+    return PATTERN_SHAPES[i % len(PATTERN_SHAPES)]
+
+
+# run_test page auto-refresh config
+MAX_AUTO_REFRESH_COUNT = 999999
+MAX_AUTO_REFRESH_INTERVAL = 5000  # 5s
+
+PAGE_TITLE = "VectorDB Benchmark"
+FAVICON = "https://assets.zilliz.com/favicon_f7f922fe27.png"
+HEADER_ICON = "https://assets.zilliz.com/vdb_benchmark_db790b5387.png"
+
+# RedisCloud icon: https://assets.zilliz.com/Redis_Cloud_74b8bfef39.png
+# Elasticsearch icon: https://assets.zilliz.com/elasticsearch_beffeadc29.png
+# Chroma icon: https://assets.zilliz.com/chroma_ceb3f06ed7.png
+DB_TO_ICON = {
+    DB.Milvus: "https://assets.zilliz.com/milvus_c30b0d1994.png",
+    DB.ZillizCloud: "https://assets.zilliz.com/zilliz_5f4cc9b050.png",
+    DB.ElasticCloud: "https://assets.zilliz.com/Elatic_Cloud_dad8d6a3a3.png",
+    DB.Pinecone: "https://assets.zilliz.com/pinecone_94d8154979.png",
+    DB.QdrantCloud: "https://assets.zilliz.com/qdrant_b691674fcd.png",
+    DB.WeaviateCloud: "https://assets.zilliz.com/weaviate_4f6f171ebe.png",
+    DB.PgVector: "https://assets.zilliz.com/PG_Vector_d464f2ef5f.png",
+}
+
+# RedisCloud color: #0D6EFD
+# Chroma color: #FFC107
+COLOR_MAP = {
+    DB.Milvus.value: "#0DCAF0",
+    DB.ZillizCloud.value: "#0D6EFD",
+    DB.ElasticCloud.value: "#04D6C8",
+    DB.Pinecone.value: "#6610F2",
+    DB.QdrantCloud.value: "#D91AD9",
+    DB.WeaviateCloud.value: "#20C997",
+    DB.PgVector.value: "#4C779A",
+}

vectordb_bench/frontend/pages/{qps_with_price.py → quries_per_dollar.py}

@@ -1,32 +1,32 @@
 import streamlit as st
+from vectordb_bench.frontend.components.check_results.footer import footer
+from vectordb_bench.frontend.components.check_results.expanderStyle import initMainExpanderStyle
 from vectordb_bench.frontend.components.check_results.priceTable import priceTable
-from vectordb_bench.frontend.const import *
+from vectordb_bench.frontend.components.check_results.stPageConfig import initResultsPageConfig
 from vectordb_bench.frontend.components.check_results.headerIcon import drawHeaderIcon
 from vectordb_bench.frontend.components.check_results.nav import NavToResults, NavToRunTest
 from vectordb_bench.frontend.components.check_results.charts import drawMetricChart
 from vectordb_bench.frontend.components.check_results.filters import getshownData
+from vectordb_bench.frontend.components.get_results.saveAsImage import getResults
+from vectordb_bench.frontend.const.styles import *
 from vectordb_bench.interface import benchMarkRunner
+from vectordb_bench.metric import QURIES_PER_DOLLAR_METRIC
 
 
 def main():
-    st.set_page_config(
-        page_title="VectorDB Benchmark",
-        page_icon="https://assets.zilliz.com/favicon_f7f922fe27.png",
-        # layout="wide",
-        # initial_sidebar_state="collapsed",
-    )
-
+    # set page config
+    initResultsPageConfig(st)
+    
     # header
     drawHeaderIcon(st)
 
     allResults = benchMarkRunner.get_results()
 
-    st.title("Vector DB Benchmark (QPS / Price)")
-    st.subheader("Price List")
+    st.title("Vector DB Benchmark (QP$)")
 
     # results selector
     resultSelectorContainer = st.sidebar.container()
-    shownData, failedTasks, showCases = getshownData(allResults, resultSelectorContainer)
+    shownData, _, showCases = getshownData(allResults, resultSelectorContainer)
 
     resultSelectorContainer.divider()
 
@@ -34,27 +34,34 @@ def main():
     navContainer = st.sidebar.container()
     NavToRunTest(navContainer)
     NavToResults(navContainer)
-    
+
+    # save or share
+    resultesContainer = st.sidebar.container()
+    getResults(resultesContainer, "vectordb_bench_qp$")
+
     # price table
+    initMainExpanderStyle(st)
     priceTableContainer = st.container()
     priceMap = priceTable(priceTableContainer, shownData)
 
     # charts
     for case in showCases:
-        chartContainer = st.container()
-        data = [data for data in shownData if data["case"] == case]
+        data = [data for data in shownData if data["case_name"] == case.name]
         dataWithMetric = []
-        metric = "qps_per_dollar (qps / price)"
+        metric = QURIES_PER_DOLLAR_METRIC
         for d in data:
             qps = d.get("qps", 0)
             price = priceMap.get(d["db"], {}).get(d["db_label"], 0)
             if qps > 0 and price > 0:
-                d[metric] = d["qps"] / price
+                d[metric] = d["qps"] / price * 3.6
                 dataWithMetric.append(d)
         if len(dataWithMetric) > 0:
-            chartContainer.subheader(case)
+            chartContainer = st.expander(case.name, True)
             drawMetricChart(data, metric, chartContainer)
 
+    # footer
+    footer(st.container())
+
 
 if __name__ == "__main__":
     main()