# -*- coding:utf-8 -*- import logging logging.basicConfig( level=logging.INFO, format="%(asctime)s [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s", ) import colorama 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.webui import * from modules.repo import * from modules.train_func import * from modules.models.models import get_model 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.State("") promptTemplates = gr.State(load_template(get_template_names(plain=True)[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(create_new_model) topic = gr.State(i18n("未命名對話歷史記錄")) with gr.Row(): gr.HTML(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") config_info = gr.HTML(get_html("config_info.html").format(bot_avatar=config.bot_avatar, user_avatar=config.user_avatar), visible=False, elem_id="config-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("好"), ), visible=check_update) with gr.Row(equal_height=True): with gr.Column(scale=5): with gr.Row(): chatbot = gr.Chatbot(label="Chuanhu Chat", elem_id="chuanhu-chatbot", latex_delimiters=latex_delimiters_set, height=700) with gr.Row(): with gr.Column(min_width=225, scale=12): user_input = gr.Textbox( elem_id="user-input-tb", show_label=False, placeholder=i18n("在這裡輸入"), container=False ) with gr.Column(min_width=42, scale=1): submitBtn = gr.Button(value="", variant="primary", elem_id="submit-btn") cancelBtn = gr.Button(value="", variant="secondary", visible=False, elem_id="cancel-btn") with gr.Row(elem_id="chatbot-buttons"): 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("🔄 重新生成")) with gr.Column(min_width=120, scale=1): delFirstBtn = gr.Button(i18n("🗑️ 刪除最舊對話")) with gr.Column(min_width=120, scale=1): delLastBtn = gr.Button(i18n("🗑️ 刪除最新對話")) with gr.Row(visible=False) as like_dislike_area: with gr.Column(min_width=20, scale=1): likeBtn = gr.Button(i18n("👍")) with gr.Column(min_width=20, scale=1): dislikeBtn = gr.Button(i18n("👎")) with gr.Column(): with gr.Column(min_width=50, scale=1): with gr.Tab(label=i18n("模型")): 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", ) 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) model_select_dropdown = gr.Dropdown( label=i18n(""), choices=MODELS, multiselect=False, value=MODELS[DEFAULT_MODEL], interactive=True ) lora_select_dropdown = gr.Dropdown( label=i18n(""), choices=[], multiselect=False, interactive=True, visible=False ) # single_turn_checkbox = False; # use_websearch_checkbox = False; with gr.Row(): single_turn_checkbox = gr.Checkbox(label=i18n(" "), value=False, elem_classes="switch-checkbox") use_websearch_checkbox = gr.Checkbox(label=i18n(" "), value=False, elem_classes="switch-checkbox") language_select_dropdown = gr.Dropdown( label=i18n(" "), choices=REPLY_LANGUAGES, multiselect=False, value=REPLY_LANGUAGES[0], ) index_files = gr.Files(label=i18n("上傳"), type="file", 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="Prompt"): systemPromptTxt = gr.Textbox( show_label=True, placeholder=i18n("在這裡輸入System Prompt..."), label="System prompt", value=INITIAL_SYSTEM_PROMPT, lines=10 ) with gr.Accordion(label=i18n("載入Prompt範本"), open=True): with gr.Column(): with gr.Row(): with gr.Column(scale=6): templateFileSelectDropdown = gr.Dropdown( label=i18n("選擇Prompt範本集合檔"), choices=get_template_names(plain=True), multiselect=False, value=get_template_names(plain=True)[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(plain=True)[0], mode=1 ), multiselect=False, container=False, ) with gr.Tab(label=i18n("保存/載入")): with gr.Accordion(label=i18n("保存/載入對話歷史記錄"), open=True): with gr.Column(): with gr.Row(): with gr.Column(scale=6): historyFileSelectDropdown = gr.Dropdown( label=i18n("從清單中載入對話"), choices=get_history_names(plain=True), multiselect=False, container=False, ) with gr.Row(): with gr.Column(min_width=42, scale=1): historyRefreshBtn = gr.Button(i18n("🔄 刷新")) with gr.Column(min_width=42, scale=1): historyDeleteBtn = gr.Button(i18n("🗑️ 刪除")) with gr.Row(): 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): saveHistoryBtn = gr.Button(i18n("💾 保存對話")) exportMarkdownBtn = gr.Button(i18n("📝 匯出為Markdown")) gr.Markdown(i18n("默認保存於history資料夾")) with gr.Row(): with gr.Column(): downloadFile = gr.File(interactive=True) with gr.Tab(label=i18n("微調")): openai_train_status = gr.Markdown(label=i18n("訓練狀態"), value=i18n("在這裡[查看使用介紹](https://github.com/GaiZhenbiao/ChuanhuChatGPT/wiki/%E4%BD%BF%E7%94%A8%E6%95%99%E7%A8%8B#%E5%BE%AE%E8%B0%83-gpt-35)")) with gr.Tab(label=i18n("準備資料集")): dataset_preview_json = gr.JSON(label=i18n("資料集預覽"), readonly=True) 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.Tab(label=i18n("高級")): gr.HTML(get_html("appearance_switcher.html").format(label=i18n("切換亮暗色主題")), elem_classes="insert-block") use_streaming_checkbox = gr.Checkbox( label=i18n("即時傳輸回答"), value=True, visible=ENABLE_STREAMING_OPTION, elem_classes="switch-checkbox" ) checkUpdateBtn = gr.Button(i18n("🔄 檢查更新..."), visible=check_update) gr.Markdown(i18n("# ⚠️ 務必謹慎更改 ⚠️"), elem_id="advanced-warning") with gr.Accordion(i18n("參數"), open=False): 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.Accordion(i18n("網路參數"), open=False): gr.Markdown(i18n("---\n⚠️ 為保證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", ) # changeAPIURLBtn = gr.Button(i18n("🔄 切換API位址")) updateChuanhuBtn = gr.Button(visible=False, elem_classes="invisible-btn", elem_id="update-chuanhu-btn") gr.Markdown(CHUANHU_DESCRIPTION, elem_id="description") gr.HTML(get_html("footer.html").format(versions=versions_html()), elem_id="footer") # 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)[0] current_model.set_user_identifier(user_name) chatbot = gr.Chatbot.update(label=MODELS[DEFAULT_MODEL]) return user_info, user_name, current_model, toggle_like_btn_visibility(DEFAULT_MODEL), *current_model.auto_load(), get_history_names(False, user_name), chatbot demo.load(create_greeting, inputs=None, outputs=[user_info, user_name, current_model, like_dislike_area, systemPromptTxt, chatbot, historyFileSelectDropdown, chatbot], 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, historyFileSelectDropdown, user_name], outputs=[saveFileName, systemPromptTxt, chatbot] ) refresh_history_args = dict( fn=get_history_names, inputs=[gr.State(False), user_name], outputs=[historyFileSelectDropdown] ) # Chatbot cancelBtn.click(interrupt, [current_model], []) 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, api_name="predict").then(**end_outputing_args) submitBtn.click(**get_usage_args) index_files.change(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], outputs=[chatbot, status_display], show_progress=True, _js='clearChatbot', ) 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) 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, 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, status_display, chatbot], show_progress=True) # Template systemPromptTxt.change(set_system_prompt, [current_model, systemPromptTxt], None) 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, [current_model, saveFileName, chatbot, user_name], downloadFile, show_progress=True, ) saveHistoryBtn.click(get_history_names, [gr.State(False), user_name], [historyFileSelectDropdown]) exportMarkdownBtn.click( export_markdown, [current_model, saveFileName, chatbot, user_name], downloadFile, show_progress=True, ) historyRefreshBtn.click(**refresh_history_args) historyDeleteBtn.click(delete_chat_history, [current_model, historyFileSelectDropdown, user_name], [status_display, historyFileSelectDropdown, chatbot], _js='(a,b,c)=>{return showConfirmationDialog(a, b, c);}') historyFileSelectDropdown.change(**load_history_from_file_args) downloadFile.change(upload_chat_history, [current_model, downloadFile, user_name], [saveFileName, systemPromptTxt, chatbot]) # 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 max_context_length_slider.change(set_token_upper_limit, [current_model, max_context_length_slider], None) temperature_slider.change(set_temperature, [current_model, temperature_slider], None) top_p_slider.change(set_top_p, [current_model, top_p_slider], None) n_choices_slider.change(set_n_choices, [current_model, n_choices_slider], None) stop_sequence_txt.change(set_stop_sequence, [current_model, stop_sequence_txt], None) max_generation_slider.change(set_max_tokens, [current_model, max_generation_slider], None) presence_penalty_slider.change(set_presence_penalty, [current_model, presence_penalty_slider], None) frequency_penalty_slider.change(set_frequency_penalty, [current_model, frequency_penalty_slider], None) logit_bias_txt.change(set_logit_bias, [current_model, logit_bias_txt], None) user_identifier_txt.change(set_user_identifier, [current_model, user_identifier_txt], None) 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, ) logging.info( colorama.Back.GREEN + "\n川虎的溫馨提示：訪問 http://localhost:7860 查看介面" + colorama.Style.RESET_ALL ) # 默認開啟本機伺服器，默認可以直接從IP訪問，預設不創建公開分享連結 demo.title = i18n("川虎Chat 🚀") if __name__ == "__main__": reload_javascript() demo.queue(concurrency_count=CONCURRENT_COUNT).launch( blocked_paths=["config.json"], favicon_path="./web_assets/favicon.ico", )