Upload ChuanhuChatbot.py

ChuanhuChatbot.py

@@ -5,288 +5,247 @@ import sys
 
 import gradio as gr
 
-from modules import config
-from modules.config import *
 from modules.utils import *
 from modules.presets import *
 from modules.overwrites import *
-from modules.models import get_model
+from modules.chat_func import *
+from modules.openai_func import get_usage
 
+logging.basicConfig(
+    level=logging.DEBUG,
+    format="%(asctime)s [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s",
+)
+
+my_api_key = "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"  # 在这里输入你的 API 密钥
+
+# if we are running in Docker
+if os.environ.get("dockerrun") == "yes":
+    dockerflag = True
+else:
+    dockerflag = False
+
+authflag = False
+auth_list = []
+
+if dockerflag:
+    my_api_key = os.environ.get("my_api_key")
+    if my_api_key == "empty":
+        logging.error("Please give a api key!")
+        sys.exit(1)
+    # auth
+    username = os.environ.get("USERNAME")
+    password = os.environ.get("PASSWORD")
+    if not (isinstance(username, type(None)) or isinstance(password, type(None))):
+        auth_list.append((os.environ.get("USERNAME"), os.environ.get("PASSWORD")))
+        authflag = True
+else:
+    if (
+        not my_api_key
+        and os.path.exists("api_key.txt")
+        and os.path.getsize("api_key.txt")
+    ):
+        with open("api_key.txt", "r") as f:
+            my_api_key = f.read().strip()
+    if os.path.exists("auth.json"):
+        authflag = True
+        with open("auth.json", "r", encoding='utf-8') as f:
+            auth = json.load(f)
+            for _ in auth:
+                if auth[_]["username"] and auth[_]["password"]:
+                    auth_list.append((auth[_]["username"], auth[_]["password"]))
+                else:
+                    logging.error("请检查auth.json文件中的用户名和密码！")
+                    sys.exit(1)
 
-gr.Chatbot._postprocess_chat_messages = postprocess_chat_messages
 gr.Chatbot.postprocess = postprocess
 PromptHelper.compact_text_chunks = compact_text_chunks
 
 with open("assets/custom.css", "r", encoding="utf-8") as f:
     customCSS = f.read()
 
-def create_new_model():
-    return get_model(model_name = MODELS[DEFAULT_MODEL], access_key = my_api_key)[0]
-
 with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
-    user_name = gr.State("")
+    history = gr.State([])
+    token_count = gr.State([])
     promptTemplates = gr.State(load_template(get_template_names(plain=True)[0], mode=2))
-    user_question = gr.State("")
     user_api_key = gr.State(my_api_key)
-    current_model = gr.State(create_new_model)
-
-    topic = gr.State(i18n("未命名对话历史记录"))
+    user_question = gr.State("")
+    outputing = gr.State(False)
+    topic = gr.State("未命名对话历史记录")
 
     with gr.Row():
-        gr.HTML(CHUANHU_TITLE, elem_id="app_title")
+        gr.HTML(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")
-
-        # https://github.com/gradio-app/gradio/pull/3296
-        def create_greeting(request: gr.Request):
-            if hasattr(request, "username") and request.username: # is not None or is not ""
-                logging.info(f"Get User Name: {request.username}")
-                return gr.Markdown.update(value=f"User: {request.username}"), request.username
-            else:
-                return gr.Markdown.update(value=f"User: default", visible=False), ""
-        demo.load(create_greeting, inputs=None, outputs=[user_info, user_name])
 
     with gr.Row().style(equal_height=True):
         with gr.Column(scale=5):
             with gr.Row():
                 chatbot = gr.Chatbot(elem_id="chuanhu_chatbot").style(height="100%")
             with gr.Row():
-                with gr.Column(min_width=225, scale=12):
+                with gr.Column(scale=12):
                     user_input = gr.Textbox(
-                        elem_id="user_input_tb",
-                        show_label=False, placeholder=i18n("在这里输入")
+                        show_label=False, placeholder="在这里输入"
                     ).style(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.Column(min_width=70, scale=1):
+                    submitBtn = gr.Button("发送", variant="primary")
+                    cancelBtn = gr.Button("取消", variant="secondary", visible=False)
             with gr.Row():
                 emptyBtn = gr.Button(
-                    i18n("🧹 新的对话"),
+                    "🧹 新的对话",
                 )
-                retryBtn = gr.Button(i18n("🔄 重新生成"))
-                delFirstBtn = gr.Button(i18n("🗑️ 删除最旧对话"))
-                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("👎"))
+                retryBtn = gr.Button("🔄 重新生成")
+                delFirstBtn = gr.Button("🗑️ 删除最旧对话")
+                delLastBtn = gr.Button("🗑️ 删除最新对话")
+                reduceTokenBtn = gr.Button("♻️ 总结对话")
 
         with gr.Column():
             with gr.Column(min_width=50, scale=1):
-                with gr.Tab(label=i18n("模型")):
+                with gr.Tab(label="ChatGPT"):
                     keyTxt = gr.Textbox(
                         show_label=True,
-                        placeholder=f"Your API-key...",
-                        value=hide_middle_chars(user_api_key.value),
+                        placeholder=f"OpenAI API-key...",
+                        value=hide_middle_chars(my_api_key),
                         type="password",
                         visible=not HIDE_MY_KEY,
                         label="API-Key",
                     )
-                    if multi_api_key:
-                        usageTxt = gr.Markdown(i18n("多账号模式已开启，无需输入key，可直接开始对话"), elem_id="usage_display", elem_classes="insert_block")
-                    else:
-                        usageTxt = gr.Markdown(i18n("**发送消息** 或 **提交key** 以显示额度"), elem_id="usage_display", elem_classes="insert_block")
+                    usageTxt = gr.Markdown("**发送消息** 或 **提交key** 以显示额度", elem_id="usage_display")
                     model_select_dropdown = gr.Dropdown(
-                        label=i18n("选择模型"), choices=MODELS, multiselect=False, value=MODELS[DEFAULT_MODEL], interactive=True
+                        label="选择模型", choices=MODELS, multiselect=False, value=MODELS[0]
                     )
-                    lora_select_dropdown = gr.Dropdown(
-                        label=i18n("选择LoRA模型"), choices=[], multiselect=False, interactive=True, visible=False
+                    use_streaming_checkbox = gr.Checkbox(
+                        label="实时传输回答", value=True, visible=enable_streaming_option
                     )
-                    with gr.Row():
-                        use_streaming_checkbox = gr.Checkbox(
-                            label=i18n("实时传输回答"), value=True, visible=ENABLE_STREAMING_OPTION
-                        )
-                        single_turn_checkbox = gr.Checkbox(label=i18n("单轮对话"), value=False)
-                        use_websearch_checkbox = gr.Checkbox(label=i18n("使用在线搜索"), value=False)
+                    use_websearch_checkbox = gr.Checkbox(label="使用在线搜索", value=False)
                     language_select_dropdown = gr.Dropdown(
-                        label=i18n("选择回复语言（针对搜索&索引功能）"),
+                        label="选择回复语言（针对搜索&索引功能）",
                         choices=REPLY_LANGUAGES,
                         multiselect=False,
                         value=REPLY_LANGUAGES[0],
                     )
-                    index_files = gr.Files(label=i18n("上传"), type="file")
-                    two_column = gr.Checkbox(label=i18n("双栏pdf"), value=advance_docs["pdf"].get("two_column", False))
-                    # TODO: 公式ocr
-                    # formula_ocr = gr.Checkbox(label=i18n("识别公式"), value=advance_docs["pdf"].get("formula_ocr", False))
+                    index_files = gr.Files(label="上传索引文件", type="file", multiple=True)
 
                 with gr.Tab(label="Prompt"):
                     systemPromptTxt = gr.Textbox(
                         show_label=True,
-                        placeholder=i18n("在这里输入System Prompt..."),
+                        placeholder=f"在这里输入System Prompt...",
                         label="System prompt",
-                        value=INITIAL_SYSTEM_PROMPT,
+                        value=initial_prompt,
                         lines=10,
                     ).style(container=False)
-                    with gr.Accordion(label=i18n("加载Prompt模板"), open=True):
+                    with gr.Accordion(label="加载Prompt模板", open=True):
                         with gr.Column():
                             with gr.Row():
                                 with gr.Column(scale=6):
                                     templateFileSelectDropdown = gr.Dropdown(
-                                        label=i18n("选择Prompt模板集合文件"),
+                                        label="选择Prompt模板集合文件",
                                         choices=get_template_names(plain=True),
                                         multiselect=False,
                                         value=get_template_names(plain=True)[0],
                                     ).style(container=False)
                                 with gr.Column(scale=1):
-                                    templateRefreshBtn = gr.Button(i18n("🔄 刷新"))
+                                    templateRefreshBtn = gr.Button("🔄 刷新")
                             with gr.Row():
                                 with gr.Column():
                                     templateSelectDropdown = gr.Dropdown(
-                                        label=i18n("从Prompt模板中加载"),
+                                        label="从Prompt模板中加载",
                                         choices=load_template(
                                             get_template_names(plain=True)[0], mode=1
                                         ),
                                         multiselect=False,
+                                        value=load_template(
+                                            get_template_names(plain=True)[0], mode=1
+                                        )[0],
                                     ).style(container=False)
 
-                with gr.Tab(label=i18n("保存/加载")):
-                    with gr.Accordion(label=i18n("保存/加载对话历史记录"), open=True):
+                with gr.Tab(label="保存/加载"):
+                    with gr.Accordion(label="保存/加载对话历史记录", open=True):
                         with gr.Column():
                             with gr.Row():
                                 with gr.Column(scale=6):
                                     historyFileSelectDropdown = gr.Dropdown(
-                                        label=i18n("从列表中加载对话"),
+                                        label="从列表中加载对话",
                                         choices=get_history_names(plain=True),
                                         multiselect=False,
                                         value=get_history_names(plain=True)[0],
                                     )
                                 with gr.Column(scale=1):
-                                    historyRefreshBtn = gr.Button(i18n("🔄 刷新"))
+                                    historyRefreshBtn = gr.Button("🔄 刷新")
                             with gr.Row():
                                 with gr.Column(scale=6):
                                     saveFileName = gr.Textbox(
                                         show_label=True,
-                                        placeholder=i18n("设置文件名: 默认为.json，可选为.md"),
-                                        label=i18n("设置保存文件名"),
-                                        value=i18n("对话历史记录"),
+                                        placeholder=f"设置文件名: 默认为.json，可选为.md",
+                                        label="设置保存文件名",
+                                        value="对话历史记录",
                                     ).style(container=True)
                                 with gr.Column(scale=1):
-                                    saveHistoryBtn = gr.Button(i18n("💾 保存对话"))
-                                    exportMarkdownBtn = gr.Button(i18n("📝 导出为Markdown"))
-                                    gr.Markdown(i18n("默认保存于history文件夹"))
+                                    saveHistoryBtn = gr.Button("💾 保存对话")
+                                    exportMarkdownBtn = gr.Button("📝 导出为Markdown")
+                                    gr.Markdown("默认保存于history文件夹")
                             with gr.Row():
                                 with gr.Column():
                                     downloadFile = gr.File(interactive=True)
 
-                with gr.Tab(label=i18n("高级")):
-                    gr.Markdown(i18n("# ⚠️ 务必谨慎更改 ⚠️\n\n如果无法使用请恢复默认设置"))
-                    gr.HTML(APPEARANCE_SWITCHER, elem_classes="insert_block")
-                    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(
+                with gr.Tab(label="高级"):
+                    gr.Markdown("# ⚠️ 务必谨慎更改 ⚠️\n\n如果无法使用请恢复默认设置")
+                    default_btn = gr.Button("🔙 恢复默认设置")
+
+                    with gr.Accordion("参数", open=False):
+                        top_p = gr.Slider(
                             minimum=-0,
                             maximum=1.0,
                             value=1.0,
                             step=0.05,
                             interactive=True,
-                            label="top-p",
+                            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,
+                        temperature = gr.Slider(
+                            minimum=-0,
                             maximum=2.0,
-                            value=0.0,
-                            step=0.01,
+                            value=1.0,
+                            step=0.1,
                             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,
+                            label="Temperature",
                         )
 
-                    with gr.Accordion(i18n("网络设置"), open=False):
-                        # 优先展示自定义的api_host
-                        apihostTxt = gr.Textbox(
+                    with gr.Accordion("网络设置", open=False):
+                        apiurlTxt = gr.Textbox(
                             show_label=True,
-                            placeholder=i18n("在这里输入API-Host..."),
-                            label="API-Host",
-                            value=config.api_host or shared.API_HOST,
-                            lines=1,
+                            placeholder=f"在这里输入API地址...",
+                            label="API地址",
+                            value="https://api.openai.com/v1/chat/completions",
+                            lines=2,
                         )
-                        changeAPIURLBtn = gr.Button(i18n("🔄 切换API地址"))
+                        changeAPIURLBtn = gr.Button("🔄 切换API地址")
                         proxyTxt = gr.Textbox(
                             show_label=True,
-                            placeholder=i18n("在这里输入代理地址..."),
-                            label=i18n("代理地址（示例：http://127.0.0.1:10809）"),
+                            placeholder=f"在这里输入代理地址...",
+                            label="代理地址（示例：http://127.0.0.1:10809）",
                             value="",
                             lines=2,
                         )
-                        changeProxyBtn = gr.Button(i18n("🔄 设置代理地址"))
-                        default_btn = gr.Button(i18n("🔙 恢复默认设置"))
+                        changeProxyBtn = gr.Button("🔄 设置代理地址")
 
-    gr.Markdown(CHUANHU_DESCRIPTION, elem_id="description")
-    gr.HTML(FOOTER.format(versions=versions_html()), elem_id="footer")
-    demo.load(refresh_ui_elements_on_load, [current_model, model_select_dropdown], [like_dislike_area], show_progress=False)
+    gr.Markdown(description)
+    gr.HTML(footer.format(versions=versions_html()), elem_id="footer")
     chatgpt_predict_args = dict(
         fn=predict,
         inputs=[
-            current_model,
+            user_api_key,
+            systemPromptTxt,
+            history,
             user_question,
             chatbot,
+            token_count,
+            top_p,
+            temperature,
             use_streaming_checkbox,
+            model_select_dropdown,
             use_websearch_checkbox,
             index_files,
             language_select_dropdown,
         ],
-        outputs=[chatbot, status_display],
+        outputs=[chatbot, history, status_display, token_count],
         show_progress=True,
     )
 
@@ -310,18 +269,12 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     )
 
     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, chatbot, user_name],
-        outputs=[saveFileName, systemPromptTxt, chatbot]
+        fn=get_usage, inputs=[user_api_key], outputs=[usageTxt], show_progress=False
     )
 
 
     # Chatbot
-    cancelBtn.click(interrupt, [current_model], [])
+    cancelBtn.click(cancel_outputing, [], [])
 
     user_input.submit(**transfer_input_args).then(**chatgpt_predict_args).then(**end_outputing_args)
     user_input.submit(**get_usage_args)
@@ -329,69 +282,69 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     submitBtn.click(**transfer_input_args).then(**chatgpt_predict_args).then(**end_outputing_args)
     submitBtn.click(**get_usage_args)
 
-    index_files.change(handle_file_upload, [current_model, index_files, chatbot], [index_files, chatbot, status_display])
-
     emptyBtn.click(
-        reset,
-        inputs=[current_model],
-        outputs=[chatbot, status_display],
+        reset_state,
+        outputs=[chatbot, history, token_count, status_display],
         show_progress=True,
     )
+    emptyBtn.click(**reset_textbox_args)
 
     retryBtn.click(**start_outputing_args).then(
         retry,
         [
-            current_model,
+            user_api_key,
+            systemPromptTxt,
+            history,
             chatbot,
+            token_count,
+            top_p,
+            temperature,
             use_streaming_checkbox,
-            use_websearch_checkbox,
-            index_files,
+            model_select_dropdown,
             language_select_dropdown,
         ],
-        [chatbot, status_display],
+        [chatbot, history, status_display, token_count],
         show_progress=True,
     ).then(**end_outputing_args)
     retryBtn.click(**get_usage_args)
 
     delFirstBtn.click(
         delete_first_conversation,
-        [current_model],
-        [status_display],
+        [history, token_count],
+        [history, token_count, 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
+        [chatbot, history, token_count],
+        [chatbot, history, token_count, status_display],
+        show_progress=True,
     )
 
-    dislikeBtn.click(
-        dislike,
-        [current_model],
-        [status_display],
-        show_progress=False
+    reduceTokenBtn.click(
+        reduce_token_size,
+        [
+            user_api_key,
+            systemPromptTxt,
+            history,
+            chatbot,
+            token_count,
+            top_p,
+            temperature,
+            gr.State(sum(token_count.value[-4:])),
+            model_select_dropdown,
+            language_select_dropdown,
+        ],
+        [chatbot, history, status_display, token_count],
+        show_progress=True,
     )
+    reduceTokenBtn.click(**get_usage_args)
 
-    two_column.change(update_doc_config, [two_column], None)
-
-    # LLM Models
-    keyTxt.change(set_key, [current_model, keyTxt], [user_api_key, status_display]).then(**get_usage_args)
+    # ChatGPT
+    keyTxt.change(submit_key, 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], [current_model, status_display, lora_select_dropdown], show_progress=True)
-    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], [current_model, status_display], show_progress=True)
 
     # Template
-    systemPromptTxt.change(set_system_prompt, [current_model, systemPromptTxt], None)
     templateRefreshBtn.click(get_template_names, None, [templateFileSelectDropdown])
     templateFileSelectDropdown.change(
         load_template,
@@ -409,39 +362,37 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     # S&L
     saveHistoryBtn.click(
         save_chat_history,
-        [current_model, saveFileName, chatbot, user_name],
+        [saveFileName, systemPromptTxt, history, chatbot],
         downloadFile,
         show_progress=True,
     )
-    saveHistoryBtn.click(get_history_names, [gr.State(False), user_name], [historyFileSelectDropdown])
+    saveHistoryBtn.click(get_history_names, None, [historyFileSelectDropdown])
     exportMarkdownBtn.click(
         export_markdown,
-        [current_model, saveFileName, chatbot, user_name],
+        [saveFileName, systemPromptTxt, history, chatbot],
         downloadFile,
         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)
+    historyRefreshBtn.click(get_history_names, None, [historyFileSelectDropdown])
+    historyFileSelectDropdown.change(
+        load_chat_history,
+        [historyFileSelectDropdown, systemPromptTxt, history, chatbot],
+        [saveFileName, systemPromptTxt, history, chatbot],
+        show_progress=True,
+    )
+    downloadFile.change(
+        load_chat_history,
+        [downloadFile, systemPromptTxt, history, chatbot],
+        [saveFileName, systemPromptTxt, history, chatbot],
+    )
 
     # 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
+        reset_default, [], [apiurlTxt, proxyTxt, status_display], show_progress=True
     )
     changeAPIURLBtn.click(
-        change_api_host,
-        [apihostTxt],
+        change_api_url,
+        [apiurlTxt],
         [status_display],
         show_progress=True,
     )
@@ -458,13 +409,22 @@ logging.info(
     + colorama.Style.RESET_ALL
 )
 # 默认开启本地服务器，默认可以直接从IP访问，默认不创建公开分享链接
-demo.title = i18n("川虎Chat 🚀")
+demo.title = "川虎ChatGPT 🚀"
 
 if __name__ == "__main__":
     reload_javascript()
-    demo.queue(concurrency_count=CONCURRENT_COUNT).launch(
-        favicon_path="./assets/favicon.ico",
-    )
-    # 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反向代理
+    # if running in Docker  demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False)
+    if dockerflag:
+        if authflag:
+            demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False)
+        else:
+            demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False)
+    # if not running in Docker
+    else:
+        if authflag:
+            demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False)
+        else:
+            demo.queue().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, 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反向代理