# -*- coding:utf-8 -*- import logging logging.basicConfig( level=logging.INFO, format="%(asctime)s [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s", ) from modules.models.models import get_model from modules.train_func import * from modules.repo import * from modules.webui import * from modules.overwrites import * from modules.presets import * from modules.utils import * from modules.config import * from modules import config import gradio as gr import colorama from modules.gradio_patch import reg_patch if not hfspaceflag: reg_patch() logging.getLogger("httpx").setLevel(logging.WARNING) gr.Chatbot._postprocess_chat_messages = postprocess_chat_messages gr.Chatbot.postprocess = postprocess # with open("web_assets/css/ChuanhuChat.css", "r", encoding="utf-8") as f: # ChuanhuChatCSS = f.read() def create_new_model(): return get_model(model_name=MODELS[DEFAULT_MODEL], access_key=my_api_key)[0] 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() topic = gr.State(i18n("未命名对话历史记录")) with gr.Row(elem_id="chuanhu-header"): gr.HTML(get_html("header_title.html").format( app_title=CHUANHU_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") update_info = gr.HTML(get_html("update.html").format( current_version=repo_tag_html(), version_time=version_time(), cancel_btn=i18n("取消"), update_btn=i18n("更新"), seenew_btn=i18n("详情"), ok_btn=i18n("好"), close_btn=i18n("关闭"), reboot_btn=i18n("立即重启"), ), visible=check_update) 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_select_dropdown = gr.Dropdown( label=i18n("选择模型"), choices=MODELS, multiselect=False, value=MODELS[DEFAULT_MODEL], interactive=True, show_label=False, container=False, elem_id="model-select-dropdown" ) lora_select_dropdown = gr.Dropdown( label=i18n("选择模型"), choices=[], multiselect=False, interactive=True, visible=False, container=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="Chuanhu Chat", elem_id="chuanhu-chatbot", latex_delimiters=latex_delimiters_set, sanitize_html=False, # height=700, show_label=False, avatar_images=[config.user_avatar, config.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="Prompt", open=True): systemPromptTxt = gr.Textbox( show_label=True, placeholder=i18n("在这里输入System Prompt..."), label="System prompt", value=INITIAL_SYSTEM_PROMPT, lines=8 ) retain_system_prompt_checkbox = gr.Checkbox( label=i18n("新建对话保留Prompt"), value=False, visible=True, elem_classes="switch-checkbox") with gr.Accordion(label=i18n("加载Prompt模板"), 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"): 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=advance_docs["pdf"].get("two_column", False)) summarize_btn = gr.Button(i18n("总结")) # TODO: 公式ocr # formula_ocr = gr.Checkbox(label=i18n("识别公式"), value=advance_docs["pdf"].get("formula_ocr", False)) 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", ) top_p_slider = gr.Slider( minimum=-0, maximum=1.0, value=1.0, step=0.05, interactive=True, label="top-p", ) n_choices_slider = gr.Slider( minimum=1, maximum=10, value=1, step=1, interactive=True, label="n choices", ) stop_sequence_txt = gr.Textbox( show_label=True, placeholder=i18n("停止符，用英文逗号隔开..."), label="stop", value="", lines=1, ) max_context_length_slider = gr.Slider( minimum=1, maximum=32768, value=2000, step=1, interactive=True, label="max context", ) max_generation_slider = gr.Slider( minimum=1, maximum=32768, value=1000, step=1, interactive=True, label="max generations", ) presence_penalty_slider = gr.Slider( minimum=-2.0, maximum=2.0, value=0.0, step=0.01, interactive=True, label="presence penalty", ) frequency_penalty_slider = gr.Slider( minimum=-2.0, maximum=2.0, value=0.0, step=0.01, interactive=True, label="frequency penalty", ) logit_bias_txt = gr.Textbox( show_label=True, placeholder=f"word:likelihood", label="logit bias", value="", lines=1, ) user_identifier_txt = gr.Textbox( show_label=True, placeholder=i18n("用于定位滥用行为"), label=i18n("用户标识符"), value=user_name.value, lines=1, ) with gr.Tab(label=i18n("拓展")): gr.Markdown( "Will be here soon...\n(We hope)\n\nAnd we hope you can help us to make more extensions!") # changeAPIURLBtn = gr.Button(i18n("🔄 切换API地址")) 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("模型")): # model_select_dropdown = gr.Dropdown( # label=i18n("选择模型"), choices=MODELS, multiselect=False, value=MODELS[DEFAULT_MODEL], interactive=True # ) # lora_select_dropdown = gr.Dropdown( # label=i18n("选择LoRA模型"), choices=[], multiselect=False, interactive=True, visible=False # ) # with gr.Row(): 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], ) 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=config.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=config.api_host or shared.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( 'Chuanhu Chat logo') gr.Markdown("# "+i18n("川虎Chat")) gr.HTML(get_html("footer.html").format( versions=versions_html()), elem_id="footer") gr.Markdown(CHUANHU_DESCRIPTION, elem_id="description") with gr.Box(elem_id="chuanhu-training"): 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-training-tabs"): with gr.Tab(label="OpenAI "+i18n("微调")): openai_train_status = gr.Markdown(label=i18n("训练状态"), value=i18n( "查看[使用介绍](https://github.com/GaiZhenbiao/ChuanhuChatGPT/wiki/使用教程#微调-gpt-35)")) with gr.Tab(label=i18n("准备数据集")): dataset_preview_json = gr.JSON( label=i18n("数据集预览")) dataset_selection = gr.Files(label=i18n("选择数据集"), file_types=[ ".xlsx", ".jsonl"], file_count="single") upload_to_openai_btn = gr.Button( i18n("上传到OpenAI"), variant="primary", interactive=False) with gr.Tab(label=i18n("训练")): openai_ft_file_id = gr.Textbox(label=i18n( "文件ID"), value="", lines=1, placeholder=i18n("上传到 OpenAI 后自动填充")) openai_ft_suffix = gr.Textbox(label=i18n( "模型名称后缀"), value="", lines=1, placeholder=i18n("可选，用于区分不同的模型")) openai_train_epoch_slider = gr.Slider(label=i18n( "训练轮数（Epochs）"), minimum=1, maximum=100, value=3, step=1, interactive=True) openai_start_train_btn = gr.Button( i18n("开始训练"), variant="primary", interactive=False) with gr.Tab(label=i18n("状态")): openai_status_refresh_btn = gr.Button(i18n("刷新状态")) openai_cancel_all_jobs_btn = gr.Button( i18n("取消所有任务")) add_to_models_btn = gr.Button( i18n("添加训练好的模型到模型列表"), interactive=False) with gr.Box(elem_id="web-config", visible=False): gr.HTML(get_html('web_config.html').format( enableCheckUpdate_config=check_update, 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/GaiZhenbiao/ChuanhuChatGPT/wiki/使用教程#手动更新)"), 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): updateChuanhuBtn = gr.Button( visible=False, elem_classes="invisible-btn", elem_id="update-chuanhu-btn") rebootChuanhuBtn = gr.Button( visible=False, elem_classes="invisible-btn", elem_id="reboot-chuanhu-btn") 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 # 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}") 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) # user_input.submit(auto_name_chat_history, [current_model, user_question, chatbot, user_name], [historySelectList], show_progress=False) 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) # submitBtn.click(auto_name_chat_history, [current_model, user_question, chatbot, user_name], [historySelectList], show_progress=False) 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) # model_select_dropdown.change( # toggle_file_type, [model_select_dropdown], [index_files], 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] ) # Train dataset_selection.upload(handle_dataset_selection, dataset_selection, [ dataset_preview_json, upload_to_openai_btn, openai_train_status]) dataset_selection.clear(handle_dataset_clear, [], [ dataset_preview_json, upload_to_openai_btn]) upload_to_openai_btn.click(upload_to_openai, [dataset_selection], [ openai_ft_file_id, openai_train_status], show_progress=True) openai_ft_file_id.change(lambda x: gr.update(interactive=True) if len( x) > 0 else gr.update(interactive=False), [openai_ft_file_id], [openai_start_train_btn]) openai_start_train_btn.click(start_training, [ openai_ft_file_id, openai_ft_suffix, openai_train_epoch_slider], [openai_train_status]) openai_status_refresh_btn.click(get_training_status, [], [ openai_train_status, add_to_models_btn]) add_to_models_btn.click(add_to_models, [], [ model_select_dropdown, openai_train_status], show_progress=True) openai_cancel_all_jobs_btn.click( cancel_all_jobs, [], [openai_train_status], show_progress=True) # 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 ) # changeAPIURLBtn.click( # change_api_host, # [apihostTxt], # [status_display], # show_progress=True, # ) # changeProxyBtn.click( # change_proxy, # [proxyTxt], # [status_display], # show_progress=True, # ) # checkUpdateBtn.click(fn=None, _js='manualCheckUpdate') # Invisible elements updateChuanhuBtn.click( update_chuanhu, [], [status_display], show_progress=True, ) rebootChuanhuBtn.click( reboot_chuanhu, [], [], show_progress=True, _js='rebootingChuanhu' ) 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"' ) # 默认开启本地服务器，默认可以直接从IP访问，默认不创建公开分享链接 demo.title = i18n("川虎Chat 🚀") if __name__ == "__main__": reload_javascript() setup_wizard() demo.queue(concurrency_count=CONCURRENT_COUNT).launch( allowed_paths=["history", "web_assets"], server_name=server_name, server_port=server_port, share=share, auth=auth_from_conf if authflag else None, favicon_path="./web_assets/favicon.ico", inbrowser=autobrowser and not dockerflag, # 禁止在docker下开启inbrowser )