Spaces:
Running
Running
| # -*- coding:utf-8 -*- | |
| import gradio as gr | |
| import os | |
| from utils import * | |
| from presets import * | |
| api_key = os.environ.get('api_key') | |
| gr.Chatbot.postprocess = postprocess | |
| with gr.Blocks(css=customCSS) as server: | |
| gr.HTML(title) | |
| with gr.Row(): | |
| with gr.Column(scale=4): | |
| keyTxt = gr.Textbox(show_label=False, placeholder=f"在这里输入你的OpenAI API-key...", | |
| value=api_key, type="password", visible=not HIDE_MY_KEY).style(container=True) | |
| with gr.Column(scale=1): | |
| use_streaming_checkbox = gr.Checkbox( | |
| label="实时传输回答", value=True, visible=enable_streaming_option) | |
| chatbot = gr.Chatbot(elem_id="chat").style( | |
| color_map=("#1D51EE", "#ffffff")) | |
| history = gr.State([]) | |
| token_count = gr.State([]) | |
| promptTemplates = gr.State(load_template( | |
| get_template_names(plain=True)[0], mode=2)) | |
| TRUECOMSTANT = gr.State(True) | |
| FALSECONSTANT = gr.State(False) | |
| topic = gr.State("未命名对话历史记录") | |
| with gr.Row(): | |
| with gr.Column(scale=12): | |
| user_input = gr.Textbox(show_label=False, placeholder="在这里输入").style( | |
| container=False) | |
| with gr.Column(min_width=50, scale=1): | |
| submitBtn = gr.Button("🚀", variant="primary") | |
| with gr.Row(): | |
| emptyBtn = gr.Button("🧹 新的对话") | |
| retryBtn = gr.Button("🔄 重新生成") | |
| delLastBtn = gr.Button("🗑️ 删除最近一条对话") | |
| reduceTokenBtn = gr.Button("♻️ 总结对话") | |
| status_display = gr.Markdown("status: ready") | |
| systemPromptTxt = gr.Textbox(show_label=True, placeholder=f"在这里输入System Prompt...", | |
| label="System prompt", value=initial_prompt).style(container=True) | |
| with gr.Accordion(label="加载Prompt模板", open=False): | |
| with gr.Column(): | |
| with gr.Row(): | |
| with gr.Column(scale=6): | |
| templateFileSelectDropdown = gr.Dropdown(label="选择Prompt模板集合文件", choices=get_template_names( | |
| plain=True), multiselect=False, value=get_template_names(plain=True)[0]) | |
| with gr.Column(scale=1): | |
| templateRefreshBtn = gr.Button("🔄 刷新") | |
| templaeFileReadBtn = gr.Button("📂 读入模板") | |
| with gr.Row(): | |
| with gr.Column(scale=6): | |
| templateSelectDropdown = gr.Dropdown(label="从Prompt模板中加载", choices=load_template(get_template_names( | |
| plain=True)[0], mode=1), multiselect=False, value=load_template(get_template_names(plain=True)[0], mode=1)[0]) | |
| with gr.Column(scale=1): | |
| templateApplyBtn = gr.Button("⬇️ 应用") | |
| with gr.Accordion(label="保存/加载对话历史记录", open=False): | |
| with gr.Column(): | |
| with gr.Row(): | |
| with gr.Column(scale=6): | |
| saveFileName = gr.Textbox( | |
| show_label=True, placeholder=f"在这里输入保存的文件名...", label="设置保存文件名", value="对话历史记录").style(container=True) | |
| with gr.Column(scale=1): | |
| saveHistoryBtn = gr.Button("💾 保存对话") | |
| with gr.Row(): | |
| with gr.Column(scale=6): | |
| historyFileSelectDropdown = gr.Dropdown(label="从列表中加载对话", choices=get_history_names( | |
| plain=True), multiselect=False, value=get_history_names(plain=True)[0]) | |
| with gr.Column(scale=1): | |
| historyRefreshBtn = gr.Button("🔄 刷新") | |
| historyReadBtn = gr.Button("📂 读入对话") | |
| # inputs, top_p, temperature, top_k, repetition_penalty | |
| with gr.Accordion("参数", open=False): | |
| top_p = gr.Slider(minimum=-0, maximum=1.0, value=1.0, step=0.05, | |
| interactive=True, label="Top-p (nucleus sampling)",) | |
| temperature = gr.Slider(minimum=-0, maximum=5.0, value=1.0, | |
| step=0.1, interactive=True, label="Temperature",) | |
| # top_k = gr.Slider( minimum=1, maximum=50, value=4, step=1, interactive=True, label="Top-k",) | |
| # repetition_penalty = gr.Slider( minimum=0.1, maximum=3.0, value=1.03, step=0.01, interactive=True, label="Repetition Penalty", ) | |
| gr.Markdown(description) | |
| user_input.submit(predict, [keyTxt, systemPromptTxt, history, user_input, chatbot, token_count, top_p, | |
| temperature, use_streaming_checkbox], [chatbot, history, status_display, token_count], show_progress=True) | |
| user_input.submit(reset_textbox, [], [user_input]) | |
| submitBtn.click(predict, [keyTxt, systemPromptTxt, history, user_input, chatbot, token_count, top_p, | |
| temperature, use_streaming_checkbox], [chatbot, history, status_display, token_count], show_progress=True) | |
| submitBtn.click(reset_textbox, [], [user_input]) | |
| emptyBtn.click(reset_state, outputs=[ | |
| chatbot, history, token_count, status_display], show_progress=True) | |
| retryBtn.click(retry, [keyTxt, systemPromptTxt, history, chatbot, token_count, top_p, temperature, | |
| use_streaming_checkbox], [chatbot, history, status_display, token_count], show_progress=True) | |
| delLastBtn.click(delete_last_conversation, [chatbot, history, token_count, use_streaming_checkbox], [ | |
| chatbot, history, token_count, status_display], show_progress=True) | |
| reduceTokenBtn.click(reduce_token_size, [keyTxt, systemPromptTxt, history, chatbot, token_count, top_p, temperature, use_streaming_checkbox], [ | |
| chatbot, history, status_display, token_count], show_progress=True) | |
| saveHistoryBtn.click(save_chat_history, [ | |
| saveFileName, systemPromptTxt, history, chatbot], None, show_progress=True) | |
| saveHistoryBtn.click(get_history_names, None, [historyFileSelectDropdown]) | |
| historyRefreshBtn.click(get_history_names, None, [ | |
| historyFileSelectDropdown]) | |
| historyReadBtn.click(load_chat_history, [historyFileSelectDropdown, systemPromptTxt, history, chatbot], [ | |
| saveFileName, systemPromptTxt, history, chatbot], show_progress=True) | |
| templateRefreshBtn.click(get_template_names, None, [ | |
| templateFileSelectDropdown]) | |
| templaeFileReadBtn.click(load_template, [templateFileSelectDropdown], [ | |
| promptTemplates, templateSelectDropdown], show_progress=True) | |
| templateApplyBtn.click(get_template_content, [ | |
| promptTemplates, templateSelectDropdown, systemPromptTxt], [systemPromptTxt], show_progress=True) | |
| server.title = "MyChatGPT" | |
| server.queue().launch(share=False) | |