update

app.py

@@ -16,16 +16,16 @@ def display_table(data, rows_per_page=10):
         height = min(40 + rows_per_page * 38, 800)  
         st.dataframe(data, height=height)
 
-
 def main():
-    st.title("Multihop-RAG Benchmark Space")
+    st.title("Multihop-RAG Benchmark 💡")
 
     data = load_data()
 
     st.sidebar.header("Search Options")
-    chat_model_query = st.sidebar.text_input("Search by Chat Model")
-    embedding_model_query = st.sidebar.text_input("Search by Embedding Model")
-    chunk_query = st.sidebar.text_input("Search by Chunk") 
+    chat_model_query = st.sidebar.text_input("Chat Model")
+    embedding_model_query = st.sidebar.text_input("Embedding Model")
+    chunk_query = st.sidebar.text_input("Chunk") 
+    frame_query = st.sidebar.text_input("Framework") 
 
     if chat_model_query:
         data = case_insensitive_search(data, chat_model_query, 'chat_model')
@@ -33,9 +33,9 @@ def main():
         data = case_insensitive_search(data, embedding_model_query, 'embedding_model')
     if chunk_query:  
         data = case_insensitive_search(data, chunk_query, 'chunk')
+    if frame_query:
+        data = case_insensitive_search(data, frame_query, 'framework')
 
-
-    st.header("Benchmark Results")
     st.write("Displaying results for MRR@10, Hit@10, and Accuracy across different frameworks, embedding models, chat models, and chunks.")
     display_table(data) 
 
@@ -48,10 +48,10 @@ def main():
     st.sidebar.header("Citation")
     st.sidebar.info(
         "Please cite this dataset as:\n"
-        "Author et al. (2024). Multihop-RAG Benchmark Dataset. Retrieved from [Source URL]."
+        "Tang, Yixuan, and Yi Yang. MultiHop-RAG: Benchmarking Retrieval-Augmented Generation for Multi-Hop Queries. ArXiv, 2024,  /abs/2401.15391."
     )
     st.markdown("---")
-    st.caption("For citation, please use: 'Author et al. (2024), Multihop-RAG Benchmark Dataset, Retrieved from [Source URL].'")
+    st.caption("For citation, please use: 'Tang, Yixuan, and Yi Yang. MultiHop-RAG: Benchmarking Retrieval-Augmented Generation for Multi-Hop Queries. ArXiv, 2024,  /abs/2401.15391. '")
 
 if __name__ == "__main__":
     main()