fix: dataframe rendering error

README.md

@@ -1,8 +1,7 @@
 ---
-title: AgentBase Platform
-emoji: 🤖
-colorFrom: blue
-colorTo: green
+title: AgentBase Search Platform
+emoji: 🔍
+color: red
 sdk: streamlit
 sdk_version: 1.44.0
 app_file: streamlit_app.py

pages/interface.py

@@ -59,7 +59,6 @@ class AgentBaseUI:
         self.indexing_config = "v1"
 
     def header_panel(self):
-        st.set_page_config(page_title="AgentBase", layout="wide")
         st.title("AgentBase Platform Demo")
         st.write("A Large-Scale Agent Collection for Automated Agent Recommendation.")
 
@@ -152,13 +151,11 @@ class AgentBaseUI:
         with st.expander(f"View AgentBase-v1.1"):
             st.dataframe(
                 self.agents_df,
-                self.agents_df.columns,
                 use_container_width=True,
                 hide_index=True
             )
             st.dataframe(
                 self.platforms_df,
-                self.platforms_df.columns,
                 use_container_width=True,
                 hide_index=True
             )

pages/retrieval.py

@@ -16,6 +16,8 @@ from retrieval.models.bm25 import BM25Retriever
 from retrieval.models.sentence_bert import DenseRetriever
 from retrieval.utils import load_queries
 
+st.set_page_config(page_title="AgentBase", layout="wide")
+
 
 @st.cache_resource()
 def load_retrievers(agentbase_path: str, index_configs: List[str]) -> Tuple[dict, dict]:
@@ -49,7 +51,6 @@ class AgentBaseUI:
         self.indexing_config = "v1"
 
     def header_panel(self):
-        st.set_page_config(page_title="AgentBase", layout="wide")
         st.title("AgentBase Retriever")
         st.write("A Large-Scale Agent Collection for Automated Agent Recommendation.")
 

streamlit_app.py

@@ -1,4 +1,4 @@
 import streamlit as st
 
-st.set_page_config(layout="wide")
+st.set_page_config(page_title="AgentBase", layout="wide")
 st.switch_page("pages/interface.py")