-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
97 lines (87 loc) · 4.97 KB
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
import gradio as gr
import os
from chatAll import config
from chatAll.utils import (add_text,
generate_response,
clear_history,
generate_response_with_file,
upload_file,
get_text_files,
choose_file,
clear_history2)
def create_chain_app():
with gr.Blocks(title="Chatbot") as demo:
with gr.Tab(label='Chat-Tab'):
text_chatbot1 = gr.Chatbot(
[],
label="chatBot",
elem_id="chatBot",
avatar_images=((os.path.join(os.path.dirname(__file__), "using_files/img", "user.png")),
(os.path.join(os.path.dirname(__file__), "using_files/img", "avatar.jpg"))),
bubble_full_width=False,
height="600px",
)
with gr.Row():
chat_text_input1 = gr.Textbox(scale=10, interactive=True, lines=3, show_label=False, render=False,
placeholder="愿起一剑杀万劫...")
gr.Examples(config.examples, chat_text_input1, label='示例')
with gr.Row():
chat_text_input1.render()
text_submit_button1 = gr.Button(value='Chat', variant='primary', scale=4)
text_clear_button1 = gr.Button(scale=2, value="Clear", variant="secondary")
with gr.Tab(label='File-Chat-Tab'):
with gr.Row():
filechatbot = gr.Chatbot(
[],
label="fileChatBot",
elem_id="fileChatBot",
avatar_images=((os.path.join(os.path.dirname(__file__), "using_files/img", "user.png")),
(os.path.join(os.path.dirname(__file__), "using_files/img", "avatar.jpg"))),
bubble_full_width=False,
height=550,
)
show_text = gr.Textbox(label='File Preview', lines=25, placeholder=config.article, interactive=True)
with gr.Row():
file_chat_input = gr.Textbox(render=False, scale=10, placeholder="因为困难多壮志...", lines=5,
show_label=False, interactive=True, )
gr.Examples(config.examples2, file_chat_input)
with gr.Row():
text_files_short, text_files = get_text_files(config.file_default_path)
file_chat_input.render()
submit_btn = gr.Button(value='Chat', variant='primary', scale=5)
with gr.Column(scale=1):
upload_type = gr.Dropdown(label='重新上传文件', choices=['否', '是'], value='否', scale=1)
upload_btn = gr.UploadButton("📁 上传文件聊天", file_types=config.upload_type, scale=1)
with gr.Column(scale=1):
choose_btn = gr.Dropdown(label='📁 选择', choices=text_files, scale=1)
preview_btn = gr.Button(value='选择文件聊天', scale=1)
upload_btn.GRADIO_CACHE = config.file_default_path
btn_clear_his = gr.Button(scale=2, value="Clear", variant="secondary")
with gr.Tab(label='Structure-Tab'):
img = gr.Image('using_files/img/img.png')
# 第一个模块
# queue=False参数，这意味着点击按钮后，只有当当前事件处理完成后，才能再次点击按钮。
text_submit_button1.click(add_text, inputs=[text_chatbot1, chat_text_input1], outputs=[text_chatbot1],
queue=False).success(generate_response, inputs=[text_chatbot1, chat_text_input1],
outputs=[text_chatbot1, chat_text_input1])
text_clear_button1.click(clear_history, inputs=text_chatbot1, outputs=[text_chatbot1, chat_text_input1])
# 第二个模块
upload_btn.upload(upload_file, [upload_btn, upload_type], show_text)
preview_btn.click(choose_file, inputs=[choose_btn, gr.Textbox(text_files_short, visible=False)],
outputs=show_text)
btn_clear_his.click(clear_history2, inputs=[filechatbot, show_text],
outputs=[filechatbot, file_chat_input, show_text])
# chat_msg = submit_btn.click(add_text_with_file, [filechatbot, file_chat_input],
# [filechatbot, show_text, file_chat_input])
return demo
if __name__ == "__main__":
app = create_chain_app()
app.launch(server_name="0.0.0.0", server_port=2333, share=False)
# start = time.time()
# split_docs = load_documents('using_files/data')
# for doc in split_docs:
# print(doc)
# end = time.time()
# print(f"数据切分时间：{(end - start) / 60 % 60:.4f}分({end - start:.4f}秒)")
# retriever = get_retriever(split_docs, embedding)
# chain = getChain(retriever)