haylit2 / utils /ui.py
sabatale's picture
Update utils/ui.py
95ba7e9 verified
raw
history blame
916 Bytes
import streamlit as st
from PIL import Image
def set_state_if_absent(key, value):
if key not in st.session_state:
st.session_state[key] = value
def set_initial_state():
set_state_if_absent("username", "Provide a Mastodon username (e.g. xyz@mastodon.social)")
set_state_if_absent("result", None)
set_state_if_absent("haystack_started", False)
def reset_results(*args):
st.session_state.result = None
def set_openai_api_key(api_key: str):
st.session_state["OPENAI_API_KEY"] = api_key
def sidebar():
with st.sidebar:
st.markdown(
"## Thanks for checking this demo! 🎈\n"
"The AI has access to the following Knowledge base: https://help.etsy.com/hc/en-us.\n\n"
"Enter a short question about Etsy and click ASK. That's it!\n\n"
"The app currently runs on Cohere AI but can be changed to any other LLM provider."
)