Spaces:
Runtime error
Runtime error
| # -*- coding: utf-8 -*- | |
| """ | |
| @author:XuMing(xuming624@qq.com) | |
| @description: | |
| """ | |
| import gradio as gr | |
| from loguru import logger | |
| # import appbuilder | |
| import time | |
| import os | |
| from src.config import ( | |
| http_proxy, | |
| hide_history_when_not_logged_in, | |
| chat_name_method_index, | |
| my_api_key, | |
| multi_api_key, | |
| server_name, | |
| server_port, | |
| share, | |
| config_file, | |
| api_host, | |
| authflag, | |
| dockerflag, | |
| show_api_billing, | |
| latex_delimiters_set, | |
| user_avatar, | |
| bot_avatar, | |
| autobrowser, | |
| update_doc_config, | |
| ) | |
| from src.gradio_patch import reg_patch | |
| from src.models import get_model | |
| from src.overwrites import ( | |
| postprocess, | |
| postprocess_chat_messages, | |
| reload_javascript, | |
| get_html, | |
| ) | |
| from src.presets import ( | |
| MODEL_ALIASES, | |
| MODELS, | |
| HISTORY_NAME_METHODS, | |
| small_and_beautiful_theme, | |
| CONCURRENT_COUNT, | |
| TITLE, | |
| HIDE_MY_KEY, | |
| DEFAULT_MODEL, | |
| REPLY_LANGUAGES, | |
| INITIAL_SYSTEM_PROMPT, | |
| ENABLE_STREAMING_OPTION, | |
| DESCRIPTION, | |
| favicon_path, | |
| API_HOST, | |
| HISTORY_DIR, | |
| assets_path, | |
| ) | |
| from src.utils import ( | |
| delete_chat_history, | |
| filter_history, | |
| get_history_list, | |
| auto_name_chat_history, | |
| get_template_dropdown, | |
| rename_chat_history, | |
| init_history_list, | |
| get_first_history_name, | |
| setup_wizard, | |
| auth_from_conf, | |
| get_geoip, | |
| get_template_names, | |
| load_template, | |
| get_history_names, | |
| reset, | |
| predict, | |
| interrupt, | |
| retry, | |
| i18n, | |
| dislike, | |
| toggle_like_btn_visibility, | |
| set_key, | |
| set_single_turn, | |
| hide_middle_chars, | |
| set_system_prompt, | |
| start_outputing, | |
| set_token_upper_limit, | |
| set_temperature, | |
| set_user_identifier, | |
| set_top_p, | |
| delete_first_conversation, | |
| delete_last_conversation, | |
| set_n_choices, | |
| set_logit_bias, | |
| load_chat_history, | |
| end_outputing, | |
| set_max_tokens, | |
| reset_default, | |
| reset_textbox, | |
| set_stop_sequence, | |
| set_presence_penalty, | |
| set_frequency_penalty, | |
| upload_chat_history, | |
| export_markdown, | |
| billing_info, | |
| get_template_content, | |
| like, | |
| transfer_input, | |
| handle_file_upload, | |
| handle_summarize_index, | |
| ) | |
| reg_patch() | |
| gr.Chatbot._postprocess_chat_messages = postprocess_chat_messages | |
| gr.Chatbot.postprocess = postprocess | |
| # 设置环境变量和默认应用ID | |
| os.environ["APPBUILDER_TOKEN"] = "bce-v3/ALTAK-QwuihwYsMjA5jBiIBVfJP/51f962e086efb6f3a2332414360552bae5f3958d" | |
| APPBUILDER_APPID_DEFAULT = "2ef66c07-b4ac-4fb7-adf9-a76b5c80b2b5" | |
| APPBUILDER_APPID_CHILD = "c2de7a91-e17e-4d31-becf-b5d014156de7" | |
| APPBUILDER_APPID_STUDENT = "93ea3085-0e79-40f0-8e3d-f47381af427a" | |
| def on_mode_change(mode): | |
| return f"已选择模式：{mode}" | |
| # def generate_image_from_text(prompt, width=1024, height=1024, image_num=1): | |
| # import appbuilder | |
| # os.environ["APPBUILDER_TOKEN"] = "bce-v3/ALTAK-QwuihwYsMjA5jBiIBVfJP/51f962e086efb6f3a2332414360552bae5f3958d" | |
| # text2Image = appbuilder.Text2Image() | |
| # content_data = {"prompt": prompt, "width": width, "height": height, "image_num": image_num} | |
| # msg = appbuilder.Message(content_data) | |
| # out = text2Image.run(msg) | |
| # return out.content['img_urls'] | |
| # def generate_image(prompt): | |
| # img_urls = generate_image_from_text(prompt) | |
| # return img_urls[0] # 假设只生成一张图片 | |
| def generate_local_image(prompt): | |
| # 模拟生成图片并保存到本地路径 | |
| # 在实际应用中，你需要调用你的图像生成函数并保存图像 | |
| # 这里我们假设生成的图像保存在 `generated_image.png` | |
| local_image_path = "assets/generate_image.png" | |
| # 模拟生成图片保存 | |
| # 这里你可以替换为实际的图像生成逻辑 | |
| # from PIL import Image, ImageDraw, ImageFont | |
| # image = Image.new('RGB', (1024, 1024), color = (73, 109, 137)) | |
| # d = ImageDraw.Draw(image) | |
| # d.text((10,10), prompt, fill=(255,255,0)) | |
| # image.save(local_image_path) | |
| time.sleep(6) | |
| return local_image_path | |
| def respond(query, app_selection, chat_history): | |
| # 根据用户选择设置应用ID | |
| if app_selection == "成人模式": | |
| app_id = APPBUILDER_APPID_DEFAULT | |
| elif app_selection == "儿童模式": | |
| app_id = APPBUILDER_APPID_CHILD | |
| elif app_selection == "学生模式": | |
| app_id = APPBUILDER_APPID_STUDENT | |
| # 初始化应用 | |
| # builder = appbuilder.AppBuilderClient(app_id) | |
| # 创建会话ID | |
| # conversation_id = builder.create_conversation() | |
| # 执行对话 | |
| # msg = builder.run(conversation_id, query) | |
| # chat_history.append((query, msg.content.answer)) | |
| time.sleep(2) | |
| return "", chat_history | |
| # # 创建独立的模式切换模块 | |
| # def mode_switch_ui(): | |
| # with gr.Tab(label=i18n("切换模式")): | |
| # gr.Markdown(i18n("# 选择运行模式 ⚙️"), elem_id="mode-selection-info") | |
| # with gr.Accordion(i18n("模式切换"), open=True): | |
| # mode_selection = gr.Radio( | |
| # choices=["默认模式", "成人模式", "儿童模式", "学生模式"], | |
| # label=i18n("运行模式"), | |
| # value="默认模式" | |
| # ) | |
| # return mode_selection | |
| with gr.Blocks(theme=small_and_beautiful_theme) as demo: | |
| user_name = gr.Textbox("", visible=False) | |
| # 激活/logout路由 | |
| logout_hidden_btn = gr.LogoutButton(visible=False) | |
| promptTemplates = gr.State(load_template(get_template_names()[0], mode=2)) | |
| user_question = gr.State("") | |
| assert type(my_api_key) == str | |
| user_api_key = gr.State(my_api_key) | |
| current_model = gr.State() | |
| # mode_selection = mode_switch_ui() | |
| topic = gr.State(i18n("未命名对话历史记录")) | |
| with gr.Row(elem_id="chuanhu-header"): | |
| gr.HTML(get_html("header_title.html").format( | |
| app_title=TITLE), elem_id="app-title") | |
| status_display = gr.Markdown(get_geoip, elem_id="status-display") | |
| with gr.Row(elem_id="float-display"): | |
| user_info = gr.Markdown( | |
| value="getting user info...", elem_id="user-info") | |
| with gr.Row(equal_height=True, elem_id="chuanhu-body"): | |
| with gr.Column(elem_id="menu-area"): | |
| with gr.Column(elem_id="chuanhu-history"): | |
| with gr.Box(): | |
| with gr.Row(elem_id="chuanhu-history-header"): | |
| with gr.Row(elem_id="chuanhu-history-search-row"): | |
| with gr.Column(min_width=150, scale=2): | |
| historySearchTextbox = gr.Textbox(show_label=False, container=False, placeholder=i18n( | |
| "搜索（支持正则）..."), lines=1, elem_id="history-search-tb") | |
| with gr.Column(min_width=52, scale=1, elem_id="gr-history-header-btns"): | |
| uploadFileBtn = gr.UploadButton( | |
| interactive=True, label="", file_types=[".json"], elem_id="gr-history-upload-btn") | |
| historyRefreshBtn = gr.Button("", elem_id="gr-history-refresh-btn") | |
| with gr.Row(elem_id="chuanhu-history-body"): | |
| with gr.Column(scale=6, elem_id="history-select-wrap"): | |
| historySelectList = gr.Radio( | |
| label=i18n("从列表中加载对话"), | |
| choices=get_history_names(), | |
| value=get_first_history_name(), | |
| # multiselect=False, | |
| container=False, | |
| elem_id="history-select-dropdown" | |
| ) | |
| with gr.Row(visible=False): | |
| with gr.Column(min_width=42, scale=1): | |
| historyDeleteBtn = gr.Button( | |
| "🗑️", elem_id="gr-history-delete-btn") | |
| with gr.Column(min_width=42, scale=1): | |
| historyDownloadBtn = gr.Button( | |
| "⏬", elem_id="gr-history-download-btn") | |
| with gr.Column(min_width=42, scale=1): | |
| historyMarkdownDownloadBtn = gr.Button( | |
| "⤵️", elem_id="gr-history-mardown-download-btn") | |
| with gr.Row(visible=False): | |
| with gr.Column(scale=6): | |
| saveFileName = gr.Textbox( | |
| show_label=True, | |
| placeholder=i18n("设置文件名: 默认为.json，可选为.md"), | |
| label=i18n("设置保存文件名"), | |
| value=i18n("对话历史记录"), | |
| elem_classes="no-container" | |
| # container=False, | |
| ) | |
| with gr.Column(scale=1): | |
| renameHistoryBtn = gr.Button( | |
| i18n("💾 保存对话"), elem_id="gr-history-save-btn") | |
| exportMarkdownBtn = gr.Button( | |
| i18n("📝 导出为 Markdown"), elem_id="gr-markdown-export-btn") | |
| with gr.Column(elem_id="chuanhu-menu-footer"): | |
| with gr.Row(elem_id="chuanhu-func-nav"): | |
| gr.HTML(get_html("func_nav.html")) | |
| # gr.HTML(get_html("footer.html").format(versions=versions_html()), elem_id="footer") | |
| # gr.Markdown(CHUANHU_DESCRIPTION, elem_id="chuanhu-author") | |
| with gr.Column(elem_id="chuanhu-area", scale=5): | |
| with gr.Column(elem_id="chatbot-area"): | |
| with gr.Row(elem_id="chatbot-header"): | |
| # 获取模型的别名列表 | |
| MODEL_ALIASES_LIST = list(MODEL_ALIASES.values()) | |
| DEFAULT_MODEL_ALIAS = MODEL_ALIASES["gpt-3.5-turbo"] | |
| model_select_dropdown = gr.Dropdown( | |
| label=i18n("选择模型"), choices=MODEL_ALIASES_LIST, multiselect=False, value=DEFAULT_MODEL_ALIAS, | |
| interactive=True, | |
| show_label=False, container=False, elem_id="model-select-dropdown" | |
| ) | |
| lora_select_dropdown = gr.Dropdown( | |
| label=i18n("选择LoRA模型"), choices=[], multiselect=False, interactive=True, | |
| container=False, visible=False, | |
| ) | |
| gr.HTML(get_html("chatbot_header_btn.html").format( | |
| json_label=i18n("历史记录（JSON）"), | |
| md_label=i18n("导出为 Markdown") | |
| ), elem_id="chatbot-header-btn-bar") | |
| with gr.Row(): | |
| chatbot = gr.Chatbot( | |
| label="ChatGPT", | |
| elem_id="chuanhu-chatbot", | |
| latex_delimiters=latex_delimiters_set, | |
| sanitize_html=False, | |
| # height=700, | |
| show_label=False, | |
| avatar_images=[user_avatar, bot_avatar], | |
| show_share_button=False, | |
| ) | |
| with gr.Row(elem_id="chatbot-footer"): | |
| with gr.Box(elem_id="chatbot-input-box"): | |
| with gr.Row(elem_id="chatbot-input-row"): | |
| gr.HTML(get_html("chatbot_more.html").format( | |
| single_turn_label=i18n("单轮对话"), | |
| websearch_label=i18n("在线搜索"), | |
| upload_file_label=i18n("上传文件"), | |
| uploaded_files_label=i18n("知识库文件"), | |
| uploaded_files_tip=i18n("在工具箱中管理知识库文件") | |
| )) | |
| with gr.Row(elem_id="chatbot-input-tb-row"): | |
| with gr.Column(min_width=225, scale=12): | |
| user_input = gr.Textbox( | |
| elem_id="user-input-tb", | |
| show_label=False, | |
| placeholder=i18n("在这里输入"), | |
| elem_classes="no-container", | |
| max_lines=5, | |
| # container=False | |
| ) | |
| with gr.Column(min_width=42, scale=1, elem_id="chatbot-ctrl-btns"): | |
| submitBtn = gr.Button( | |
| value="", variant="primary", elem_id="submit-btn") | |
| cancelBtn = gr.Button( | |
| value="", variant="secondary", visible=False, elem_id="cancel-btn") | |
| # Note: Buttons below are set invisible in UI. But they are used in JS. | |
| with gr.Row(elem_id="chatbot-buttons", visible=False): | |
| with gr.Column(min_width=120, scale=1): | |
| emptyBtn = gr.Button( | |
| i18n("🧹 新的对话"), elem_id="empty-btn" | |
| ) | |
| with gr.Column(min_width=120, scale=1): | |
| retryBtn = gr.Button( | |
| i18n("🔄 重新生成"), elem_id="gr-retry-btn") | |
| with gr.Column(min_width=120, scale=1): | |
| delFirstBtn = gr.Button(i18n("🗑️ 删除最旧对话")) | |
| with gr.Column(min_width=120, scale=1): | |
| delLastBtn = gr.Button( | |
| i18n("🗑️ 删除最新对话"), elem_id="gr-dellast-btn") | |
| with gr.Row(visible=False) as like_dislike_area: | |
| with gr.Column(min_width=20, scale=1): | |
| likeBtn = gr.Button( | |
| "👍", elem_id="gr-like-btn") | |
| with gr.Column(min_width=20, scale=1): | |
| dislikeBtn = gr.Button( | |
| "👎", elem_id="gr-dislike-btn") | |
| with gr.Column(elem_id="toolbox-area", scale=1): | |
| # For CSS setting, there is an extra box. Don't remove it. | |
| with gr.Box(elem_id="chuanhu-toolbox"): | |
| with gr.Row(): | |
| gr.Markdown("## " + i18n("工具箱")) | |
| gr.HTML(get_html("close_btn.html").format( | |
| obj="toolbox"), elem_classes="close-btn") | |
| with gr.Tabs(elem_id="chuanhu-toolbox-tabs"): | |
| with gr.Tab(label=i18n("主题")): | |
| with gr.Accordion(label=i18n("模型"), open=not HIDE_MY_KEY, visible=not HIDE_MY_KEY): | |
| keyTxt = gr.Textbox( | |
| show_label=True, | |
| placeholder=f"Your API-key...", | |
| value=hide_middle_chars(user_api_key.value), | |
| type="password", | |
| visible=not HIDE_MY_KEY, | |
| label="API-Key", | |
| elem_id="api-key" | |
| ) | |
| if multi_api_key: | |
| usageTxt = gr.Markdown(i18n( | |
| "多账号模式已开启，无需输入key，可直接开始对话"), elem_id="usage-display", | |
| elem_classes="insert-block", visible=show_api_billing) | |
| else: | |
| usageTxt = gr.Markdown(i18n( | |
| "**发送消息** 或 **提交key** 以显示额度"), elem_id="usage-display", | |
| elem_classes="insert-block", visible=show_api_billing) | |
| gr.Markdown("---", elem_classes="hr-line", visible=not HIDE_MY_KEY) | |
| with gr.Accordion(label="讨论主题展示", open=True): | |
| systemPromptTxt = gr.Textbox( | |
| show_label=True, | |
| placeholder=i18n("在这里输入System Prompt..."), | |
| label="古诗词", | |
| value=INITIAL_SYSTEM_PROMPT, | |
| lines=8 | |
| ) | |
| retain_system_prompt_checkbox = gr.Checkbox( | |
| label=i18n("新建对话保留当前讨论主题"), value=False, visible=True, | |
| elem_classes="switch-checkbox") | |
| with gr.Accordion(label=i18n("加载自定义讨论主题"), open=False): | |
| with gr.Column(): | |
| with gr.Row(): | |
| with gr.Column(scale=6): | |
| templateFileSelectDropdown = gr.Dropdown( | |
| label=i18n("选择Prompt模板集合文件"), | |
| choices=get_template_names(), | |
| multiselect=False, | |
| value=get_template_names()[0], | |
| container=False, | |
| ) | |
| with gr.Column(scale=1): | |
| templateRefreshBtn = gr.Button( | |
| i18n("🔄 刷新")) | |
| with gr.Row(): | |
| with gr.Column(): | |
| templateSelectDropdown = gr.Dropdown( | |
| label=i18n("从Prompt模板中加载"), | |
| choices=load_template( | |
| get_template_names()[ | |
| 0], mode=1 | |
| ), | |
| multiselect=False, | |
| container=False, | |
| ) | |
| gr.Markdown("---", elem_classes="hr-line") | |
| with gr.Accordion(label=i18n("知识库"), open=True, elem_id="gr-kb-accordion", visible=True): | |
| use_websearch_checkbox = gr.Checkbox(label=i18n( | |
| "使用在线搜索"), value=False, elem_classes="switch-checkbox", elem_id="gr-websearch-cb", | |
| visible=False) | |
| index_files = gr.Files(label=i18n( | |
| "上传"), type="file", | |
| file_types=[".pdf", ".docx", ".pptx", ".epub", ".xlsx", ".txt", "text", "image"], | |
| elem_id="upload-index-file") | |
| two_column = gr.Checkbox(label=i18n( | |
| "双栏pdf"), value=False) | |
| summarize_btn = gr.Button(i18n("总结"), visible=False) | |
| with gr.Tab(label=i18n("模式")): # 将标题修改为“模式” | |
| gr.Markdown(i18n("# 选择运行模式 ⚙️"), | |
| elem_id="mode-selection-info") | |
| with gr.Accordion(i18n("模式切换"), open=True): | |
| mode_selection = gr.Radio(choices=["默认模式", "成人模式", "儿童模式", "学生模式"], label=i18n("运行模式"), value="默认模式") | |
| submit_button = gr.Button(i18n("确认选择")) | |
| result = gr.Textbox(label="选择结果") | |
| submit_button.click(on_mode_change, inputs=mode_selection, outputs=result) | |
| gr.Markdown("### 智能图片生成") | |
| # 添加文本输入框用于输入生成图片的文本 | |
| image_output = gr.Image(label="古诗文意象图") | |
| text_input = gr.Textbox(label="你的描述") | |
| generate_button = gr.Button("生成图片") | |
| # 绑定按钮点击事件 | |
| # generate_button.click(generate_image, inputs=text_input, outputs=image_output) | |
| # 做一个假本地返回效果 | |
| generate_button.click(generate_local_image, inputs=text_input, outputs=image_output) | |
| # with gr.Tab(label=i18n("参数")): | |
| # gr.Markdown(i18n("# ⚠️ 务必谨慎更改 ⚠️"), | |
| # elem_id="advanced-warning") | |
| # with gr.Accordion(i18n("参数"), open=True): | |
| temperature_slider = gr.Slider( | |
| minimum=-0, | |
| maximum=2.0, | |
| value=1.0, | |
| step=0.1, | |
| interactive=True, | |
| label="temperature", | |
| visible=False | |
| ) | |
| top_p_slider = gr.Slider( | |
| minimum=-0, | |
| maximum=1.0, | |
| value=1.0, | |
| step=0.05, | |
| interactive=True, | |
| label="top-p", | |
| visible=False | |
| ) | |
| n_choices_slider = gr.Slider( | |
| minimum=1, | |
| maximum=10, | |
| value=1, | |
| step=1, | |
| interactive=True, | |
| label="n choices", | |
| visible=False | |
| ) | |
| stop_sequence_txt = gr.Textbox( | |
| show_label=True, | |
| placeholder=i18n("停止符，用英文逗号隔开..."), | |
| label="stop", | |
| value="", | |
| lines=1, | |
| visible=False | |
| ) | |
| max_context_length_slider = gr.Slider( | |
| minimum=1, | |
| maximum=32768, | |
| value=2000, | |
| step=1, | |
| interactive=True, | |
| label="max context", | |
| visible=False | |
| ) | |
| max_generation_slider = gr.Slider( | |
| minimum=1, | |
| maximum=32768, | |
| value=1000, | |
| step=1, | |
| interactive=True, | |
| label="max generations", | |
| visible=False | |
| ) | |
| presence_penalty_slider = gr.Slider( | |
| minimum=-2.0, | |
| maximum=2.0, | |
| value=0.0, | |
| step=0.01, | |
| interactive=True, | |
| label="presence penalty", | |
| visible=False | |
| ) | |
| frequency_penalty_slider = gr.Slider( | |
| minimum=-2.0, | |
| maximum=2.0, | |
| value=0.0, | |
| step=0.01, | |
| interactive=True, | |
| label="frequency penalty", | |
| visible=False | |
| ) | |
| logit_bias_txt = gr.Textbox( | |
| show_label=True, | |
| placeholder=f"word:likelihood", | |
| label="logit bias", | |
| value="", | |
| lines=1, | |
| visible=False | |
| ) | |
| user_identifier_txt = gr.Textbox( | |
| show_label=True, | |
| placeholder=i18n("用于定位滥用行为"), | |
| label=i18n("用户标识符"), | |
| value=user_name.value, | |
| lines=1, | |
| visible=False | |
| ) | |
| with gr.Tab(label=i18n("关于")): | |
| gr.Markdown("#### " + i18n("PoetryChat Github地址")) | |
| gr.Markdown(DESCRIPTION) | |
| with gr.Row(elem_id="popup-wrapper"): | |
| with gr.Box(elem_id="chuanhu-popup"): | |
| with gr.Box(elem_id="chuanhu-setting"): | |
| with gr.Row(): | |
| gr.Markdown("## " + i18n("设置")) | |
| gr.HTML(get_html("close_btn.html").format( | |
| obj="box"), elem_classes="close-btn") | |
| with gr.Tabs(elem_id="chuanhu-setting-tabs"): | |
| with gr.Tab(label=i18n("高级")): | |
| gr.HTML(get_html("appearance_switcher.html").format( | |
| label=i18n("切换亮暗色主题")), elem_classes="insert-block", visible=False) | |
| use_streaming_checkbox = gr.Checkbox( | |
| label=i18n("实时传输回答"), value=True, visible=ENABLE_STREAMING_OPTION, | |
| elem_classes="switch-checkbox" | |
| ) | |
| language_select_dropdown = gr.Dropdown( | |
| label=i18n("选择回复语言（针对搜索&索引功能）"), | |
| choices=REPLY_LANGUAGES, | |
| multiselect=False, | |
| value=REPLY_LANGUAGES[0], | |
| visible=False, | |
| ) | |
| name_chat_method = gr.Dropdown( | |
| label=i18n("对话命名方式"), | |
| choices=HISTORY_NAME_METHODS, | |
| multiselect=False, | |
| interactive=True, | |
| value=HISTORY_NAME_METHODS[chat_name_method_index], | |
| ) | |
| single_turn_checkbox = gr.Checkbox(label=i18n( | |
| "单轮对话"), value=False, elem_classes="switch-checkbox", elem_id="gr-single-session-cb", | |
| visible=False) | |
| # checkUpdateBtn = gr.Button(i18n("🔄 检查更新..."), visible=check_update) | |
| logout_btn = gr.Button(i18n("退出用户"), variant="primary", visible=authflag) | |
| with gr.Tab(i18n("网络")): | |
| gr.Markdown( | |
| i18n("⚠️ 为保证API-Key安全，请在配置文件`config.json`中修改网络设置"), | |
| elem_id="netsetting-warning") | |
| default_btn = gr.Button(i18n("🔙 恢复默认网络设置")) | |
| # 网络代理 | |
| proxyTxt = gr.Textbox( | |
| show_label=True, | |
| placeholder=i18n("未设置代理..."), | |
| label=i18n("代理地址"), | |
| value=http_proxy, | |
| lines=1, | |
| interactive=False, | |
| # container=False, | |
| elem_classes="view-only-textbox no-container", | |
| ) | |
| # changeProxyBtn = gr.Button(i18n("🔄 设置代理地址")) | |
| # 优先展示自定义的api_host | |
| apihostTxt = gr.Textbox( | |
| show_label=True, | |
| placeholder="api.openai.com", | |
| label="OpenAI API-Host", | |
| value=api_host or API_HOST, | |
| lines=1, | |
| interactive=False, | |
| # container=False, | |
| elem_classes="view-only-textbox no-container", | |
| ) | |
| with gr.Tab(label=i18n("关于"), elem_id="about-tab"): | |
| gr.Markdown("# " + i18n("PoetryChat")) | |
| gr.Markdown(DESCRIPTION, elem_id="description") | |
| with gr.Box(elem_id="web-config", visible=False): | |
| gr.HTML(get_html('web_config.html').format( | |
| enableCheckUpdate_config=False, | |
| hideHistoryWhenNotLoggedIn_config=hide_history_when_not_logged_in, | |
| forView_i18n=i18n("仅供查看"), | |
| deleteConfirm_i18n_pref=i18n("你真的要删除 "), | |
| deleteConfirm_i18n_suff=i18n(" 吗？"), | |
| usingLatest_i18n=i18n("您使用的就是最新版！"), | |
| updatingMsg_i18n=i18n("正在尝试更新..."), | |
| updateSuccess_i18n=i18n("更新成功，请重启本程序"), | |
| updateFailure_i18n=i18n( | |
| "更新失败，请尝试[手动更新](https://github.com/shibing624/chatgpt-webui/"), | |
| regenerate_i18n=i18n("重新生成"), | |
| deleteRound_i18n=i18n("删除这轮问答"), | |
| renameChat_i18n=i18n("重命名该对话"), | |
| validFileName_i18n=i18n("请输入有效的文件名，不要包含以下特殊字符："), | |
| clearFileHistoryMsg_i18n=i18n("⚠️请先删除知识库中的历史文件，再尝试上传！"), | |
| dropUploadMsg_i18n=i18n("释放文件以上传"), | |
| )) | |
| with gr.Box(elem_id="fake-gradio-components", visible=False): | |
| changeSingleSessionBtn = gr.Button( | |
| visible=False, elem_classes="invisible-btn", elem_id="change-single-session-btn") | |
| changeOnlineSearchBtn = gr.Button( | |
| visible=False, elem_classes="invisible-btn", elem_id="change-online-search-btn") | |
| historySelectBtn = gr.Button( | |
| visible=False, elem_classes="invisible-btn", elem_id="history-select-btn") # Not used | |
| def create_greeting(request: gr.Request): | |
| if hasattr(request, "username") and request.username: | |
| logger.info(f"Get User Name: {request.username}") | |
| user_info, user_name = gr.Markdown.update( | |
| value=f"User: {request.username}"), request.username | |
| else: | |
| user_info, user_name = gr.Markdown.update( | |
| value=f"", visible=False), "" | |
| current_model = get_model( | |
| model_name=MODELS[DEFAULT_MODEL], access_key=my_api_key, user_name=user_name)[0] | |
| if not hide_history_when_not_logged_in or user_name: | |
| loaded_stuff = current_model.auto_load() | |
| else: | |
| loaded_stuff = [gr.update(), gr.update(), gr.Chatbot.update(label=MODELS[DEFAULT_MODEL]), | |
| current_model.single_turn, current_model.temperature, current_model.top_p, | |
| current_model.n_choices, current_model.stop_sequence, current_model.token_upper_limit, | |
| current_model.max_generation_token, current_model.presence_penalty, | |
| current_model.frequency_penalty, current_model.logit_bias, current_model.user_identifier] | |
| return user_info, user_name, current_model, toggle_like_btn_visibility( | |
| DEFAULT_MODEL), *loaded_stuff, init_history_list(user_name, prepend=current_model.history_file_path[:-5]) | |
| demo.load(create_greeting, inputs=None, outputs=[ | |
| user_info, user_name, current_model, like_dislike_area, saveFileName, systemPromptTxt, chatbot, | |
| single_turn_checkbox, temperature_slider, top_p_slider, n_choices_slider, stop_sequence_txt, | |
| max_context_length_slider, max_generation_slider, presence_penalty_slider, frequency_penalty_slider, | |
| logit_bias_txt, user_identifier_txt, historySelectList], api_name="load") | |
| chatgpt_predict_args = dict( | |
| fn=predict, | |
| inputs=[ | |
| current_model, | |
| user_question, | |
| chatbot, | |
| use_streaming_checkbox, | |
| use_websearch_checkbox, | |
| index_files, | |
| language_select_dropdown, | |
| ], | |
| outputs=[chatbot, status_display], | |
| 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=billing_info, inputs=[current_model], outputs=[ | |
| usageTxt], show_progress=False | |
| ) | |
| load_history_from_file_args = dict( | |
| fn=load_chat_history, | |
| inputs=[current_model, historySelectList], | |
| outputs=[saveFileName, systemPromptTxt, chatbot, single_turn_checkbox, temperature_slider, top_p_slider, | |
| n_choices_slider, stop_sequence_txt, max_context_length_slider, max_generation_slider, | |
| presence_penalty_slider, frequency_penalty_slider, logit_bias_txt, user_identifier_txt], | |
| ) | |
| refresh_history_args = dict( | |
| fn=get_history_list, inputs=[user_name], outputs=[historySelectList] | |
| ) | |
| auto_name_chat_history_args = dict( | |
| fn=auto_name_chat_history, | |
| inputs=[current_model, name_chat_method, user_question, chatbot, single_turn_checkbox], | |
| outputs=[historySelectList], | |
| show_progress=False, | |
| ) | |
| # Chatbot | |
| cancelBtn.click(interrupt, [current_model], []) | |
| user_input.submit( | |
| **transfer_input_args).then( | |
| **chatgpt_predict_args).then( | |
| **end_outputing_args).then( | |
| **auto_name_chat_history_args) | |
| user_input.submit(**get_usage_args) | |
| submitBtn.click(**transfer_input_args).then( | |
| **chatgpt_predict_args, api_name="predict").then( | |
| **end_outputing_args).then( | |
| **auto_name_chat_history_args) | |
| submitBtn.click(**get_usage_args) | |
| index_files.upload(handle_file_upload, [current_model, index_files, chatbot, language_select_dropdown], [ | |
| index_files, chatbot, status_display]) | |
| summarize_btn.click(handle_summarize_index, [ | |
| current_model, index_files, chatbot, language_select_dropdown], [chatbot, status_display]) | |
| emptyBtn.click( | |
| reset, | |
| inputs=[current_model, retain_system_prompt_checkbox], | |
| outputs=[chatbot, status_display, historySelectList, systemPromptTxt, single_turn_checkbox, temperature_slider, | |
| top_p_slider, n_choices_slider, stop_sequence_txt, max_context_length_slider, max_generation_slider, | |
| presence_penalty_slider, frequency_penalty_slider, logit_bias_txt, user_identifier_txt], | |
| show_progress=True, | |
| _js='(a,b)=>{return clearChatbot(a,b);}', | |
| ) | |
| retryBtn.click(**start_outputing_args).then( | |
| retry, | |
| [ | |
| current_model, | |
| chatbot, | |
| use_streaming_checkbox, | |
| use_websearch_checkbox, | |
| index_files, | |
| language_select_dropdown, | |
| ], | |
| [chatbot, status_display], | |
| show_progress=True, | |
| ).then(**end_outputing_args) | |
| retryBtn.click(**get_usage_args) | |
| delFirstBtn.click( | |
| delete_first_conversation, | |
| [current_model], | |
| [status_display], | |
| ) | |
| delLastBtn.click( | |
| delete_last_conversation, | |
| [current_model, chatbot], | |
| [chatbot, status_display], | |
| show_progress=False | |
| ) | |
| likeBtn.click( | |
| like, | |
| [current_model], | |
| [status_display], | |
| show_progress=False | |
| ) | |
| dislikeBtn.click( | |
| dislike, | |
| [current_model], | |
| [status_display], | |
| show_progress=False | |
| ) | |
| two_column.change(update_doc_config, [two_column], None) | |
| # LLM Models | |
| keyTxt.change(set_key, [current_model, keyTxt], [ | |
| user_api_key, status_display], api_name="set_key").then(**get_usage_args) | |
| keyTxt.submit(**get_usage_args) | |
| single_turn_checkbox.change( | |
| set_single_turn, [current_model, single_turn_checkbox], None, show_progress=False) | |
| model_select_dropdown.change(get_model, | |
| [model_select_dropdown, lora_select_dropdown, user_api_key, temperature_slider, | |
| top_p_slider, systemPromptTxt, user_name, current_model], [ | |
| current_model, status_display, chatbot, lora_select_dropdown, user_api_key, | |
| keyTxt], show_progress=True, api_name="get_model") | |
| model_select_dropdown.change(toggle_like_btn_visibility, [model_select_dropdown], [ | |
| like_dislike_area], show_progress=False) | |
| lora_select_dropdown.change(get_model, | |
| [model_select_dropdown, lora_select_dropdown, user_api_key, temperature_slider, | |
| top_p_slider, systemPromptTxt, user_name, current_model], | |
| [current_model, status_display, chatbot], show_progress=True) | |
| # Template | |
| systemPromptTxt.change(set_system_prompt, [ | |
| current_model, systemPromptTxt], None) | |
| templateRefreshBtn.click(get_template_dropdown, None, [ | |
| templateFileSelectDropdown]) | |
| templateFileSelectDropdown.input( | |
| load_template, | |
| [templateFileSelectDropdown], | |
| [promptTemplates, templateSelectDropdown], | |
| show_progress=True, | |
| ) | |
| templateSelectDropdown.change( | |
| get_template_content, | |
| [promptTemplates, templateSelectDropdown, systemPromptTxt], | |
| [systemPromptTxt], | |
| show_progress=True, | |
| ) | |
| # S&L | |
| renameHistoryBtn.click( | |
| rename_chat_history, | |
| [current_model, saveFileName, chatbot], | |
| [historySelectList], | |
| show_progress=True, | |
| _js='(a,b,c,d)=>{return saveChatHistory(a,b,c,d);}' | |
| ) | |
| exportMarkdownBtn.click( | |
| export_markdown, | |
| [current_model, saveFileName, chatbot], | |
| [], | |
| show_progress=True, | |
| ) | |
| historyRefreshBtn.click(**refresh_history_args) | |
| historyDeleteBtn.click(delete_chat_history, [current_model, historySelectList], | |
| [status_display, historySelectList, chatbot], | |
| _js='(a,b,c)=>{return showConfirmationDialog(a, b, c);}').then( | |
| reset, | |
| inputs=[current_model, retain_system_prompt_checkbox], | |
| outputs=[chatbot, status_display, historySelectList, systemPromptTxt], | |
| show_progress=True, | |
| _js='(a,b)=>{return clearChatbot(a,b);}', | |
| ) | |
| historySelectList.input(**load_history_from_file_args) | |
| uploadFileBtn.upload(upload_chat_history, [current_model, uploadFileBtn], [ | |
| saveFileName, systemPromptTxt, chatbot, single_turn_checkbox, temperature_slider, top_p_slider, | |
| n_choices_slider, stop_sequence_txt, max_context_length_slider, max_generation_slider, presence_penalty_slider, | |
| frequency_penalty_slider, logit_bias_txt, user_identifier_txt]).then(**refresh_history_args) | |
| historyDownloadBtn.click(None, [ | |
| user_name, historySelectList], None, _js='(a,b)=>{return downloadHistory(a,b,".json");}') | |
| historyMarkdownDownloadBtn.click(None, [ | |
| user_name, historySelectList], None, _js='(a,b)=>{return downloadHistory(a,b,".md");}') | |
| historySearchTextbox.input( | |
| filter_history, | |
| [user_name, historySearchTextbox], | |
| [historySelectList] | |
| ) | |
| # Advanced | |
| temperature_slider.input( | |
| set_temperature, [current_model, temperature_slider], None, show_progress=False) | |
| top_p_slider.input(set_top_p, [current_model, top_p_slider], None, show_progress=False) | |
| n_choices_slider.input( | |
| set_n_choices, [current_model, n_choices_slider], None, show_progress=False) | |
| stop_sequence_txt.input( | |
| set_stop_sequence, [current_model, stop_sequence_txt], None, show_progress=False) | |
| max_context_length_slider.input( | |
| set_token_upper_limit, [current_model, max_context_length_slider], None, show_progress=False) | |
| max_generation_slider.input( | |
| set_max_tokens, [current_model, max_generation_slider], None, show_progress=False) | |
| presence_penalty_slider.input( | |
| set_presence_penalty, [current_model, presence_penalty_slider], None, show_progress=False) | |
| frequency_penalty_slider.input( | |
| set_frequency_penalty, [current_model, frequency_penalty_slider], None, show_progress=False) | |
| logit_bias_txt.input( | |
| set_logit_bias, [current_model, logit_bias_txt], None, show_progress=False) | |
| user_identifier_txt.input(set_user_identifier, [ | |
| current_model, user_identifier_txt], None, show_progress=False) | |
| default_btn.click( | |
| reset_default, [], [apihostTxt, proxyTxt, status_display], show_progress=True | |
| ) | |
| # Invisible elements | |
| changeSingleSessionBtn.click( | |
| fn=lambda value: gr.Checkbox.update(value=value), | |
| inputs=[single_turn_checkbox], | |
| outputs=[single_turn_checkbox], | |
| _js='(a)=>{return bgChangeSingleSession(a);}' | |
| ) | |
| changeOnlineSearchBtn.click( | |
| fn=lambda value: gr.Checkbox.update(value=value), | |
| inputs=[use_websearch_checkbox], | |
| outputs=[use_websearch_checkbox], | |
| _js='(a)=>{return bgChangeOnlineSearch(a);}' | |
| ) | |
| historySelectBtn.click( # This is an experimental feature... Not actually used. | |
| fn=load_chat_history, | |
| inputs=[current_model, historySelectList], | |
| outputs=[saveFileName, systemPromptTxt, chatbot, single_turn_checkbox, temperature_slider, top_p_slider, | |
| n_choices_slider, stop_sequence_txt, max_context_length_slider, max_generation_slider, | |
| presence_penalty_slider, frequency_penalty_slider, logit_bias_txt, user_identifier_txt], | |
| _js='(a,b)=>{return bgSelectHistory(a,b);}' | |
| ) | |
| logout_btn.click( | |
| fn=None, | |
| inputs=[], | |
| outputs=[], | |
| _js='self.location="/logout"' | |
| ) | |
| demo.title = TITLE | |
| if __name__ == "__main__": | |
| reload_javascript() | |
| setup_wizard() | |
| demo.queue(concurrency_count=CONCURRENT_COUNT).launch( | |
| allowed_paths=[HISTORY_DIR, assets_path], | |
| server_name=server_name, | |
| server_port=server_port, | |
| share=share, | |
| blocked_paths=[config_file], | |
| auth=auth_from_conf if authflag else None, | |
| favicon_path=favicon_path, | |
| inbrowser=autobrowser and not dockerflag, | |
| ) | |