refactor

app.py

@@ -21,6 +21,28 @@ sys.modules["sqlite3"] = sys.modules.pop("pysqlite3")
 st.set_page_config(page_title="Hello", page_icon="👋", layout="wide")
 
 
+def show_sidebar():
+    # Sidebar
+    st.sidebar.header(("About"))
+    st.sidebar.markdown(
+        (
+            "[Brian](https://www.brianknows.org/) Built on top of Brian API, Brian App offers an user interface for performing transactions in a non-custodial way, researching web3 info, and deploying smart contracts by prompt."
+        )
+    )
+
+    st.sidebar.header(("Resources"))
+    st.sidebar.markdown(
+        (
+            """
+    - [Brian Documentation](https://docs.brianknows.org/)
+    - [X (Twitter)](https://x.com/BrianknowsAI?mx=2)
+    - [Linkedin](https://www.linkedin.com/company/brianknowsai/)
+    - [Medium](https://medium.com/@BrianknowsAI) 
+    """
+        )
+    )
+
+
 def check_password():
     """Returns `True` if the user had the correct password."""
 
@@ -48,71 +70,63 @@ def check_password():
 if not check_password():
     st.stop()  # Do not continue if check_password is not True.
 
+
 # Main Streamlit app starts here
 client = get_chroma_client()
 default_embedding_function = get_embedding_function()
 
-col1, col2, col3 = st.columns((1, 4, 1))
-with col2:
-    st.image("https://brianknows.org/brian_logo.png", width=300)
-st.write("# Brian Knowledge Base System! 👋")
-
-# Sidebar
-st.sidebar.header(("About"))
-st.sidebar.markdown(
-    (
-        "[Brian](https://www.brianknows.org/) Built on top of Brian API, Brian App offers an user interface for performing transactions in a non-custodial way, researching web3 info, and deploying smart contracts by prompt."
-    )
-)
-
-st.sidebar.header(("Resources"))
-st.sidebar.markdown(
-    (
-        """
-- [Brian Documentation](https://docs.brianknows.org/)
-- [X (Twitter)](https://x.com/BrianknowsAI?mx=2)
-- [Linkedin](https://www.linkedin.com/company/brianknowsai/)
-- [Medium](https://medium.com/@BrianknowsAI) 
-"""
-    )
-)
 
-components.iframe("https://www.brianknows.org/", height=650, scrolling=True)
+# Function to load a page
+def load_page(page_name):
+    with open(f"pages/{page_name}", "r") as file:
+        exec(file.read(), globals())
 
-st.divider()
 
+client = get_chroma_client()
+default_embedding_function = get_embedding_function()
 
-st.markdown("## Ask Brian Anything")
-kb_name = "public-knowledge-box"
+show_sidebar()
 
-load_dotenv()
-api_key = os.getenv("BRIAN_API_KEY")
+col1, col2, col3 = st.columns((1, 4, 1))
+with col2:
+    st.image("https://brianknows.org/brian_logo.png", width=300)
+st.write("# Brian Knowledge Base System! 👋")
 
 
-def send_post_request(prompt, kb):
-    url = " https://api.brianknows.org/api/v0/agent/knowledge"
-    data = {"prompt": prompt, "kb": kb}
-    headers = {
-        "Content-Type": "application/json",
-        "X-Brian-Api-Key": api_key,  # Include the API key in the headers
-    }
+tab1, tab2 = st.tabs(["AskBrian", "BrianApp"])
+with tab1:
+    st.markdown("## Ask Brian Anything")
+    kb_name = "public-knowledge-box"
 
-    response = requests.post(url, json=data, headers=headers)
+    load_dotenv()
+    api_key = os.getenv("BRIAN_API_KEY")
 
-    if response.status_code == 200:
-        return response.json()  # Returns the JSON response if successful
-    else:
-        return (
-            response.status_code,
-            response.text,
-        )  # Returns the status code and error if not successful
+    def send_post_request(prompt, kb):
+        url = " https://api.brianknows.org/api/v0/agent/knowledge"
+        data = {"prompt": prompt, "kb": kb}
+        headers = {
+            "Content-Type": "application/json",
+            "X-Brian-Api-Key": api_key,  # Include the API key in the headers
+        }
 
+        response = requests.post(url, json=data, headers=headers)
 
-# Example usage:
-kbs = get_current_knowledge_bases(client=client)
-kbs = (kb.name for kb in kbs)
-kb_name = st.selectbox("Select knowledge box", kbs)
-query = st.text_input(label="query")
-if st.button("askbrian"):
-    result = send_post_request(query, kb_name)
-    st.json(result)
+        if response.status_code == 200:
+            return response.json()  # Returns the JSON response if successful
+        else:
+            return (
+                response.status_code,
+                response.text,
+            )  # Returns the status code and error if not successful
+
+    # Example usage:
+    kbs = get_current_knowledge_bases(client=client)
+    kbs = (kb.name for kb in kbs)
+    kb_name = st.selectbox("Select knowledge box", kbs)
+    query = st.text_input(label="query")
+    if st.button("askbrian"):
+        result = send_post_request(query, kb_name)
+        st.json(result)
+
+with tab2:
+    components.iframe("https://www.brianknows.org/", height=650, scrolling=True)

pages/manage_knowledge_box.py

@@ -11,15 +11,14 @@ from openai import OpenAI
 import wave
 from dotenv import load_dotenv
 
+
 load_dotenv()
 openai_key = os.getenv("OPENAI_API_KEY")
 
-st.title("Get knowledge boxes")
-if st.button("Get current knowledge bases"):
-    kbs = get_current_knowledge_bases(client=client)
-    st.json(kbs)
-
-collection_name = st.text_input(label="knowledge base name")
+st.title("Manage collections")
+kbs = get_current_knowledge_bases(client=client)
+kbs = (kb.name for kb in kbs)
+collection_name = st.selectbox("Select knowledge box", kbs)
 info = {}
 collection = None