webui.py

import streamlit as st

from llama_index.core.base.llms.types import ChatMessage

st.set_page_config(page_title="NoC RAGbot", page_icon=":robot_face:")

st.title("NoC RAGbot: Porozprávaj sa so svojimi dátami")
st.caption("Powered by LlamaIndex & Streamlit & OpenAI")


@st.cache_resource(show_spinner=False)
def init():
    with st.spinner(text="Načítavam dokumenty, počkajte prosím."):
        pass

        # -------------------------
        # Init RAGbot here
        #
        #
        # -------------------------


bot = init()


def add_msg_to_ui_conversation(role: str, content: str) -> None:
    st.session_state.messages.append(ChatMessage(role=role, content=str(content)))


# Initialize the chat message history
if "messages" not in st.session_state.keys():
    initial_msg = ChatMessage(
        role="assistant",
        content="Ahoj, spýtaj sa niečo svojich dát!",
    )
    st.session_state.messages = [initial_msg]
    # ---------------------------
    # ---------------------------

if question := st.chat_input(
    "Vaša otázka"
):  # Prompt for user input and save to chat history
    add_msg_to_ui_conversation(role="user", content=question)

# Display the prior chat messages
for message in st.session_state.messages:
    with st.chat_message(message.role):
        st.write(message.content)

# If last message is not from assistant, generate a new response
if st.session_state.messages[-1].role != "assistant":
    with st.chat_message("assistant"):
        with st.spinner("Pracujem na odpovedi..."):
            placeholder = st.empty()

            # -------------------------------------
            # Ask question and retrieve asnwer here
            #
            #
            # -------------------------------------
            response = "Zatiaľ nič nie je implementované :/"

            placeholder.markdown(response)
            add_msg_to_ui_conversation(role="assistant", content=response)