Langchain-Chatchat

1
import streamlit as st
2
from webui_pages.utils import *
3
from streamlit_option_menu import option_menu
4
from webui_pages.dialogue.dialogue import dialogue_page, chat_box
5
from webui_pages.knowledge_base.knowledge_base import knowledge_base_page
6
import os
7
import sys
8
from configs import VERSION
9
from server.utils import api_address
10

11

12
api = ApiRequest(base_url=api_address())
13

14
if __name__ == "__main__":
15
    is_lite = "lite" in sys.argv
16

17
    st.set_page_config(
18
        "Langchain-Chatchat WebUI",
19
        os.path.join("img", "chatchat_icon_blue_square_v2.png"),
20
        initial_sidebar_state="expanded",
21
        menu_items={
22
            'Get Help': 'https://github.com/chatchat-space/Langchain-Chatchat',
23
            'Report a bug': "https://github.com/chatchat-space/Langchain-Chatchat/issues",
24
            'About': f"""欢迎使用 Langchain-Chatchat WebUI {VERSION}！"""
25
        }
26
    )
27

28
    pages = {
29
        "对话": {
30
            "icon": "chat",
31
            "func": dialogue_page,
32
        },
33
        "知识库管理": {
34
            "icon": "hdd-stack",
35
            "func": knowledge_base_page,
36
        },
37
    }
38

39
    with st.sidebar:
40
        st.image(
41
            os.path.join(
42
                "img",
43
                "logo-long-chatchat-trans-v2.png"
44
            ),
45
            use_column_width=True
46
        )
47
        st.caption(
48
            f"""<p align="right">当前版本：{VERSION}</p>""",
49
            unsafe_allow_html=True,
50
        )
51
        options = list(pages)
52
        icons = [x["icon"] for x in pages.values()]
53

54
        default_index = 0
55
        selected_page = option_menu(
56
            "",
57
            options=options,
58
            icons=icons,
59
            # menu_icon="chat-quote",
60
            default_index=default_index,
61
        )
62

63
    if selected_page in pages:
64
        pages[selected_page]["func"](api=api, is_lite=is_lite)
65