X-Pipe/flash

1import logging
2import sys
3
4import streamlit as st
5from dotenv import load_dotenv
6
7from xpipe_wiki.manager_factory import XPipeRobotManagerFactory, XPipeRobotRevision
8
9logging.basicConfig(
10    stream=sys.stdout, level=logging.INFO
11)  # logging.DEBUG for more verbose output
12# logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
13
14# # Sidebar contents
15with st.sidebar:
16    st.title("🤗💬 LLM Chat App")
17    st.markdown(
18        """
19    ## About
20    This app is an LLM-powered chatbot built using:
21    - [Streamlit](https://streamlit.io/)
22    - [LangChain](https://python.langchain.com/)
23    - [X-Pipe](https://github.com/ctripcorp/x-pipe)
24    """
25    )
26    # add_vertical_space(5)
27    st.write("Made by Nick")
28
29
30def main() -> None:
31    st.header("X-Pipe Wiki 机器人 💬")
32
33    robot_manager = XPipeRobotManagerFactory.get_or_create(
34        XPipeRobotRevision.SIMPLE_OPENAI_VERSION_0
35    )
36    robot = robot_manager.get_robot()
37    query = st.text_input("X-Pipe Wiki 问题:")
38    if query:
39        response = robot.ask(question=query)
40        st.write(response)
41
42
43if __name__ == "__main__":
44    load_dotenv()
45    main()
46