# -*- coding:utf-8 -*- import os import logging import sys import gradio as gr from modules import config from modules.config import * from modules.utils import * from modules.presets import * from modules.overwrites import * from modules.chat_func import * from modules.openai_func import get_usage gr.Chatbot.postprocess = postprocess PromptHelper.compact_text_chunks = compact_text_chunks with open("assets/custom.css", "r", encoding="utf-8") as f: customCSS = f.read() with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo: user_name = gr.State("") history = gr.State([]) token_count = gr.State([]) promptTemplates = gr.State(load_template(get_template_names(plain=True)[0], mode=2)) user_api_key = gr.State(my_api_key) user_question = gr.State("") outputing = gr.State(False) topic = gr.State("未命名对话历史记录") with gr.Row(): with gr.Column(): gr.HTML(title) user_info = gr.Markdown(value="", elem_id="user_info") gr.HTML('
Duplicate Space
') status_display = gr.Markdown(get_geoip(), elem_id="status_display") # https://github.com/gradio-app/gradio/pull/3296 def create_greeting(request: gr.Request): if hasattr(request, "username") and request.username: # is not None or is not "" logging.info(f"Get User Name: {request.username}") return gr.Markdown.update(value=f"User: {request.username}"), request.username else: return gr.Markdown.update(value=f"User: default", visible=False), "" demo.load(create_greeting, inputs=None, outputs=[user_info, user_name]) with gr.Row().style(equal_height=True): with gr.Column(scale=5): with gr.Row(): chatbot = gr.Chatbot(elem_id="chuanhu_chatbot").style(height="100%") with gr.Row(): with gr.Column(scale=12): user_input = gr.Textbox( elem_id="user_input_tb", show_label=False, placeholder="在这里输入" ).style(container=False) with gr.Column(min_width=70, scale=1): submitBtn = gr.Button("发送", variant="primary") cancelBtn = gr.Button("取消", variant="secondary", visible=False) with gr.Row(): emptyBtn = gr.Button( "🧹 新的对话", ) retryBtn = gr.Button("🔄 重新生成") delFirstBtn = gr.Button("🗑️ 删除最旧对话") delLastBtn = gr.Button("🗑️ 删除最新对话") reduceTokenBtn = gr.Button("♻️ 总结对话") with gr.Column(): with gr.Column(min_width=50, scale=1): with gr.Tab(label="ChatGPT"): keyTxt = gr.Textbox( show_label=True, placeholder=f"OpenAI API-key...", value=hide_middle_chars(my_api_key), type="password", visible=not HIDE_MY_KEY, label="API-Key", ) if multi_api_key: usageTxt = gr.Markdown("多账号模式已开启,无需输入key,可直接开始对话", elem_id="usage_display") else: usageTxt = gr.Markdown("**发送消息** 或 **提交key** 以显示额度", elem_id="usage_display") model_select_dropdown = gr.Dropdown( label="选择模型", choices=MODELS, multiselect=False, value=MODELS[0] ) use_streaming_checkbox = gr.Checkbox( label="实时传输回答", value=True, visible=enable_streaming_option ) use_websearch_checkbox = gr.Checkbox(label="使用在线搜索", value=False) language_select_dropdown = gr.Dropdown( label="选择回复语言(针对搜索&索引功能)", choices=REPLY_LANGUAGES, multiselect=False, value=REPLY_LANGUAGES[0], ) index_files = gr.Files(label="上传索引文件", type="file", multiple=True) two_column = gr.Checkbox(label="双栏pdf", value=advance_docs["pdf"].get("two_column", False)) # TODO: 公式ocr # formula_ocr = gr.Checkbox(label="识别公式", value=advance_docs["pdf"].get("formula_ocr", False)) with gr.Tab(label="Prompt"): systemPromptTxt = gr.Textbox( show_label=True, placeholder=f"在这里输入System Prompt...", # label="System prompt", value=initial_prompt, lines=5, ).style(container=False) 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], ).style(container=False) 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, ).style(container=False) with gr.Tab(label="保存/加载"): with gr.Accordion(label="保存/加载对话历史记录", open=True): with gr.Column(): 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("🔄 刷新") with gr.Row(): with gr.Column(scale=6): saveFileName = gr.Textbox( show_label=True, placeholder=f"设置文件名: 默认为.json,可选为.md", label="设置保存文件名", value="对话历史记录", ).style(container=True) with gr.Column(scale=1): saveHistoryBtn = gr.Button("💾 保存对话") exportMarkdownBtn = gr.Button("📝 导出为Markdown") gr.Markdown("默认保存于history文件夹") with gr.Row(): with gr.Column(): downloadFile = gr.File(interactive=True) with gr.Tab(label="高级"): gr.Markdown("# ⚠️ 务必谨慎更改 ⚠️\n\n如果无法使用请恢复默认设置") default_btn = gr.Button("🔙 恢复默认设置") 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", ) temperature = gr.Slider( minimum=-0, maximum=2.0, value=1.0, step=0.1, interactive=True, label="Temperature", ) with gr.Accordion("网络设置", open=False, visible=False): # 优先展示自定义的api_host apihostTxt = gr.Textbox( show_label=True, placeholder=f"在这里输入API-Host...", label="API-Host", value=config.api_host or shared.API_HOST, lines=1, ) changeAPIURLBtn = gr.Button("🔄 切换API地址") proxyTxt = gr.Textbox( show_label=True, placeholder=f"在这里输入代理地址...", label="代理地址(示例:http://127.0.0.1:10809)", value="", lines=2, ) changeProxyBtn = gr.Button("🔄 设置代理地址") gr.Markdown(description) gr.HTML(footer.format(versions=versions_html()), elem_id="footer") chatgpt_predict_args = dict( fn=predict, inputs=[ user_api_key, systemPromptTxt, history, user_question, chatbot, token_count, top_p, temperature, use_streaming_checkbox, model_select_dropdown, use_websearch_checkbox, index_files, language_select_dropdown, ], outputs=[chatbot, history, status_display, token_count], show_progress=True, ) start_outputing_args = dict( fn=start_outputing, inputs=[], outputs=[submitBtn, cancelBtn], show_progress=True, ) end_outputing_args = dict( fn=end_outputing, inputs=[], outputs=[submitBtn, cancelBtn] ) reset_textbox_args = dict( fn=reset_textbox, inputs=[], outputs=[user_input] ) transfer_input_args = dict( fn=transfer_input, inputs=[user_input], outputs=[user_question, user_input, submitBtn, cancelBtn], show_progress=True ) get_usage_args = dict( fn=get_usage, inputs=[user_api_key], outputs=[usageTxt], show_progress=False ) # Chatbot cancelBtn.click(cancel_outputing, [], []) user_input.submit(**transfer_input_args).then(**chatgpt_predict_args).then(**end_outputing_args) user_input.submit(**get_usage_args) submitBtn.click(**transfer_input_args).then(**chatgpt_predict_args).then(**end_outputing_args) submitBtn.click(**get_usage_args) emptyBtn.click( reset_state, outputs=[chatbot, history, token_count, status_display], show_progress=True, ) emptyBtn.click(**reset_textbox_args) retryBtn.click(**start_outputing_args).then( retry, [ user_api_key, systemPromptTxt, history, chatbot, token_count, top_p, temperature, use_streaming_checkbox, model_select_dropdown, language_select_dropdown, ], [chatbot, history, status_display, token_count], show_progress=True, ).then(**end_outputing_args) retryBtn.click(**get_usage_args) delFirstBtn.click( delete_first_conversation, [history, token_count], [history, token_count, status_display], ) delLastBtn.click( delete_last_conversation, [chatbot, history, token_count], [chatbot, history, token_count, status_display], show_progress=True, ) reduceTokenBtn.click( reduce_token_size, [ user_api_key, systemPromptTxt, history, chatbot, token_count, top_p, temperature, gr.State(sum(token_count.value[-4:])), model_select_dropdown, language_select_dropdown, ], [chatbot, history, status_display, token_count], show_progress=True, ) reduceTokenBtn.click(**get_usage_args) two_column.change(update_doc_config, [two_column], None) # ChatGPT keyTxt.change(submit_key, keyTxt, [user_api_key, status_display]).then(**get_usage_args) keyTxt.submit(**get_usage_args) # Template 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, ) # S&L saveHistoryBtn.click( save_chat_history, [saveFileName, systemPromptTxt, history, chatbot, user_name], downloadFile, show_progress=True, ) saveHistoryBtn.click(get_history_names, [gr.State(False), user_name], [historyFileSelectDropdown]) exportMarkdownBtn.click( export_markdown, [saveFileName, systemPromptTxt, history, chatbot, user_name], downloadFile, show_progress=True, ) historyRefreshBtn.click(get_history_names, [gr.State(False), user_name], [historyFileSelectDropdown]) historyFileSelectDropdown.change( load_chat_history, [historyFileSelectDropdown, systemPromptTxt, history, chatbot, user_name], [saveFileName, systemPromptTxt, history, chatbot], show_progress=True, ) downloadFile.change( load_chat_history, [downloadFile, systemPromptTxt, history, chatbot, user_name], [saveFileName, systemPromptTxt, history, chatbot], ) # Advanced default_btn.click( reset_default, [], [apihostTxt, proxyTxt, status_display], show_progress=True ) changeAPIURLBtn.click( change_api_host, [apihostTxt], [status_display], show_progress=True, ) changeProxyBtn.click( change_proxy, [proxyTxt], [status_display], show_progress=True, ) logging.info( colorama.Back.GREEN + "\nMyRecipe的温馨提示:访问 http://localhost:7860 查看界面" + colorama.Style.RESET_ALL ) # 默认开启本地服务器,默认可以直接从IP访问,默认不创建公开分享链接 demo.title = "MyRecipe🍥" if __name__ == "__main__": reload_javascript() # if running in Docker if dockerflag: if authflag: demo.queue(concurrency_count=CONCURRENT_COUNT).launch( server_name="0.0.0.0", server_port=7860, auth=auth_list, favicon_path="./assets/favicon.ico", ) else: demo.queue(concurrency_count=CONCURRENT_COUNT).launch( server_name="0.0.0.0", server_port=7860, share=False, favicon_path="./assets/favicon.ico", ) # if not running in Docker else: if authflag: demo.queue(concurrency_count=CONCURRENT_COUNT).launch( share=False, auth=auth_list, favicon_path="./assets/favicon.ico", inbrowser=True, ) else: demo.queue(concurrency_count=CONCURRENT_COUNT).launch( share=False, favicon_path="./assets/favicon.ico", inbrowser=True ) # 改为 share=True 可以创建公开分享链接 # demo.queue(concurrency_count=CONCURRENT_COUNT).launch(server_name="0.0.0.0", server_port=7860, share=False) # 可自定义端口 # demo.queue(concurrency_count=CONCURRENT_COUNT).launch(server_name="0.0.0.0", server_port=7860,auth=("在这里填写用户名", "在这里填写密码")) # 可设置用户名与密码 # demo.queue(concurrency_count=CONCURRENT_COUNT).launch(auth=("在这里填写用户名", "在这里填写密码")) # 适合Nginx反向代理