update

app.py

@@ -9,12 +9,18 @@ def case_insensitive_search(data, query, column):
         return data[data[column].str.lower().str.contains(query.lower())]
     return data
 
-def display_table(data, rows_per_page=10, height=None):
-    container = st.container()
-    with container:
-        if height is None:
-            height = min(40 + rows_per_page * 38, 800)
-        st.dataframe(data, height=height)
+def display_table(data):
+    # 设置多级列索引
+    data.columns = pd.MultiIndex.from_tuples([
+        ("Settings", "Framework"),
+        ("Settings", "Chat Model"),
+        ("Settings", "Embedding Model"),
+        ("Settings", "Chunk"),
+        ("Retrieval Metrics", "MRR@10"),
+        ("Retrieval Metrics", "Hit@10"),
+        ("Response Metrics", "Accuracy")
+    ])
+    st.dataframe(data, height=600)
 
 def main():
     st.title("Multihop-RAG Benchmark 💡")
@@ -36,17 +42,13 @@ def main():
     if frame_query:
         data = case_insensitive_search(data, frame_query, 'framework')
 
-    # Display settings
-    st.header("Settings")
-    display_table(data[['framework', 'chat_model', 'embedding_model', 'chunk']])
+    # 根据需要选择列
+    data = data[['framework', 'chat_model', 'embedding_model', 'chunk', 'MRR@10', 'Hit@10', 'Accuracy']]
+    display_table(data)
 
-    # Display retrieval metrics
-    st.header("Retrieval Metrics")
-    display_table(data[['MRR@10', 'Hit@10']])
-
-    # Display response metrics
-    st.header("Response Metrics")
-    display_table(data[['Accuracy']])
+    if st.sidebar.checkbox("Show Metrics Distribution"):
+        st.subheader("Metrics Distribution")
+        st.bar_chart(data[['MRR@10', 'Hit@10', 'Accuracy']]) 
 
     st.sidebar.header("Citation")
     st.sidebar.info(