|
|
| Line 1: |
Line 1: |
| import streamlit as st
| | Kotoba |
| import app_utils as a_t
| |
| from tabs.home import home_tab
| |
| from tabs.play import playground_tab
| |
| from tabs.generate import generate_code_tab
| |
| | |
| def session_add(key, value, is_func=False):
| |
| """
| |
| Adds a key-value pair to the session state.
| |
| | |
| Args:
| |
| - key (str): The key to add to the session state.
| |
| - value (str): The value to add to the session state.
| |
| - is_func (bool): If True, calls the function `value` and adds the result to the session state.
| |
| """
| |
| if key not in st.session_state:
| |
| if is_func:
| |
| st.session_state[key] = value()
| |
| else:
| |
| st.session_state[key] = value
| |
| | |
| | |
| def init_keys():
| |
| """Initializes session keys."""
| |
| # All new session variables should be added here.
| |
| session_add("chroma_persisted", False)
| |
| session_add("vector_selection", "FAISS")
| |
| session_add("agent_selection", "🧑🔧 technical")
| |
| session_add("embedding_model", "HuggingFaceEmbeddings")
| |
| session_add("chunk_size", 1000)
| |
| session_add("chunk_overlap", 100)
| |
| session_add("messages", [])
| |
| session_add("model_temperature", 0.3)
| |
| session_add("store", {})
| |
| session_add("search_type", "similarity")
| |
| session_add("k", 4)
| |
| session_add("fetch_k", 20)
| |
| session_add("lambda_mult", 0.5)
| |
| session_add("score_threshold", 0.5)
| |
| session_add("history_unaware", False)
| |
| session_add("search_kwargs", {})
| |
| | |
| def render_site():
| |
| """Configures and displays the landing page."""
| |
| st.set_page_config("Document checker", page_icon="👁️🗨️")
| |
| with open("tabs/custom.css") as f:
| |
| st.markdown(f'<style>{f.read()}</style>', unsafe_allow_html=True)
| |
| st.title("Knowledge base LLM 💬")
| |
| st.text("Chat with your PDF file using the LLM of your choice")
| |
| st.write('''
| |
| Usage:
| |
| 1. export or define your UNIFY_KEY
| |
| 2. Select the **Model** and endpoint provider of your choice from the drop down.
| |
| 3. Upload your document(s) and click the Submit button
| |
| 4. Chat
| |
| ''')
| |
| | |
| for message in st.session_state.messages:
| |
| st.chat_message('human').write(message[0])
| |
| st.chat_message('assistant').write(message[1])
| |
|
| |
| with st.sidebar:
| |
| tab1, tab2, tab3 = st.tabs(["🏠Home", "🕹️Tuning", "👾Code"])
| |
| with tab1:
| |
| home_tab()
| |
| with tab2:
| |
| playground_tab()
| |
| with tab3:
| |
| generate_code_tab()
| |
|
| |
| a_t.chat_bot()
| |
| | |
| | |
| def main():
| |
| st.set_page_config(page_title="audit compliance check",page_icon=":books:")
| |
| st.header("metric comparison")
| |
| st.text_input("ask a question")
| |
| with st.sidebar:
| |
| st.subheader("read doc")
| |
| st.file_uploader("upload pdf")
| |
|
| |
| | |
| if __name__ == '__main__':
| |
| init_keys()
| |
| render_site()
| |