app.py

import logging
import sys

import streamlit as st
from dotenv import load_dotenv

from faq.manager_factory import FAQRobotManagerFactory, FAQRobotRevision

logging.basicConfig(
    stream=sys.stdout, level=logging.INFO
)  # logging.DEBUG for more verbose output
# logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))

# # Sidebar contents
with st.sidebar:
    st.title("🤗💬 LLM Chat App")
    st.markdown(
        """
    ## About
    This app is an LLM-powered chatbot built using:
    - [Streamlit](https://streamlit.io/)
    - [LangChain](https://python.langchain.com/)
    """
    )
    # add_vertical_space(5)
    st.write("Made by Nick")


def main() -> None:
    st.header("星巴克门店伙伴小蜜 💬")

    robot_manager = FAQRobotManagerFactory.get_or_create(
        FAQRobotRevision.SIMPLE_OPENAI_VERSION_0
    )
    robot = robot_manager.get_robot()
    query = st.text_input("请输入你的问题:")
    if query:
        response = robot.ask(question=query)
        st.write(response)


if __name__ == "__main__":
    load_dotenv()
    main()