weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update README.md

This commit is contained in:
imClumsyPanda 2023-04-12 21:09:37 +08:00
parent 26f398f3a8
commit 3fbbf1d74a
1 changed files with 49 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
webui.py
View File

@ -70,16 +70,26 @@ with gr.Blocks(css="""
show_label=False).style(height=600) show_label=False).style(height=600)
with gr.Column(scale=1): with gr.Column(scale=1):
with gr.Column(): with gr.Column():
llm_model = gr.Radio( llm_model = gr.Radio(llm_model_dict_list,
llm_model_dict_list, label="llm model", label="llm model",
value="chatglm-6b", interactive=True) value="chatglm-6b",
LLM_HISTORY_LEN = gr.Slider( interactive=True)
1, 10, value=3, step=1, label="LLM history len", interactive=True) LLM_HISTORY_LEN = gr.Slider(1,
embedding_model = gr.Radio( 10,
embedding_model_dict_list, label="embedding model", value=3,
value="text2vec", interactive=True) step=1,
VECTOR_SEARCH_TOP_K = gr.Slider( label="LLM history len",
1, 20, value=6, step=1, label="vector search top k", interactive=True) interactive=True)
embedding_model = gr.Radio(embedding_model_dict_list,
label="embedding model",
value="text2vec",
interactive=True)
VECTOR_SEARCH_TOP_K = gr.Slider(1,
20,
value=6,
step=1,
label="vector search top k",
interactive=True)
load_model_button = gr.Button("step.1：setting") load_model_button = gr.Button("step.1：setting")
load_model_button.click(lambda *args: load_model_button.click(lambda *args:
kb.init_cfg(args[0], args[1], args[2], args[3]), kb.init_cfg(args[0], args[1], args[2], args[3]),
@ -92,14 +102,18 @@ with gr.Blocks(css="""
with gr.Column(): with gr.Column():
with gr.Tab("select"): with gr.Tab("select"):
selectFile = gr.Dropdown( selectFile = gr.Dropdown(file_list,
file_list, label="content file", interactive=True, label="content file",
value=file_list[0] if len(file_list) > 0 else None) interactive=True,
value=file_list[0] if len(file_list) > 0 else None)
with gr.Tab("upload"): with gr.Tab("upload"):
file = gr.File(label="content file", file_types=[ file = gr.File(label="content file",
'.txt', '.md', '.docx']).style(height=100) file_types=['.txt', '.md', '.docx']
).style(height=100)
# 将上传的文件保存到content文件夹下,并更新下拉框 # 将上传的文件保存到content文件夹下,并更新下拉框
file.upload(upload_file, inputs=file, outputs=selectFile) file.upload(upload_file,
inputs=file,
outputs=selectFile)
history = gr.State([]) history = gr.State([])
vector_store = gr.State() vector_store = gr.State()
load_button = gr.Button("step.2：loading") load_button = gr.Button("step.2：loading")
@ -108,19 +122,30 @@ with gr.Blocks(css="""
"content/" + fileName), "content/" + fileName),
show_progress=True, show_progress=True,
api_name="init_knowledge_vector_store", api_name="init_knowledge_vector_store",
inputs=selectFile, outputs=vector_store).then( inputs=selectFile,
get_file_status, chatbot, chatbot, show_progress=True, outputs=vector_store
).then(
get_file_status,
chatbot,
chatbot,
show_progress=True,
) )
with gr.Row(): with gr.Row():
with gr.Column(scale=2): with gr.Column(scale=2):
query = gr.Textbox(show_label=False, placeholder="Prompts", lines=1, value="用200字总结一下").style( query = gr.Textbox(show_label=False,
container=False) placeholder="Prompts",
lines=1,
value="用200字总结一下"
).style(container=False)
with gr.Column(scale=1): with gr.Column(scale=1):
generate_button = gr.Button( generate_button = gr.Button("step.3：asking",
"step.3：asking", elem_classes="importantButton") elem_classes="importantButton")
generate_button.click(get_answer, [query, vector_store, chatbot], generate_button.click(get_answer,
[chatbot, history], api_name="get_knowledge_based_answer") [query, vector_store, chatbot],
[chatbot, history],
api_name="get_knowledge_based_answer"
)
demo.queue(concurrency_count=3).launch( demo.queue(concurrency_count=3).launch(
server_name='0.0.0.0', share=False, inbrowser=False) server_name='0.0.0.0', share=False, inbrowser=False)