')
- 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("请输入有效的文件名，不要包含以下特殊字符："),
- ))
- with gr.Box(elem_id="fake-gradio-components", visible=False):
- updateChuanhuBtn = gr.Button(
- visible=False, elem_classes="invisible-btn", elem_id="update-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)[0]
- current_model.set_user_identifier(user_name)
- if not hide_history_when_not_logged_in or user_name:
- filename, system_prompt, chatbot = current_model.auto_load()
- else:
- system_prompt = gr.update()
- filename = gr.update()
- chatbot = gr.Chatbot.update(label=MODELS[DEFAULT_MODEL])
- return user_info, user_name, current_model, toggle_like_btn_visibility(DEFAULT_MODEL), filename, system_prompt, chatbot, init_history_list(user_name)
- demo.load(create_greeting, inputs=None, outputs=[
- user_info, user_name, current_model, like_dislike_area, saveFileName, systemPromptTxt, chatbot, historySelectList], api_name="load")
+ gr.Markdown(CHUANHU_DESCRIPTION, elem_id="description")
+ gr.HTML(FOOTER.format(versions=versions_html()), elem_id="footer")
chatgpt_predict_args = dict(
fn=predict,
inputs=[
@@ -538,59 +300,38 @@ with gr.Blocks(theme=small_and_beautiful_theme) as demo:
)
transfer_input_args = dict(
- fn=transfer_input, inputs=[user_input], outputs=[
- user_question, user_input, submitBtn, cancelBtn], show_progress=True
+ 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
+ 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, user_name],
+ inputs=[current_model, historyFileSelectDropdown, chatbot, user_name],
outputs=[saveFileName, systemPromptTxt, chatbot]
)
- 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, user_name, 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(**transfer_input_args).then(**chatgpt_predict_args).then(**end_outputing_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(**transfer_input_args).then(**chatgpt_predict_args).then(**end_outputing_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])
+ index_files.change(handle_file_upload, [current_model, index_files, chatbot], [index_files, chatbot, status_display])
emptyBtn.click(
reset,
- inputs=[current_model, retain_system_prompt_checkbox],
- outputs=[chatbot, status_display, historySelectList, systemPromptTxt],
+ inputs=[current_model],
+ outputs=[chatbot, status_display],
show_progress=True,
- _js='(a,b)=>{return clearChatbot(a,b);}',
)
+ emptyBtn.click(**reset_textbox_args)
retryBtn.click(**start_outputing_args).then(
retry,
@@ -620,43 +361,19 @@ with gr.Blocks(theme=small_and_beautiful_theme) as demo:
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.change(set_key, [current_model, keyTxt], [user_api_key, status_display]).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], [
- 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)
+ 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], [current_model, status_display, lora_select_dropdown], show_progress=True)
+ lora_select_dropdown.change(get_model, [model_select_dropdown, lora_select_dropdown, user_api_key, temperature_slider, top_p_slider, systemPromptTxt], [current_model, status_display], show_progress=True)
# Template
- systemPromptTxt.change(set_system_prompt, [
- current_model, systemPromptTxt], None)
- templateRefreshBtn.click(get_template_dropdown, None, [
- templateFileSelectDropdown])
- templateFileSelectDropdown.input(
+ systemPromptTxt.change(set_system_prompt, [current_model, systemPromptTxt], None)
+ templateRefreshBtn.click(get_template_names, None, [templateFileSelectDropdown])
+ templateFileSelectDropdown.change(
load_template,
[templateFileSelectDropdown],
[promptTemplates, templateSelectDropdown],
@@ -670,125 +387,51 @@ with gr.Blocks(theme=small_and_beautiful_theme) as demo:
)
# S&L
- renameHistoryBtn.click(
- rename_chat_history,
+ saveHistoryBtn.click(
+ save_chat_history,
[current_model, saveFileName, chatbot, user_name],
- [historySelectList],
+ downloadFile,
show_progress=True,
- _js='(a,b,c,d)=>{return saveChatHistory(a,b,c,d);}'
)
+ saveHistoryBtn.click(get_history_names, [gr.State(False), user_name], [historyFileSelectDropdown])
exportMarkdownBtn.click(
export_markdown,
[current_model, saveFileName, chatbot, user_name],
- [],
- show_progress=True,
- )
- historyRefreshBtn.click(**refresh_history_args)
- historyDeleteBtn.click(delete_chat_history, [current_model, historySelectList, user_name], [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],
+ downloadFile,
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, user_name], [
- saveFileName, systemPromptTxt, chatbot]).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)
+ historyRefreshBtn.click(get_history_names, [gr.State(False), user_name], [historyFileSelectDropdown])
+ historyFileSelectDropdown.change(**load_history_from_file_args)
+ downloadFile.change(**load_history_from_file_args)
# 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)
+ 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)
+ 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,
- [],
+ changeAPIURLBtn.click(
+ change_api_host,
+ [apihostTxt],
[status_display],
show_progress=True,
)
- 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],
- _js='(a,b)=>{return bgSelectHistory(a,b);}'
+ changeProxyBtn.click(
+ change_proxy,
+ [proxyTxt],
+ [status_display],
+ show_progress=True,
)
-
logging.info(
colorama.Back.GREEN
+ "\n川虎的温馨提示：访问 http://localhost:7860 查看界面"
@@ -800,9 +443,10 @@ demo.title = i18n("川虎Chat 🚀")
if __name__ == "__main__":
reload_javascript()
demo.queue(concurrency_count=CONCURRENT_COUNT).launch(
- allowed_paths=["history", "web_assets"],
- share=share,
- auth=auth_from_conf if authflag else None,
- favicon_path="./web_assets/favicon.ico",
- inbrowser=not dockerflag, # 禁止在docker下开启inbrowser
+ auth=auth_list if authflag else None,
+ favicon_path="./assets/favicon.ico",
+ inbrowser=not dockerflag, # 禁止在docker下开启inbrowser
)
+ # 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反向代理
diff --git a/Dockerfile b/Dockerfile
index 85d5045d5316ac160277af1e7d60afa823c0f953..335c2dba28ba8c365de9306858462a59dea25f28 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,18 +1,15 @@
-FROM python:3.9-slim-buster as builder
-RUN apt-get update \
- && apt-get install -y build-essential \
- && apt-get clean \
- && rm -rf /var/lib/apt/lists/*
+FROM python:3.9 as builder
+RUN apt-get update && apt-get install -y build-essential
COPY requirements.txt .
COPY requirements_advanced.txt .
-RUN pip install --user --no-cache-dir -r requirements.txt
-# RUN pip install --user --no-cache-dir -r requirements_advanced.txt
+RUN pip install --user -r requirements.txt
+# RUN pip install --user -r requirements_advanced.txt
-FROM python:3.9-slim-buster
-LABEL maintainer="iskoldt"
+FROM python:3.9
+MAINTAINER iskoldt
COPY --from=builder /root/.local /root/.local
ENV PATH=/root/.local/bin:$PATH
COPY . /app
WORKDIR /app
-ENV dockerrun=yes
-CMD ["python3", "-u", "ChuanhuChatbot.py","2>&1", "|", "tee", "/var/log/application.log"]
+ENV dockerrun yes
+CMD ["python3", "-u", "ChuanhuChatbot.py", "2>&1", "|", "tee", "/var/log/application.log"]
diff --git a/README.md b/README.md
index 4a50bab62991332378df0aec2b9f14a46109676b..bf82f4590c81d425816e3de0cc5dd411dd65340b 100644
--- a/README.md
+++ b/README.md
@@ -1,11 +1,12 @@
---
title: ChuanhuChatGPT
emoji: 🐯
-colorFrom: yellow
-colorTo: yellow
+colorFrom: green
+colorTo: red
sdk: gradio
-sdk_version: 3.43.2
+sdk_version: 3.25.0
app_file: ChuanhuChatbot.py
+pinned: false
license: gpl-3.0
---
diff --git a/assets/custom.css b/assets/custom.css
index 22108488886cfc8d7772214dd9b83727b3fca6a3..af5e9f2118b843b3bbd7627ed45e970c20b13bef 100644
--- a/assets/custom.css
+++ b/assets/custom.css
@@ -1,12 +1,6 @@
:root {
- --chatbot-color-light: #000000;
- --chatbot-color-dark: #FFFFFF;
- --chatbot-background-color-light: #F3F3F3;
- --chatbot-background-color-dark: #121111;
- --message-user-background-color-light: #95EC69;
- --message-user-background-color-dark: #26B561;
- --message-bot-background-color-light: #FFFFFF;
- --message-bot-background-color-dark: #2C2C2C;
+ --chatbot-color-light: #F3F3F3;
+ --chatbot-color-dark: #121111;
}
#app_title {
@@ -19,15 +13,13 @@
}
#description {
text-align: center;
- margin: 32px 0 4px 0;
+ margin:16px 0
}
-/* gradio的页脚信息 */
-footer {
- /* display: none !important; */
- margin-top: .2em !important;
- font-size: 85%;
-}
+/* 覆盖gradio的页脚信息QAQ */
+/* footer {
+ display: none !important;
+} */
#footer {
text-align: center;
}
@@ -36,7 +28,7 @@ footer {
}
#footer .versions{
font-size: 85%;
- opacity: 0.60;
+ opacity: 0.85;
}
#float_display {
@@ -78,8 +70,7 @@ footer {
}
#status_display p {
font-size: .85em;
- font-family: ui-monospace, "SF Mono", "SFMono-Regular", "Menlo", "Consolas", "Liberation Mono", "Microsoft Yahei UI", "Microsoft Yahei", monospace;
- /* Windows下中文的monospace会fallback为新宋体，实在太丑，这里折中使用微软雅黑 */
+ font-family: monospace;
color: var(--body-text-color-subdued);
}
@@ -111,7 +102,7 @@ footer {
}
.progress-bar {
background-color: var(--input-background-fill);;
- margin: .5em 0 !important;
+ margin: 0 1em;
height: 20px;
border-radius: 10px;
overflow: hidden;
@@ -144,7 +135,7 @@ footer {
display: none !important;
}
.apSlider {
- background-color: var(--neutral-200);
+ background-color: var(--block-label-background-fill);
bottom: 0;
cursor: pointer;
left: 0;
@@ -163,47 +154,13 @@ footer {
content: "🌞";
}
input:checked + .apSlider {
- background-color: var(--primary-600);
+ background-color: var(--block-label-background-fill);
}
input:checked + .apSlider::before {
transform: translateX(23px);
content:"🌚";
}
-/* Override Slider Styles (for webkit browsers like Safari and Chrome)
- * 好希望这份提案能早日实现 https://github.com/w3c/csswg-drafts/issues/4410
- * 进度滑块在各个平台还是太不统一了
- */
-input[type="range"] {
- -webkit-appearance: none;
- height: 4px;
- background: var(--input-background-fill);
- border-radius: 5px;
- background-image: linear-gradient(var(--primary-500),var(--primary-500));
- background-size: 0% 100%;
- background-repeat: no-repeat;
-}
-input[type="range"]::-webkit-slider-thumb {
- -webkit-appearance: none;
- height: 20px;
- width: 20px;
- border-radius: 50%;
- border: solid 0.5px #ddd;
- background-color: white;
- cursor: ew-resize;
- box-shadow: var(--input-shadow);
- transition: background-color .1s ease;
-}
-input[type="range"]::-webkit-slider-thumb:hover {
- background: var(--neutral-50);
-}
-input[type=range]::-webkit-slider-runnable-track {
- -webkit-appearance: none;
- box-shadow: none;
- border: none;
- background: transparent;
-}
-
#submit_btn, #cancel_btn {
height: 42px !important;
}
@@ -222,25 +179,25 @@ ol:not(.options), ul:not(.options) {
/* 亮色（默认） */
#chuanhu_chatbot {
- background-color: var(--chatbot-background-color-light) !important;
- color: var(--chatbot-color-light) !important;
+ background-color: var(--chatbot-color-light) !important;
+ color: #000000 !important;
}
[data-testid = "bot"] {
- background-color: var(--message-bot-background-color-light) !important;
+ background-color: #FFFFFF !important;
}
[data-testid = "user"] {
- background-color: var(--message-user-background-color-light) !important;
+ background-color: #95EC69 !important;
}
/* 暗色 */
.dark #chuanhu_chatbot {
- background-color: var(--chatbot-background-color-dark) !important;
- color: var(--chatbot-color-dark) !important;
+ background-color: var(--chatbot-color-dark) !important;
+ color: #FFFFFF !important;
}
.dark [data-testid = "bot"] {
- background-color: var(--message-bot-background-color-dark) !important;
+ background-color: #2C2C2C !important;
}
.dark [data-testid = "user"] {
- background-color: var(--message-user-background-color-dark) !important;
+ background-color: #26B561 !important;
}
/* 屏幕宽度大于等于500px的设备 */
@@ -262,17 +219,14 @@ ol:not(.options), ul:not(.options) {
max-height: calc(100vh - 140px - var(--line-sm)*1rem - 2*var(--block-label-margin) );
}
[data-testid = "bot"] {
- max-width: 95% !important;
+ max-width: 98% !important;
}
#app_title h1{
letter-spacing: -1px; font-size: 22px;
}
}
-#chuanhu_chatbot .wrap {
- overflow-x: hidden;
-}
/* 对话气泡 */
-.message {
+[class *= "message"] {
border-radius: var(--radius-xl) !important;
border: none;
padding: var(--spacing-xl) !important;
@@ -290,116 +244,6 @@ ol:not(.options), ul:not(.options) {
width: auto !important;
border-bottom-right-radius: 0 !important;
}
-
-.message.user p {
- white-space: pre-wrap;
-}
-.message .user-message {
- display: block;
- padding: 0 !important;
- white-space: pre-wrap;
-}
-
-.message .md-message p {
- margin-top: 0.6em !important;
- margin-bottom: 0.6em !important;
-}
-.message .md-message p:first-child { margin-top: 0 !important; }
-.message .md-message p:last-of-type { margin-bottom: 0 !important; }
-
-.message .md-message {
- display: block;
- padding: 0 !important;
-}
-.message .raw-message p {
- margin:0 !important;
-}
-.message .raw-message {
- display: block;
- padding: 0 !important;
- white-space: pre-wrap;
-}
-.raw-message.hideM, .md-message.hideM {
- display: none;
-}
-
-/* custom buttons */
-.chuanhu-btn {
- border-radius: 5px;
- /* background-color: #E6E6E6 !important; */
- color: rgba(120, 120, 120, 0.64) !important;
- padding: 4px !important;
- position: absolute;
- right: -22px;
- cursor: pointer !important;
- transition: color .2s ease, background-color .2s ease;
-}
-.chuanhu-btn:hover {
- background-color: rgba(167, 167, 167, 0.25) !important;
- color: unset !important;
-}
-.chuanhu-btn:active {
- background-color: rgba(167, 167, 167, 0.5) !important;
-}
-.chuanhu-btn:focus {
- outline: none;
-}
-.copy-bot-btn {
- /* top: 18px; */
- bottom: 0;
-}
-.toggle-md-btn {
- /* top: 0; */
- bottom: 20px;
-}
-.copy-code-btn {
- position: relative;
- float: right;
- font-size: 1em;
- cursor: pointer;
-}
-
-.message-wrap>div img{
- border-radius: 10px !important;
-}
-
-/* history message */
-.wrap>.history-message {
- padding: 10px !important;
-}
-.history-message {
- /* padding: 0 !important; */
- opacity: 80%;
- display: flex;
- flex-direction: column;
-}
-.history-message>.history-message {
- padding: 0 !important;
-}
-.history-message>.message-wrap {
- padding: 0 !important;
- margin-bottom: 16px;
-}
-.history-message>.message {
- margin-bottom: 16px;
-}
-.wrap>.history-message::after {
- content: "";
- display: block;
- height: 2px;
- background-color: var(--body-text-color-subdued);
- margin-bottom: 10px;
- margin-top: -10px;
- clear: both;
-}
-.wrap>.history-message>:last-child::after {
- content: "仅供查看";
- display: block;
- text-align: center;
- color: var(--body-text-color-subdued);
- font-size: 0.8em;
-}
-
/* 表格 */
table {
margin: 1em 0;
@@ -417,52 +261,93 @@ thead th {
padding: .5em .2em;
}
/* 行内代码 */
-.message :not(pre) code {
+code {
display: inline;
white-space: break-spaces;
- font-family: var(--font-mono);
border-radius: 6px;
margin: 0 2px 0 2px;
padding: .2em .4em .1em .4em;
background-color: rgba(175,184,193,0.2);
}
/* 代码块 */
-.message pre,
-.message pre[class*=language-] {
- color: #fff;
- overflow-x: auto;
- overflow-y: hidden;
- margin: .8em 1em 1em 0em !important;
- padding: var(--spacing-xl) 1.2em !important;
- border-radius: var(--radius-lg) !important;
-}
-.message pre code,
-.message pre code[class*=language-] {
- color: #fff;
- padding: 0;
- margin: 0;
- background-color: unset;
- text-shadow: none;
- font-family: var(--font-mono);
-}
-/* 覆盖 gradio 丑陋的复制按钮样式 */
-pre button[title="copy"] {
- border-radius: 5px;
- transition: background-color .2s ease;
-}
-pre button[title="copy"]:hover {
- background-color: #333232;
-}
-pre button .check {
- color: #fff !important;
- background: var(--neutral-950) !important;
-}
-
-/* 覆盖prism.css */
-.language-css .token.string,
-.style .token.string,
-.token.entity,
-.token.operator,
-.token.url {
- background: none !important;
-}
+pre code {
+ display: block;
+ overflow: auto;
+ white-space: pre;
+ background-color: hsla(0, 0%, 0%, 80%)!important;
+ border-radius: 10px;
+ padding: 1.4em 1.2em 0em 1.4em;
+ margin: 1.2em 2em 1.2em 0.5em;
+ color: #FFF;
+ box-shadow: 6px 6px 16px hsla(0, 0%, 0%, 0.2);
+}
+/* 代码高亮样式 */
+.highlight .hll { background-color: #49483e }
+.highlight .c { color: #75715e } /* Comment */
+.highlight .err { color: #960050; background-color: #1e0010 } /* Error */
+.highlight .k { color: #66d9ef } /* Keyword */
+.highlight .l { color: #ae81ff } /* Literal */
+.highlight .n { color: #f8f8f2 } /* Name */
+.highlight .o { color: #f92672 } /* Operator */
+.highlight .p { color: #f8f8f2 } /* Punctuation */
+.highlight .ch { color: #75715e } /* Comment.Hashbang */
+.highlight .cm { color: #75715e } /* Comment.Multiline */
+.highlight .cp { color: #75715e } /* Comment.Preproc */
+.highlight .cpf { color: #75715e } /* Comment.PreprocFile */
+.highlight .c1 { color: #75715e } /* Comment.Single */
+.highlight .cs { color: #75715e } /* Comment.Special */
+.highlight .gd { color: #f92672 } /* Generic.Deleted */
+.highlight .ge { font-style: italic } /* Generic.Emph */
+.highlight .gi { color: #a6e22e } /* Generic.Inserted */
+.highlight .gs { font-weight: bold } /* Generic.Strong */
+.highlight .gu { color: #75715e } /* Generic.Subheading */
+.highlight .kc { color: #66d9ef } /* Keyword.Constant */
+.highlight .kd { color: #66d9ef } /* Keyword.Declaration */
+.highlight .kn { color: #f92672 } /* Keyword.Namespace */
+.highlight .kp { color: #66d9ef } /* Keyword.Pseudo */
+.highlight .kr { color: #66d9ef } /* Keyword.Reserved */
+.highlight .kt { color: #66d9ef } /* Keyword.Type */
+.highlight .ld { color: #e6db74 } /* Literal.Date */
+.highlight .m { color: #ae81ff } /* Literal.Number */
+.highlight .s { color: #e6db74 } /* Literal.String */
+.highlight .na { color: #a6e22e } /* Name.Attribute */
+.highlight .nb { color: #f8f8f2 } /* Name.Builtin */
+.highlight .nc { color: #a6e22e } /* Name.Class */
+.highlight .no { color: #66d9ef } /* Name.Constant */
+.highlight .nd { color: #a6e22e } /* Name.Decorator */
+.highlight .ni { color: #f8f8f2 } /* Name.Entity */
+.highlight .ne { color: #a6e22e } /* Name.Exception */
+.highlight .nf { color: #a6e22e } /* Name.Function */
+.highlight .nl { color: #f8f8f2 } /* Name.Label */
+.highlight .nn { color: #f8f8f2 } /* Name.Namespace */
+.highlight .nx { color: #a6e22e } /* Name.Other */
+.highlight .py { color: #f8f8f2 } /* Name.Property */
+.highlight .nt { color: #f92672 } /* Name.Tag */
+.highlight .nv { color: #f8f8f2 } /* Name.Variable */
+.highlight .ow { color: #f92672 } /* Operator.Word */
+.highlight .w { color: #f8f8f2 } /* Text.Whitespace */
+.highlight .mb { color: #ae81ff } /* Literal.Number.Bin */
+.highlight .mf { color: #ae81ff } /* Literal.Number.Float */
+.highlight .mh { color: #ae81ff } /* Literal.Number.Hex */
+.highlight .mi { color: #ae81ff } /* Literal.Number.Integer */
+.highlight .mo { color: #ae81ff } /* Literal.Number.Oct */
+.highlight .sa { color: #e6db74 } /* Literal.String.Affix */
+.highlight .sb { color: #e6db74 } /* Literal.String.Backtick */
+.highlight .sc { color: #e6db74 } /* Literal.String.Char */
+.highlight .dl { color: #e6db74 } /* Literal.String.Delimiter */
+.highlight .sd { color: #e6db74 } /* Literal.String.Doc */
+.highlight .s2 { color: #e6db74 } /* Literal.String.Double */
+.highlight .se { color: #ae81ff } /* Literal.String.Escape */
+.highlight .sh { color: #e6db74 } /* Literal.String.Heredoc */
+.highlight .si { color: #e6db74 } /* Literal.String.Interpol */
+.highlight .sx { color: #e6db74 } /* Literal.String.Other */
+.highlight .sr { color: #e6db74 } /* Literal.String.Regex */
+.highlight .s1 { color: #e6db74 } /* Literal.String.Single */
+.highlight .ss { color: #e6db74 } /* Literal.String.Symbol */
+.highlight .bp { color: #f8f8f2 } /* Name.Builtin.Pseudo */
+.highlight .fm { color: #a6e22e } /* Name.Function.Magic */
+.highlight .vc { color: #f8f8f2 } /* Name.Variable.Class */
+.highlight .vg { color: #f8f8f2 } /* Name.Variable.Global */
+.highlight .vi { color: #f8f8f2 } /* Name.Variable.Instance */
+.highlight .vm { color: #f8f8f2 } /* Name.Variable.Magic */
+.highlight .il { color: #ae81ff } /* Literal.Number.Integer.Long */
diff --git a/assets/custom.js b/assets/custom.js
index f013209931218fd054979e290706f1945de76856..b8071034f3618c541e3f4169c7fc6d6593d56f44 100644
--- a/assets/custom.js
+++ b/assets/custom.js
@@ -13,48 +13,22 @@ var user_input_tb = null;
var userInfoDiv = null;
var appTitleDiv = null;
var chatbot = null;
-var chatbotWrap = null;
var apSwitch = null;
-var empty_botton = null;
-var messageBotDivs = null;
-var loginUserForm = null;
-var logginUser = null;
-
-var userLogged = false;
-var usernameGotten = false;
-var historyLoaded = false;
var ga = document.getElementsByTagName("gradio-app");
var targetNode = ga[0];
var isInIframe = (window.self !== window.top);
-var language = navigator.language.slice(0,2);
-
-var forView_i18n = {
- 'zh': "仅供查看",
- 'en': "For viewing only",
- 'ja': "閲覧専用",
- 'fr': "Pour consultation seulement",
- 'es': "Solo para visualización",
-};
// gradio 页面加载好了么??? 我能动你的元素了么??
function gradioLoaded(mutations) {
for (var i = 0; i < mutations.length; i++) {
- if (mutations[i].addedNodes.length) {
- loginUserForm = document.querySelector(".gradio-container > .main > .wrap > .panel > .form")
+ if (mutations[i].addedNodes.length) {
gradioContainer = document.querySelector(".gradio-container");
user_input_tb = document.getElementById('user_input_tb');
userInfoDiv = document.getElementById("user_info");
appTitleDiv = document.getElementById("app_title");
chatbot = document.querySelector('#chuanhu_chatbot');
- chatbotWrap = document.querySelector('#chuanhu_chatbot > .wrap');
apSwitch = document.querySelector('.apSwitch input[type="checkbox"]');
- empty_botton = document.getElementById("empty_btn")
-
- if (loginUserForm) {
- localStorage.setItem("userLogged", true);
- userLogged = true;
- }
if (gradioContainer && apSwitch) { // gradioCainter 加载出来了没?
adjustDarkMode();
@@ -63,38 +37,15 @@ function gradioLoaded(mutations) {
selectHistory();
}
if (userInfoDiv && appTitleDiv) { // userInfoDiv 和 appTitleDiv 加载出来了没?
- if (!usernameGotten) {
- getUserInfo();
- }
setTimeout(showOrHideUserInfo(), 2000);
}
if (chatbot) { // chatbot 加载出来了没?
- setChatbotHeight();
- }
- if (chatbotWrap) {
- if (!historyLoaded) {
- loadHistoryHtml();
- }
- setChatbotScroll();
- }
- if (empty_botton) {
- emptyHistory();
+ setChatbotHeight()
}
}
}
}
-function webLocale() {
- console.log("webLocale", language);
- if (forView_i18n.hasOwnProperty(language)) {
- var forView = forView_i18n[language];
- var forViewStyle = document.createElement('style');
- forViewStyle.innerHTML = '.wrap>.history-message>:last-child::after { content: "' + forView + '"!important; }';
- document.head.appendChild(forViewStyle);
- // console.log("added forViewStyle", forView);
- }
-}
-
function selectHistory() {
user_input_ta = user_input_tb.querySelector("textarea");
if (user_input_ta) {
@@ -143,34 +94,6 @@ function selectHistory() {
}
}
-var username = null;
-function getUserInfo() {
- if (usernameGotten) {
- return;
- }
- userLogged = localStorage.getItem('userLogged');
- if (userLogged) {
- username = userInfoDiv.innerText;
- if (username) {
- if (username.includes("getting user info…")) {
- setTimeout(getUserInfo, 500);
- return;
- } else if (username === " ") {
- localStorage.removeItem("username");
- localStorage.removeItem("userLogged")
- userLogged = false;
- usernameGotten = true;
- return;
- } else {
- username = username.match(/User:\s*(.*)/)[1] || username;
- localStorage.setItem("username", username);
- usernameGotten = true;
- clearHistoryHtml();
- }
- }
- }
-}
-
function toggleUserInfoVisibility(shouldHide) {
if (userInfoDiv) {
if (shouldHide) {
@@ -217,12 +140,12 @@ function showOrHideUserInfo() {
appTitleDiv.ontouchend = function () {
setTimeout(function () {
toggleUserInfoVisibility(true);
- }, 3000);
+ }, 3000);
};
userInfoDiv.ontouchend = function () {
setTimeout(function () {
toggleUserInfoVisibility(true);
- }, 3000);
+ }, 3000);
};
sendBtn.ontouchend = function () {
setTimeout(function () {
@@ -238,10 +161,10 @@ function showOrHideUserInfo() {
function toggleDarkMode(isEnabled) {
if (isEnabled) {
- document.body.classList.add("dark");
+ gradioContainer.classList.add("dark");
document.body.style.setProperty("background-color", "var(--neutral-950)", "important");
} else {
- document.body.classList.remove("dark");
+ gradioContainer.classList.remove("dark");
document.body.style.backgroundColor = "";
}
}
@@ -285,200 +208,6 @@ function setChatbotHeight() {
}
}
}
-function setChatbotScroll() {
- var scrollHeight = chatbotWrap.scrollHeight;
- chatbotWrap.scrollTo(0,scrollHeight)
-}
-var rangeInputs = null;
-var numberInputs = null;
-function setSlider() {
- rangeInputs = document.querySelectorAll('input[type="range"]');
- numberInputs = document.querySelectorAll('input[type="number"]')
- setSliderRange();
- rangeInputs.forEach(rangeInput => {
- rangeInput.addEventListener('input', setSliderRange);
- });
- numberInputs.forEach(numberInput => {
- numberInput.addEventListener('input', setSliderRange);
- })
-}
-function setSliderRange() {
- var range = document.querySelectorAll('input[type="range"]');
- range.forEach(range => {
- range.style.backgroundSize = (range.value - range.min) / (range.max - range.min) * 100 + '% 100%';
- });
-}
-
-function addChuanhuButton(botElement) {
- var rawMessage = null;
- var mdMessage = null;
- rawMessage = botElement.querySelector('.raw-message');
- mdMessage = botElement.querySelector('.md-message');
- if (!rawMessage) {
- var buttons = botElement.querySelectorAll('button.chuanhu-btn');
- for (var i = 0; i < buttons.length; i++) {
- buttons[i].parentNode.removeChild(buttons[i]);
- }
- return;
- }
- var copyButton = null;
- var toggleButton = null;
- copyButton = botElement.querySelector('button.copy-bot-btn');
- toggleButton = botElement.querySelector('button.toggle-md-btn');
- if (copyButton) copyButton.remove();
- if (toggleButton) toggleButton.remove();
-
- // Copy bot button
- var copyButton = document.createElement('button');
- copyButton.classList.add('chuanhu-btn');
- copyButton.classList.add('copy-bot-btn');
- copyButton.setAttribute('aria-label', 'Copy');
- copyButton.innerHTML = copyIcon;
- copyButton.addEventListener('click', () => {
- const textToCopy = rawMessage.innerText;
- navigator.clipboard
- .writeText(textToCopy)
- .then(() => {
- copyButton.innerHTML = copiedIcon;
- setTimeout(() => {
- copyButton.innerHTML = copyIcon;
- }, 1500);
- })
- .catch(() => {
- console.error("copy failed");
- });
- });
- botElement.appendChild(copyButton);
-
- // Toggle button
- var toggleButton = document.createElement('button');
- toggleButton.classList.add('chuanhu-btn');
- toggleButton.classList.add('toggle-md-btn');
- toggleButton.setAttribute('aria-label', 'Toggle');
- var renderMarkdown = mdMessage.classList.contains('hideM');
- toggleButton.innerHTML = renderMarkdown ? mdIcon : rawIcon;
- toggleButton.addEventListener('click', () => {
- renderMarkdown = mdMessage.classList.contains('hideM');
- if (renderMarkdown){
- renderMarkdownText(botElement);
- toggleButton.innerHTML=rawIcon;
- } else {
- removeMarkdownText(botElement);
- toggleButton.innerHTML=mdIcon;
- }
- });
- botElement.insertBefore(toggleButton, copyButton);
-}
-
-function renderMarkdownText(message) {
- var mdDiv = message.querySelector('.md-message');
- if (mdDiv) mdDiv.classList.remove('hideM');
- var rawDiv = message.querySelector('.raw-message');
- if (rawDiv) rawDiv.classList.add('hideM');
-}
-function removeMarkdownText(message) {
- var rawDiv = message.querySelector('.raw-message');
- if (rawDiv) rawDiv.classList.remove('hideM');
- var mdDiv = message.querySelector('.md-message');
- if (mdDiv) mdDiv.classList.add('hideM');
-}
-
-let timeoutId;
-let isThrottled = false;
-var mmutation
-// 监听所有元素中 bot message 的变化，为 bot 消息添加复制按钮。
-var mObserver = new MutationObserver(function (mutationsList) {
- for (mmutation of mutationsList) {
- if (mmutation.type === 'childList') {
- for (var node of mmutation.addedNodes) {
- if (node.nodeType === 1 && node.classList.contains('message') && node.getAttribute('data-testid') === 'bot') {
- saveHistoryHtml();
- document.querySelectorAll('#chuanhu_chatbot>.wrap>.message-wrap .message.bot').forEach(addChuanhuButton);
- }
- if (node.tagName === 'INPUT' && node.getAttribute('type') === 'range') {
- setSlider();
- }
- }
- for (var node of mmutation.removedNodes) {
- if (node.nodeType === 1 && node.classList.contains('message') && node.getAttribute('data-testid') === 'bot') {
- saveHistoryHtml();
- document.querySelectorAll('#chuanhu_chatbot>.wrap>.message-wrap .message.bot').forEach(addChuanhuButton);
- }
- }
- } else if (mmutation.type === 'attributes') {
- if (mmutation.target.nodeType === 1 && mmutation.target.classList.contains('message') && mmutation.target.getAttribute('data-testid') === 'bot') {
- if (isThrottled) break; // 为了防止重复不断疯狂渲染，加上等待_(:з」∠)_
- isThrottled = true;
- clearTimeout(timeoutId);
- timeoutId = setTimeout(() => {
- isThrottled = false;
- document.querySelectorAll('#chuanhu_chatbot>.wrap>.message-wrap .message.bot').forEach(addChuanhuButton);
- saveHistoryHtml();
- }, 500);
- }
- }
- }
-});
-mObserver.observe(document.documentElement, { attributes: true, childList: true, subtree: true });
-
-var loadhistorytime = 0; // for debugging
-function saveHistoryHtml() {
- var historyHtml = document.querySelector('#chuanhu_chatbot > .wrap');
- localStorage.setItem('chatHistory', historyHtml.innerHTML);
- // console.log("History Saved")
- historyLoaded = false;
-}
-function loadHistoryHtml() {
- var historyHtml = localStorage.getItem('chatHistory');
- if (!historyHtml) {
- historyLoaded = true;
- return; // no history, do nothing
- }
- userLogged = localStorage.getItem('userLogged');
- if (userLogged){
- historyLoaded = true;
- return; // logged in, do nothing
- }
- if (!historyLoaded) {
- var tempDiv = document.createElement('div');
- tempDiv.innerHTML = historyHtml;
- var buttons = tempDiv.querySelectorAll('button.chuanhu-btn');
- var gradioCopyButtons = tempDiv.querySelectorAll('button.copy_code_button');
- for (var i = 0; i < buttons.length; i++) {
- buttons[i].parentNode.removeChild(buttons[i]);
- }
- for (var i = 0; i < gradioCopyButtons.length; i++) {
- gradioCopyButtons[i].parentNode.removeChild(gradioCopyButtons[i]);
- }
- var fakeHistory = document.createElement('div');
- fakeHistory.classList.add('history-message');
- fakeHistory.innerHTML = tempDiv.innerHTML;
- webLocale();
- chatbotWrap.insertBefore(fakeHistory, chatbotWrap.firstChild);
- // var fakeHistory = document.createElement('div');
- // fakeHistory.classList.add('history-message');
- // fakeHistory.innerHTML = historyHtml;
- // chatbotWrap.insertBefore(fakeHistory, chatbotWrap.firstChild);
- historyLoaded = true;
- console.log("History Loaded");
- loadhistorytime += 1; // for debugging
- } else {
- historyLoaded = false;
- }
-}
-function clearHistoryHtml() {
- localStorage.removeItem("chatHistory");
- historyMessages = chatbotWrap.querySelector('.history-message');
- if (historyMessages) {
- chatbotWrap.removeChild(historyMessages);
- console.log("History Cleared");
- }
-}
-function emptyHistory() {
- empty_botton.addEventListener("click", function () {
- clearHistoryHtml();
- });
-}
// 监视页面内部 DOM 变动
var observer = new MutationObserver(function (mutations) {
@@ -489,14 +218,7 @@ observer.observe(targetNode, { childList: true, subtree: true });
// 监视页面变化
window.addEventListener("DOMContentLoaded", function () {
isInIframe = (window.self !== window.top);
- historyLoaded = false;
});
window.addEventListener('resize', setChatbotHeight);
window.addEventListener('scroll', setChatbotHeight);
-window.matchMedia("(prefers-color-scheme: dark)").addEventListener("change", adjustDarkMode);
-
-// button svg code
-const copyIcon = '';
-const copiedIcon = '';
-const mdIcon = '';
-const rawIcon = '';
+window.matchMedia("(prefers-color-scheme: dark)").addEventListener("change", adjustDarkMode);
\ No newline at end of file
diff --git a/assets/external-scripts.js b/assets/external-scripts.js
deleted file mode 100644
index 8d0352669045537af5698b1824dbc1dba21df478..0000000000000000000000000000000000000000
--- a/assets/external-scripts.js
+++ /dev/null
@@ -1,2 +0,0 @@
-
-// external javascript here
diff --git a/assets/html/appearance_switcher.html b/assets/html/appearance_switcher.html
deleted file mode 100644
index 9375071fbdfda7bfd622d7f7bd2dfdd0c494341b..0000000000000000000000000000000000000000
--- a/assets/html/appearance_switcher.html
+++ /dev/null
@@ -1,11 +0,0 @@
-{action_name}: {action_input}\n
' - else: - return "" - - -class ChuanhuCallbackHandler(BaseCallbackHandler): - - def __init__(self, callback) -> None: - """Initialize callback handler.""" - self.callback = callback - - def on_agent_action( - self, action: AgentAction, color: Optional[str] = None, **kwargs: Any - ) -> Any: - self.callback(get_action_description(action.log)) - - def on_tool_end( - self, - output: str, - color: Optional[str] = None, - observation_prefix: Optional[str] = None, - llm_prefix: Optional[str] = None, - **kwargs: Any, - ) -> None: - """If not the final action, print out observation.""" - # if observation_prefix is not None: - # self.callback(f"\n\n{observation_prefix}") - # self.callback(output) - # if llm_prefix is not None: - # self.callback(f"\n\n{llm_prefix}") - if observation_prefix is not None: - logging.info(observation_prefix) - self.callback(output) - if llm_prefix is not None: - logging.info(llm_prefix) - - def on_agent_finish( - self, finish: AgentFinish, color: Optional[str] = None, **kwargs: Any - ) -> None: - # self.callback(f"{finish.log}\n\n") - logging.info(finish.log) - - def on_llm_new_token(self, token: str, **kwargs: Any) -> None: - """Run on new LLM token. Only available when streaming is enabled.""" - self.callback(token) - - def on_chat_model_start(self, serialized: Dict[str, Any], messages: List[List[BaseMessage]], **kwargs: Any) -> Any: - """Run when a chat model starts running.""" - pass - - -class ModelType(Enum): - Unknown = -1 - OpenAI = 0 - ChatGLM = 1 - LLaMA = 2 - XMChat = 3 - StableLM = 4 - MOSS = 5 - YuanAI = 6 - Minimax = 7 - ChuanhuAgent = 8 - GooglePaLM = 9 - LangchainChat = 10 - Midjourney = 11 - Spark = 12 - OpenAIInstruct = 13 - Claude = 14 - Qwen = 15 - OpenAIVision = 16 - ERNIE = 17 - - @classmethod - def get_type(cls, model_name: str): - model_type = None - model_name_lower = model_name.lower() - if "gpt" in model_name_lower: - if "instruct" in model_name_lower: - model_type = ModelType.OpenAIInstruct - elif "vision" in model_name_lower: - model_type = ModelType.OpenAIVision - else: - model_type = ModelType.OpenAI - elif "chatglm" in model_name_lower: - model_type = ModelType.ChatGLM - elif "llama" in model_name_lower or "alpaca" in model_name_lower: - model_type = ModelType.LLaMA - elif "xmchat" in model_name_lower: - model_type = ModelType.XMChat - elif "stablelm" in model_name_lower: - model_type = ModelType.StableLM - elif "moss" in model_name_lower: - model_type = ModelType.MOSS - elif "yuanai" in model_name_lower: - model_type = ModelType.YuanAI - elif "minimax" in model_name_lower: - model_type = ModelType.Minimax - elif "川虎助理" in model_name_lower: - model_type = ModelType.ChuanhuAgent - elif "palm" in model_name_lower: - model_type = ModelType.GooglePaLM - elif "midjourney" in model_name_lower: - model_type = ModelType.Midjourney - elif "azure" in model_name_lower or "api" in model_name_lower: - model_type = ModelType.LangchainChat - elif "星火大模型" in model_name_lower: - model_type = ModelType.Spark - elif "claude" in model_name_lower: - model_type = ModelType.Claude - elif "qwen" in model_name_lower: - model_type = ModelType.Qwen - elif "ernie" in model_name_lower: - model_type = ModelType.ERNIE - else: - model_type = ModelType.LLaMA - return model_type - - -class BaseLLMModel: - def __init__( - self, - model_name, - system_prompt=INITIAL_SYSTEM_PROMPT, - temperature=1.0, - top_p=1.0, - n_choices=1, - stop=None, - max_generation_token=None, - presence_penalty=0, - frequency_penalty=0, - logit_bias=None, - user="", - ) -> None: - self.history = [] - self.all_token_counts = [] - if model_name in MODEL_METADATA: - self.model_name = MODEL_METADATA[model_name]["model_name"] - else: - self.model_name = model_name - self.model_type = ModelType.get_type(model_name) - try: - self.token_upper_limit = MODEL_METADATA[model_name]["token_limit"] - except KeyError: - self.token_upper_limit = DEFAULT_TOKEN_LIMIT - self.interrupted = False - self.system_prompt = system_prompt - self.api_key = None - self.need_api_key = False - self.single_turn = False - self.history_file_path = get_first_history_name(user) - - self.temperature = temperature - self.top_p = top_p - self.n_choices = n_choices - self.stop_sequence = stop - self.max_generation_token = None - self.presence_penalty = presence_penalty - self.frequency_penalty = frequency_penalty - self.logit_bias = logit_bias - self.user_identifier = user - - def get_answer_stream_iter(self): - """stream predict, need to be implemented - conversations are stored in self.history, with the most recent question, in OpenAI format - should return a generator, each time give the next word (str) in the answer - """ - logging.warning( - "stream predict not implemented, using at once predict instead") - response, _ = self.get_answer_at_once() - yield response - - def get_answer_at_once(self): - """predict at once, need to be implemented - conversations are stored in self.history, with the most recent question, in OpenAI format - Should return: - the answer (str) - total token count (int) - """ - logging.warning( - "at once predict not implemented, using stream predict instead") - response_iter = self.get_answer_stream_iter() - count = 0 - for response in response_iter: - count += 1 - return response, sum(self.all_token_counts) + count - - def billing_info(self): - """get billing infomation, inplement if needed""" - # logging.warning("billing info not implemented, using default") - return BILLING_NOT_APPLICABLE_MSG - - def count_token(self, user_input): - """get token count from input, implement if needed""" - # logging.warning("token count not implemented, using default") - return len(user_input) - - def stream_next_chatbot(self, inputs, chatbot, fake_input=None, display_append=""): - def get_return_value(): - return chatbot, status_text - - status_text = i18n("开始实时传输回答……") - if fake_input: - chatbot.append((fake_input, "")) - else: - chatbot.append((inputs, "")) - - user_token_count = self.count_token(inputs) - self.all_token_counts.append(user_token_count) - logging.debug(f"输入token计数: {user_token_count}") - - stream_iter = self.get_answer_stream_iter() - - if display_append: - display_append = '\n\n