Function biz_bot_ui_app

solutions/use_cases/biz_bot.py:267–338

(db="postgres", table_name=None, fp=None,doc=None)

def biz_bot_ui_app (db="postgres", table_name=None, fp=None,doc=None):

    st.title(f"Biz Bot")

    parser_output = None

    if os.path.exists(os.path.join(fp,doc)):
        if not parser_output:
            parser_output = Parser().parse_one(fp, doc, save_history=False)

    prompter = load_prompt_model()
    reranker_model = load_reranker_model()
    agent = load_agent_model()

    # initialize chat history
    if "messages" not in st.session_state:
        st.session_state.messages = []

    # display chat messages from history on app rerun
    for message in st.session_state.messages:
        with st.chat_message(message["role"]):
            st.markdown(message["content"])

    with st.sidebar:

        st.write("Biz Bot")
        model_type = st.selectbox("Pick your mode", ("RAG","SQL"), index=0)

        uploaded_doc = st.file_uploader("Upload Document")
        uploaded_table = st.file_uploader("Upload CSV")

        if uploaded_doc:
            fp = LLMWareConfig().get_llmware_path()
            doc = uploaded_doc.name
            f = open(os.path.join(fp,doc), "wb")
            f.write(uploaded_doc.getvalue())
            f.close()
            parser_output = parse_file(fp, doc)
            st.write(f"Document Parsed and Ready - {len(parser_output)}")

        if uploaded_table:
            fp = LLMWareConfig().get_llmware_path()
            tab = uploaded_table.name
            f = open(os.path.join(fp,tab),"wb")
            f.write(uploaded_table.getvalue())
            f.close()
            table_name = tab.split(".")[0]
            st.write("Building Table - ", tab, table_name)
            st.write(st.session_state['loaded_tables'])
            row_count = build_table(db=db, table_name=table_name, load_fp=fp, load_file=tab)
            st.write(f"Completed - Table - {table_name} - Rows - {row_count} - is Ready.")

    # accept user input
    prompt = st.chat_input("Say something")
    if prompt:

        with st.chat_message("user"):
            st.markdown(prompt)