chatbot_app.py

import streamlit as st
import chatbot_lib as glib

st.set_page_config(page_title="Chatbot", page_icon="🤖")
st.title("Chatbot")

if 'memory' not in st.session_state:
    st.session_state.memory = glib.get_memory()

if 'chat_history' not in st.session_state:
    st.session_state.chat_history = []

for message in st.session_state.chat_history:
    with st.chat_message(message["role"]):
        st.markdown(message["text"])

input_text = st.chat_input("Chat with your assistant here")

if input_text:

    with st.chat_message("human"):
        st.markdown(input_text)

    st.session_state.chat_history.append({"role": "user", "text": input_text})

    chat_response = glib.get_chat_response(
        input_text=input_text, memory=st.session_state.memory)

    with st.chat_message("assistant"):
        st.markdown(chat_response)

    st.session_state.chat_history.append(
        {"role": "assistant", "text": chat_response})