|
|
|
import gradio as gr |
|
import os |
|
import logging |
|
import sys |
|
import argparse |
|
from utils import * |
|
from presets import * |
|
|
|
logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s") |
|
|
|
my_api_key = "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" |
|
|
|
|
|
if os.environ.get('dockerrun') == 'yes': |
|
dockerflag = True |
|
else: |
|
dockerflag = False |
|
|
|
authflag = False |
|
|
|
if dockerflag: |
|
my_api_key = os.environ.get('my_api_key') |
|
if my_api_key == "empty": |
|
logging.error("Please give a api key!") |
|
sys.exit(1) |
|
|
|
username = os.environ.get('USERNAME') |
|
password = os.environ.get('PASSWORD') |
|
if not (isinstance(username, type(None)) or isinstance(password, type(None))): |
|
authflag = True |
|
else: |
|
if not my_api_key and os.path.exists("api_key.txt") and os.path.getsize("api_key.txt"): |
|
with open("api_key.txt", "r") as f: |
|
my_api_key = f.read().strip() |
|
if os.path.exists("auth.json"): |
|
with open("auth.json", "r") as f: |
|
auth = json.load(f) |
|
username = auth["username"] |
|
password = auth["password"] |
|
if username != "" and password != "": |
|
authflag = True |
|
|
|
gr.Chatbot.postprocess = postprocess |
|
|
|
with gr.Blocks(css=customCSS,) as demo: |
|
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("未命名对话历史记录") |
|
|
|
|
|
|
|
|
|
gr.HTML(title) |
|
|
|
with gr.Row(scale=1).style(equal_height=True): |
|
|
|
with gr.Column(scale=5): |
|
with gr.Row(scale=1): |
|
chatbot = gr.Chatbot().style(height=600) |
|
with gr.Row(scale=1): |
|
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(scale=1): |
|
emptyBtn = gr.Button("🧹 新的对话",) |
|
retryBtn = gr.Button("🔄 重新生成") |
|
delLastBtn = gr.Button("🗑️ 删除一条对话") |
|
reduceTokenBtn = gr.Button("♻️ 总结对话") |
|
|
|
|
|
|
|
with gr.Column(): |
|
with gr.Column(min_width=50,scale=1): |
|
status_display = gr.Markdown("status: ready") |
|
with gr.Tab(label="ChatGPT"): |
|
keyTxt = gr.Textbox(show_label=True, placeholder=f"OpenAI API-key...",value=my_api_key, type="password", visible=not HIDE_MY_KEY, label="API-Key") |
|
model_select_dropdown = gr.Dropdown(label="选择模型", choices=MODELS, multiselect=False, value=MODELS[0]) |
|
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=2.0, value=1.0, |
|
step=0.1, interactive=True, label="Temperature",) |
|
use_streaming_checkbox = gr.Checkbox(label="实时传输回答", value=True, visible=enable_streaming_option) |
|
use_websearch_checkbox = gr.Checkbox(label="使用在线搜索", value=False) |
|
|
|
with gr.Tab(label="Prompt"): |
|
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=True): |
|
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("🔄 刷新") |
|
with gr.Row(): |
|
with gr.Column(): |
|
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.Tab(label="保存/加载"): |
|
with gr.Accordion(label="保存/加载对话历史记录", open=True): |
|
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("🔄 刷新") |
|
|
|
|
|
|
|
gr.HTML(""" |
|
<div style="text-align: center; margin-top: 20px; margin-bottom: 20px;"> |
|
""") |
|
gr.Markdown(description) |
|
|
|
|
|
user_input.submit(predict, [keyTxt, systemPromptTxt, history, user_input, chatbot, token_count, top_p, temperature, use_streaming_checkbox, model_select_dropdown, use_websearch_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, model_select_dropdown, use_websearch_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, model_select_dropdown], [chatbot, history, status_display, token_count], show_progress=True) |
|
|
|
delLastBtn.click(delete_last_conversation, [chatbot, history, token_count], [ |
|
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, model_select_dropdown], [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]) |
|
|
|
historyFileSelectDropdown.change(load_chat_history, [historyFileSelectDropdown, systemPromptTxt, history, chatbot], [saveFileName, systemPromptTxt, history, chatbot], show_progress=True) |
|
|
|
templateRefreshBtn.click(get_template_names, None, [templateFileSelectDropdown]) |
|
|
|
templateFileSelectDropdown.change(load_template, [templateFileSelectDropdown], [promptTemplates, templateSelectDropdown], show_progress=True) |
|
|
|
templateSelectDropdown.change(get_template_content, [promptTemplates, templateSelectDropdown, systemPromptTxt], [systemPromptTxt], show_progress=True) |
|
|
|
logging.info(colorama.Back.GREEN + "\n温馨提示：访问 http://localhost:7860 查看界面" + colorama.Style.RESET_ALL) |
|
|
|
demo.title = "川虎ChatGPT 🚀" |
|
|
|
if __name__ == "__main__": |
|
|
|
if dockerflag: |
|
if authflag: |
|
demo.queue().launch(server_name="0.0.0.0", server_port=7860,auth=(username, password)) |
|
else: |
|
demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False) |
|
|
|
else: |
|
if authflag: |
|
demo.queue().launch(share=False, auth=(username, password)) |
|
else: |
|
demo.queue().launch(share=False) |
|
|
|
|
|
|
|
|
