diff --git a/README.md b/README.md
index 7369db3..1e9602f 100644
--- a/README.md
+++ b/README.md
@@ -255,7 +255,7 @@ Web UI 可以实现如下功能:
- [x] VUE 前端
## 项目交流群
-
+
🎉 langchain-ChatGLM 项目微信交流群,如果你也对本项目感兴趣,欢迎加入群聊参与讨论交流。
diff --git a/img/qr_code_44.jpg b/img/qr_code_44.jpg
new file mode 100644
index 0000000..58ccd6c
Binary files /dev/null and b/img/qr_code_44.jpg differ
diff --git a/webui_st.py b/webui_st.py
index 541851f..1584a55 100644
--- a/webui_st.py
+++ b/webui_st.py
@@ -23,6 +23,7 @@ def get_vs_list():
if not os.path.exists(KB_ROOT_PATH):
return lst_default
lst = os.listdir(KB_ROOT_PATH)
+ lst = [x for x in lst if os.path.isdir(os.path.join(KB_ROOT_PATH, x))]
if not lst:
return lst_default
lst.sort()
@@ -283,20 +284,24 @@ with st.sidebar:
def on_new_kb():
name = st.session_state.kb_name
- if name in vs_list:
- st.error(f'名为“{name}”的知识库已存在。')
+ if not name:
+ st.sidebar.error(f'新建知识库名称不能为空!')
+ elif name in vs_list:
+ st.sidebar.error(f'名为“{name}”的知识库已存在。')
else:
- vs_list.append(name)
st.session_state.vs_path = name
+ st.session_state.kb_name = ''
+ new_kb_dir = os.path.join(KB_ROOT_PATH, name)
+ if not os.path.exists(new_kb_dir):
+ os.makedirs(new_kb_dir)
+ st.sidebar.success(f'名为“{name}”的知识库创建成功,您可以开始添加文件。')
def on_vs_change():
chat_box.robot_say(f'已加载知识库: {st.session_state.vs_path}')
with st.expander('知识库配置', True):
cols = st.columns([12, 10])
kb_name = cols[0].text_input(
- '新知识库名称', placeholder='新知识库名称', label_visibility='collapsed')
- if 'kb_name' not in st.session_state:
- st.session_state.kb_name = kb_name
+ '新知识库名称', placeholder='新知识库名称', label_visibility='collapsed', key='kb_name')
cols[1].button('新建知识库', on_click=on_new_kb)
index = 0
try: