restyle app

app.py

@@ -6,6 +6,12 @@ from utils import get_chroma_client, get_embedding_function
 import hmac
 import streamlit as st
 import os
+import streamlit.components.v1 as components
+from retrieve_kb import get_current_knowledge_bases, get_knowledge_base_information
+import streamlit as st
+import requests
+import os
+from dotenv import load_dotenv
 
 __import__("pysqlite3")
 import sys
@@ -43,6 +49,70 @@ if not check_password():
     st.stop()  # Do not continue if check_password is not True.
 
 # Main Streamlit app starts here
-st.write("# Brian Knowledge Base System! 👋")
 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)
+
+st.divider()
+
+
+st.markdown("## Ask Brian Anything")
+kb_name = "public-knowledge-box"
+
+load_dotenv()
+api_key = os.getenv("BRIAN_API_KEY")
+
+
+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)
+
+    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)

pages/ask_brian.py

@@ -1,31 +0,0 @@
-import streamlit as st
-import requests
-
-api_key = st.secrets["BRIAN_API_KEY"]
-
-
-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)
-
-    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:
-kb_name = st.text_input(label="kb_name")
-query = st.text_input(label="query")
-if st.button("askbrian"):
-    result = send_post_request(query, kb_name)
-    st.json(result)