Update app.py

app.py

@@ -96,366 +96,344 @@ def main():
             # 添加一个隐藏的 Textbox 用于存储 URL 参数
             url_params = gr.Textbox(label="URL Parameters", visible=False)
 
-            with gr_L1():
-                with gr_L2(scale=2, elem_id="gpt-chat"):
-                    chatbot = gr.Chatbot(label=f"当前模型：{LLM_MODEL}", elem_id="gpt-chatbot")
-                    if LAYOUT == "TOP-DOWN":  
-                        chatbot.style(height=CHATBOT_HEIGHT)
-                    history = gr.State([])
-                with gr_L2(scale=1, elem_id="gpt-panel"):
-                    with gr.Accordion("输入区", open=True, elem_id="input-panel") as area_input_primary:
-                        with gr.Row():
-                            txt = gr.Textbox(show_label=False, lines=2, placeholder="输入问题或API密钥，输入多个密钥时，用英文逗号间隔。支持多个OpenAI密钥共存。").style(container=False)
-                        with gr.Row():
-                            submitBtn = gr.Button("提交", elem_id="elem_submit", variant="primary")
-                        with gr.Row():
-                            resetBtn = gr.Button("重置", elem_id="elem_reset", variant="secondary"); resetBtn.style(size="sm")
-                            stopBtn = gr.Button("停止", elem_id="elem_stop", variant="secondary"); stopBtn.style(size="sm")
-                            clearBtn = gr.Button("清除", elem_id="elem_clear", variant="secondary", visible=False); clearBtn.style(size="sm")
-                        # 如果需要启用音频，可以取消下面的注释
-                        # if ENABLE_AUDIO:
-                        #     with gr.Row():
-                        #         audio_mic = gr.Audio(source="microphone", type="numpy", elem_id="elem_audio", streaming=True, show_label=False).style(container=False)
-                        with gr.Row():
-                            status = gr.Markdown(f"Tip: 按Enter提交, 按Shift+Enter换行。当前模型: {LLM_MODEL} \n {proxy_info}", elem_id="state-panel")
-
-                    with gr.Accordion("基础功能区", open=True, elem_id="basic-panel") as area_basic_fn:
-                        with gr.Row():
-                            for k in range(NUM_CUSTOM_BASIC_BTN):
-                                customize_btn = gr.Button("自定义按钮" + str(k+1), visible=False, variant="secondary", info_str=f'基础功能区: 自定义按钮')
-                                customize_btn.style(size="sm")
-                                customize_btns.update({"自定义按钮" + str(k+1): customize_btn})
-                            for k in functional:
-                                if ("Visible" in functional[k]) and (not functional[k]["Visible"]): continue
-                                variant = functional[k]["Color"] if "Color" in functional[k] else "secondary"
-                                functional[k]["Button"] = gr.Button(k, variant=variant, info_str=f'基础功能区: {k}')
-                                functional[k]["Button"].style(size="sm")
-                                predefined_btns.update({k: functional[k]["Button"]})
-                    with gr.Accordion("函数插件区", open=True, elem_id="plugin-panel") as area_crazy_fn:
-                        with gr.Row():
-                            gr.Markdown("插件可读取“输入区”文本/路径作为参数（上传文件自动修正路径）")
-                        with gr.Row(elem_id="input-plugin-group"):
-                            plugin_group_sel = gr.Dropdown(choices=all_plugin_groups, label='', show_label=False, value=DEFAULT_FN_GROUPS,
-                                                          multiselect=True, interactive=True, elem_classes='normal_mut_select').style(container=False)
+                    with gr_L1():
+            with gr_L2(scale=2, elem_id="gpt-chat"):
+                chatbot = gr.Chatbot(label=f"当前模型：{LLM_MODEL}", elem_id="gpt-chatbot")
+                if LAYOUT == "TOP-DOWN":  chatbot.style(height=CHATBOT_HEIGHT)
+                history = gr.State([])
+            with gr_L2(scale=1, elem_id="gpt-panel"):
+                with gr.Accordion("输入区", open=True, elem_id="input-panel") as area_input_primary:
+                    with gr.Row():
+                        txt = gr.Textbox(show_label=False, lines=2, placeholder="输入问题或API密钥，输入多个密钥时，用英文逗号间隔。支持多个OpenAI密钥共存。").style(container=False)
+                    with gr.Row():
+                        submitBtn = gr.Button("提交", elem_id="elem_submit", variant="primary")
+                    with gr.Row():
+                        resetBtn = gr.Button("重置", elem_id="elem_reset", variant="secondary"); resetBtn.style(size="sm")
+                        stopBtn = gr.Button("停止", elem_id="elem_stop", variant="secondary"); stopBtn.style(size="sm")
+                        clearBtn = gr.Button("清除", elem_id="elem_clear", variant="secondary", visible=False); clearBtn.style(size="sm")
+                    if ENABLE_AUDIO:
                         with gr.Row():
+                            audio_mic = gr.Audio(source="microphone", type="numpy", elem_id="elem_audio", streaming=True, show_label=False).style(container=False)
+                    with gr.Row():
+                        status = gr.Markdown(f"Tip: 按Enter提交, 按Shift+Enter换行。当前模型: {LLM_MODEL} \n {proxy_info}", elem_id="state-panel")
+
+                with gr.Accordion("基础功能区", open=True, elem_id="basic-panel") as area_basic_fn:
+                    with gr.Row():
+                        for k in range(NUM_CUSTOM_BASIC_BTN):
+                            customize_btn = gr.Button("自定义按钮" + str(k+1), visible=False, variant="secondary", info_str=f'基础功能区: 自定义按钮')
+                            customize_btn.style(size="sm")
+                            customize_btns.update({"自定义按钮" + str(k+1): customize_btn})
+                        for k in functional:
+                            if ("Visible" in functional[k]) and (not functional[k]["Visible"]): continue
+                            variant = functional[k]["Color"] if "Color" in functional[k] else "secondary"
+                            functional[k]["Button"] = gr.Button(k, variant=variant, info_str=f'基础功能区: {k}')
+                            functional[k]["Button"].style(size="sm")
+                            predefined_btns.update({k: functional[k]["Button"]})
+                with gr.Accordion("函数插件区", open=True, elem_id="plugin-panel") as area_crazy_fn:
+                    with gr.Row():
+                        gr.Markdown("插件可读取“输入区”文本/路径作为参数（上传文件自动修正路径）")
+                    with gr.Row(elem_id="input-plugin-group"):
+                        plugin_group_sel = gr.Dropdown(choices=all_plugin_groups, label='', show_label=False, value=DEFAULT_FN_GROUPS,
+                                                      multiselect=True, interactive=True, elem_classes='normal_mut_select').style(container=False)
+                    with gr.Row():
+                        for k, plugin in plugins.items():
+                            if not plugin.get("AsButton", True): continue
+                            visible = True if match_group(plugin['Group'], DEFAULT_FN_GROUPS) else False
+                            variant = plugins[k]["Color"] if "Color" in plugin else "secondary"
+                            info = plugins[k].get("Info", k)
+                            plugin['Button'] = plugins[k]['Button'] = gr.Button(k, variant=variant,
+                                visible=visible, info_str=f'函数插件区: {info}').style(size="sm")
+                    with gr.Row():
+                        with gr.Accordion("更多函数插件", open=True):
+                            dropdown_fn_list = []
                             for k, plugin in plugins.items():
-                                if not plugin.get("AsButton", True): continue
-                                visible = True if match_group(plugin['Group'], DEFAULT_FN_GROUPS) else False
-                                variant = plugins[k]["Color"] if "Color" in plugin else "secondary"
-                                info = plugins[k].get("Info", k)
-                                plugin['Button'] = plugins[k]['Button'] = gr.Button(k, variant=variant,
-                                    visible=visible, info_str=f'函数插件区: {info}').style(size="sm")
-                        with gr.Row():
-                            with gr.Accordion("更多函数插件", open=True):
-                                dropdown_fn_list = []
-                                for k, plugin in plugins.items():
-                                    if not match_group(plugin['Group'], DEFAULT_FN_GROUPS): continue
-                                    if not plugin.get("AsButton", True): dropdown_fn_list.append(k)     # 排除已经是按钮的插件
-                                    elif plugin.get('AdvancedArgs', False): dropdown_fn_list.append(k)  # 对于需要高级参数的插件，亦在下拉菜单中显示
-                                with gr.Row():
-                                    dropdown = gr.Dropdown(dropdown_fn_list, value=r"打开插件列表", label="", show_label=False).style(container=False)
-                                with gr.Row():
-                                    plugin_advanced_arg = gr.Textbox(show_label=True, label="高级参数输入区", visible=False,
-                                                                     placeholder="这里是特殊函数插件的高级参数输入区").style(container=False)
-                                with gr.Row():
-                                    switchy_bt = gr.Button(r"请先从插件列表中选择", variant="secondary").style(size="sm")
-                        with gr.Row():
-                            with gr.Accordion("点击展开“文件下载区”。", open=False) as area_file_up:
-                                file_upload = gr.Files(label="任何文件, 推荐上传压缩文件(zip, tar)", file_count="multiple", elem_id="elem_upload")
-
-            with gr.Floating(init_x="0%", init_y="0%", visible=True, width=None, drag="forbidden", elem_id="tooltip"):
-                with gr.Row():
-                    with gr.Tab("上传文件", elem_id="interact-panel"):
-                        gr.Markdown("请上传本地文件/压缩包供“函数插件区”功能调用。请注意: 上传文件后会自动把输入区修改为相应路径。")
-                        file_upload_2 = gr.Files(label="任何文件, 推荐上传压缩文件(zip, tar)", file_count="multiple", elem_id="elem_upload_float")
-
-                    with gr.Tab("更换模型", elem_id="interact-panel"):
-                        md_dropdown = gr.Dropdown(AVAIL_LLM_MODELS, value=LLM_MODEL, label="更换LLM模型/请求源").style(container=False)
-                        top_p = gr.Slider(minimum=0, maximum=1.0, value=1.0, step=0.01,interactive=True, label="Top-p (nucleus sampling)",)
-                        temperature = gr.Slider(minimum=0, maximum=2.0, value=1.0, step=0.01, interactive=True, label="Temperature",)
-                        max_length_sl = gr.Slider(minimum=256, maximum=1024*32, value=4096, step=128, interactive=True, label="Local LLM MaxLength",)
-                        system_prompt = gr.Textbox(show_label=True, lines=2, placeholder=f"System Prompt", label="System prompt", value=INIT_SYS_PROMPT)
-
-                    with gr.Tab("界面外观", elem_id="interact-panel"):
-                        theme_dropdown = gr.Dropdown(AVAIL_THEMES, value=THEME, label="更换UI主题").style(container=False)
-                        checkboxes = gr.CheckboxGroup(["基础功能区", "函数插件区", "浮动输入区", "输入清除键", "插件参数区"], value=["基础功能区", "函数插件区"], label="显示/隐藏功能区", elem_id='cbs').style(container=False)
-                        opt = ["自定义菜单"]
-                        value=[]
-                        if ADD_WAIFU: opt += ["添加Live2D形象"]; value += ["添加Live2D形象"]
-                        checkboxes_2 = gr.CheckboxGroup(opt, value=value, label="显示/隐藏自定义菜单", elem_id='cbsc').style(container=False)
-                        dark_mode_btn = gr.Button("切换界面明暗 ☀", variant="secondary").style(size="sm")
-                        dark_mode_btn.click(None, None, None, _js=js_code_for_toggle_darkmode)
-                    with gr.Tab("帮助", elem_id="interact-panel"):
-                        gr.Markdown(help_menu_description)
-
-            with gr.Floating(init_x="20%", init_y="50%", visible=False, width="40%", drag="top") as area_input_secondary:
-                with gr.Accordion("浮动输入区", open=True, elem_id="input-panel2"):
-                    with gr.Row() as row:
-                        row.style(equal_height=True)
-                        with gr.Column(scale=10):
-                            txt2 = gr.Textbox(show_label=False, placeholder="Input question here.",
-                                              elem_id='user_input_float', lines=8, label="输入区2").style(container=False)
-                        with gr.Column(scale=1, min_width=40):
-                            submitBtn2 = gr.Button("提交", variant="primary"); submitBtn2.style(size="sm")
-                            resetBtn2 = gr.Button("重置", variant="secondary"); resetBtn2.style(size="sm")
-                            stopBtn2 = gr.Button("停止", variant="secondary"); stopBtn2.style(size="sm")
-                            clearBtn2 = gr.Button("清除", elem_id="elem_clear2", variant="secondary", visible=False); clearBtn2.style(size="sm")
-
-            with gr.Floating(init_x="20%", init_y="50%", visible=False, width="40%", drag="top") as area_customize:
-                with gr.Accordion("自定义菜单", open=True, elem_id="edit-panel"):
-                    with gr.Row() as row:
-                        with gr.Column(scale=10):
-                            AVAIL_BTN = [btn for btn in customize_btns.keys()] + [k for k in functional]
-                            basic_btn_dropdown = gr.Dropdown(AVAIL_BTN, value="自定义按钮1", label="选择一个需要自定义基础功能区按钮").style(container=False)
-                            basic_fn_title = gr.Textbox(show_label=False, placeholder="输入新按钮名称", lines=1).style(container=False)
-                            basic_fn_prefix = gr.Textbox(show_label=False, placeholder="输入新提示前缀", lines=4).style(container=False)
-                            basic_fn_suffix = gr.Textbox(show_label=False, placeholder="输入新提示后缀", lines=4).style(container=False)
-                        with gr.Column(scale=1, min_width=70):
-                            basic_fn_confirm = gr.Button("确认并保存", variant="primary"); basic_fn_confirm.style(size="sm")
-                            basic_fn_clean   = gr.Button("恢复默认", variant="primary"); basic_fn_clean.style(size="sm")
-
-                            def assign_btn(persistent_cookie_, cookies_, basic_btn_dropdown_, basic_fn_title, basic_fn_prefix, basic_fn_suffix, clean_up=False):
-                                ret = {}
-                                # 读取之前的自定义按钮
-                                customize_fn_overwrite_ = cookies_['customize_fn_overwrite']
-                                # 更新新的自定义按钮
-                                customize_fn_overwrite_.update({
-                                    basic_btn_dropdown_:
-                                        {
-                                            "Title":basic_fn_title,
-                                            "Prefix":basic_fn_prefix,
-                                            "Suffix":basic_fn_suffix,
-                                        }
+                                if not match_group(plugin['Group'], DEFAULT_FN_GROUPS): continue
+                                if not plugin.get("AsButton", True): dropdown_fn_list.append(k)     # 排除已经是按钮的插件
+                                elif plugin.get('AdvancedArgs', False): dropdown_fn_list.append(k)  # 对于需要高级参数的插件，亦在下拉菜单中显示
+                            with gr.Row():
+                                dropdown = gr.Dropdown(dropdown_fn_list, value=r"打开插件列表", label="", show_label=False).style(container=False)
+                            with gr.Row():
+                                plugin_advanced_arg = gr.Textbox(show_label=True, label="高级参数输入区", visible=False,
+                                                                 placeholder="这里是特殊函数插件的高级参数输入区").style(container=False)
+                            with gr.Row():
+                                switchy_bt = gr.Button(r"请先从插件列表中选择", variant="secondary").style(size="sm")
+                    with gr.Row():
+                        with gr.Accordion("点击展开“文件下载区”。", open=False) as area_file_up:
+                            file_upload = gr.Files(label="任何文件, 推荐上传压缩文件(zip, tar)", file_count="multiple", elem_id="elem_upload")
+
+        with gr.Floating(init_x="0%", init_y="0%", visible=True, width=None, drag="forbidden", elem_id="tooltip"):
+            with gr.Row():
+                with gr.Tab("上传文件", elem_id="interact-panel"):
+                    gr.Markdown("请上传本地文件/压缩包供“函数插件区”功能调用。请注意: 上传文件后会自动把输入区修改为相应路径。")
+                    file_upload_2 = gr.Files(label="任何文件, 推荐上传压缩文件(zip, tar)", file_count="multiple", elem_id="elem_upload_float")
+
+                with gr.Tab("更换模型", elem_id="interact-panel"):
+                    md_dropdown = gr.Dropdown(AVAIL_LLM_MODELS, value=LLM_MODEL, label="更换LLM模型/请求源").style(container=False)
+                    top_p = gr.Slider(minimum=-0, maximum=1.0, value=1.0, step=0.01,interactive=True, label="Top-p (nucleus sampling)",)
+                    temperature = gr.Slider(minimum=-0, maximum=2.0, value=1.0, step=0.01, interactive=True, label="Temperature",)
+                    max_length_sl = gr.Slider(minimum=256, maximum=1024*32, value=4096, step=128, interactive=True, label="Local LLM MaxLength",)
+                    system_prompt = gr.Textbox(show_label=True, lines=2, placeholder=f"System Prompt", label="System prompt", value=INIT_SYS_PROMPT)
+
+                with gr.Tab("界面外观", elem_id="interact-panel"):
+                    theme_dropdown = gr.Dropdown(AVAIL_THEMES, value=THEME, label="更换UI主题").style(container=False)
+                    checkboxes = gr.CheckboxGroup(["基础功能区", "函数插件区", "浮动输入区", "输入清除键", "插件参数区"], value=["基础功能区", "函数插件区"], label="显示/隐藏功能区", elem_id='cbs').style(container=False)
+                    opt = ["自定义菜单"]
+                    value=[]
+                    if ADD_WAIFU: opt += ["添加Live2D形象"]; value += ["添加Live2D形象"]
+                    checkboxes_2 = gr.CheckboxGroup(opt, value=value, label="显示/隐藏自定义菜单", elem_id='cbsc').style(container=False)
+                    dark_mode_btn = gr.Button("切换界面明暗 ☀", variant="secondary").style(size="sm")
+                    dark_mode_btn.click(None, None, None, _js=js_code_for_toggle_darkmode)
+                with gr.Tab("帮助", elem_id="interact-panel"):
+                    gr.Markdown(help_menu_description)
+
+        with gr.Floating(init_x="20%", init_y="50%", visible=False, width="40%", drag="top") as area_input_secondary:
+            with gr.Accordion("浮动输入区", open=True, elem_id="input-panel2"):
+                with gr.Row() as row:
+                    row.style(equal_height=True)
+                    with gr.Column(scale=10):
+                        txt2 = gr.Textbox(show_label=False, placeholder="Input question here.",
+                                          elem_id='user_input_float', lines=8, label="输入区2").style(container=False)
+                    with gr.Column(scale=1, min_width=40):
+                        submitBtn2 = gr.Button("提交", variant="primary"); submitBtn2.style(size="sm")
+                        resetBtn2 = gr.Button("重置", variant="secondary"); resetBtn2.style(size="sm")
+                        stopBtn2 = gr.Button("停止", variant="secondary"); stopBtn2.style(size="sm")
+                        clearBtn2 = gr.Button("清除", elem_id="elem_clear2", variant="secondary", visible=False); clearBtn2.style(size="sm")
+
+
+        with gr.Floating(init_x="20%", init_y="50%", visible=False, width="40%", drag="top") as area_customize:
+            with gr.Accordion("自定义菜单", open=True, elem_id="edit-panel"):
+                with gr.Row() as row:
+                    with gr.Column(scale=10):
+                        AVAIL_BTN = [btn for btn in customize_btns.keys()] + [k for k in functional]
+                        basic_btn_dropdown = gr.Dropdown(AVAIL_BTN, value="自定义按钮1", label="选择一个需要自定义基础功能区按钮").style(container=False)
+                        basic_fn_title = gr.Textbox(show_label=False, placeholder="输入新按钮名称", lines=1).style(container=False)
+                        basic_fn_prefix = gr.Textbox(show_label=False, placeholder="输入新提示前缀", lines=4).style(container=False)
+                        basic_fn_suffix = gr.Textbox(show_label=False, placeholder="输入新提示后缀", lines=4).style(container=False)
+                    with gr.Column(scale=1, min_width=70):
+                        basic_fn_confirm = gr.Button("确认并保存", variant="primary"); basic_fn_confirm.style(size="sm")
+                        basic_fn_clean   = gr.Button("恢复默认", variant="primary"); basic_fn_clean.style(size="sm")
+                        def assign_btn(persistent_cookie_, cookies_, basic_btn_dropdown_, basic_fn_title, basic_fn_prefix, basic_fn_suffix, clean_up=False):
+                            ret = {}
+                            # 读取之前的自定义按钮
+                            customize_fn_overwrite_ = cookies_['customize_fn_overwrite']
+                            # 更新新的自定义按钮
+                            customize_fn_overwrite_.update({
+                                basic_btn_dropdown_:
+                                    {
+                                        "Title":basic_fn_title,
+                                        "Prefix":basic_fn_prefix,
+                                        "Suffix":basic_fn_suffix,
                                     }
-                                )
-                                if clean_up:
-                                    customize_fn_overwrite_ = {}
-                                cookies_.update(customize_fn_overwrite_)    # 更新cookie
-                                visible = (not clean_up) and (basic_fn_title != "")
-                                if basic_btn_dropdown_ in customize_btns:
-                                    # 是自定义按钮，不是预定义按钮
-                                    ret.update({customize_btns[basic_btn_dropdown_]: gr.update(visible=visible, value=basic_fn_title)})
-                                else:
-                                    # 是预定义按钮
-                                    ret.update({predefined_btns[basic_btn_dropdown_]: gr.update(visible=visible, value=basic_fn_title)})
-                                ret.update({cookies: cookies_})
-                                try: persistent_cookie_ = from_cookie_str(persistent_cookie_)   # persistent cookie to dict
-                                except: persistent_cookie_ = {}
-                                persistent_cookie_["custom_bnt"] = customize_fn_overwrite_      # dict update new value
-                                persistent_cookie_ = to_cookie_str(persistent_cookie_)          # persistent cookie to dict
-                                ret.update({py_pickle_cookie: persistent_cookie_})             # write persistent cookie
-                                return ret
-
-                            # update btn
-                            h = basic_fn_confirm.click(assign_btn, [py_pickle_cookie, cookies, basic_btn_dropdown, basic_fn_title, basic_fn_prefix, basic_fn_suffix],
-                                                       [py_pickle_cookie, cookies, *customize_btns.values(), *predefined_btns.values()])
-                            h.then(None, [py_pickle_cookie], None, _js="""(py_pickle_cookie)=>{setCookie("py_pickle_cookie", py_pickle_cookie, 365);}""")
-                            # clean up btn
-                            h2 = basic_fn_clean.click(assign_btn, [py_pickle_cookie, cookies, basic_btn_dropdown, basic_fn_title, basic_fn_prefix, basic_fn_suffix, gr.State(True)],
-                                                       [py_pickle_cookie, cookies, *customize_btns.values(), *predefined_btns.values()])
-                            h2.then(None, [py_pickle_cookie], None, _js="""(py_pickle_cookie)=>{setCookie("py_pickle_cookie", py_pickle_cookie, 365);}""")
-
-                            def persistent_cookie_reload(persistent_cookie_, cookies_):
-                                ret = {}
-                                for k in customize_btns:
-                                    ret.update({customize_btns[k]: gr.update(visible=False, value="")})
-
-                                try: persistent_cookie_ = from_cookie_str(persistent_cookie_)    # persistent cookie to dict
-                                except: return ret
-
-                                customize_fn_overwrite_ = persistent_cookie_.get("custom_bnt", {})
-                                cookies_['customize_fn_overwrite'] = customize_fn_overwrite_
-                                ret.update({cookies: cookies_})
-
-                                for k,v in persistent_cookie_["custom_bnt"].items():
-                                    if v['Title'] == "": continue
-                                    if k in customize_btns: ret.update({customize_btns[k]: gr.update(visible=True, value=v['Title'])})
-                                    else: ret.update({predefined_btns[k]: gr.update(visible=True, value=v['Title'])})
-                                return ret
-
-            # 添加 JavaScript 来解析 URL 参数并设置到隐藏的 Textbox
-            url_param_js = """
-            <script>
-            window.onload = function() {
-                const params = new URLSearchParams(window.location.search);
-                let base_url = params.get('base_url') || '';
-                let api_key = params.get('api_key') || '';
-                let combined = `base_url=${encodeURIComponent(base_url)}&api_key=${encodeURIComponent(api_key)}`;
-                // 将解析后的参数设置到隐藏的 Textbox
-                const urlParamsTextbox = document.querySelector('input[aria-label="URL Parameters"]');
-                if (urlParamsTextbox) {
-                    urlParamsTextbox.value = combined;
-                }
-            }
-            </script>
-            """
-            gr.HTML(url_param_js)
-
-            # 定义从隐藏的 Textbox 中提取 base_url 和 api_key 的函数
-            def extract_api_credentials(url_params, cookies):
-                try:
-                    # 解析 URL 参数
-                    parsed = dict(param.split('=') for param in url_params.split('&') if '=' in param)
-                    base_url = parsed.get('base_url', None)
-                    api_key = parsed.get('api_key', None)
-
-                    print(f"Extracted base_url: {base_url}")
-                    print(f"Extracted api_key: {api_key}")
-
-                    # 更新 cookies 或应用程序状态
-                    if base_url:
-                        cookies['base_url'] = base_url
-                    if api_key:
-                        cookies['api_key'] = api_key  # 警告：通过 URL 传递 API 密钥是不安全的
-
-                    return cookies
-                except Exception as e:
-                    print(f"Error in extract_api_credentials: {e}")
-                    return cookies
-
-            # 在应用加载时调用该函数，通过隐藏的 Textbox 获取 URL 参数
-            demo.load(
-                fn=extract_api_credentials,
-                inputs=[url_params, cookies],
-                outputs=[cookies]
-            )
-
-            # 准备传递给 predict 函数的输入
-            input_combo = [cookies, max_length_sl, md_dropdown, txt, txt2, top_p, temperature, chatbot, history, system_prompt, plugin_advanced_arg]
-            output_combo = [cookies, chatbot, history, status]
-            predict_args = dict(fn=ArgsGeneralWrapper(predict), inputs=[*input_combo, gr.State(True)], outputs=output_combo)
-
-            # 提交按钮和重置按钮
-            cancel_handles.append(txt.submit(**predict_args))
-            cancel_handles.append(txt2.submit(**predict_args))
-            cancel_handles.append(submitBtn.click(**predict_args))
-            cancel_handles.append(submitBtn2.click(**predict_args))
-            resetBtn.click(None, None, [chatbot, history, status], _js=js_code_reset)   # 先在前端快速清除chatbot&status
-            resetBtn2.click(None, None, [chatbot, history, status], _js=js_code_reset)  # 先在前端快速清除chatbot&status
-            resetBtn.click(lambda: ([], [], "已重置"), None, [chatbot, history, status])    # 再在后端清除history
-            resetBtn2.click(lambda: ([], [], "已重置"), None, [chatbot, history, status])   # 再在后端清除history
-            clearBtn.click(None, None, [txt, txt2], _js=js_code_clear)
-            clearBtn2.click(None, None, [txt, txt2], _js=js_code_clear)
-            if AUTO_CLEAR_TXT:
-                submitBtn.click(None, None, [txt, txt2], _js=js_code_clear)
-                submitBtn2.click(None, None, [txt, txt2], _js=js_code_clear)
-                txt.submit(None, None, [txt, txt2], _js=js_code_clear)
-                txt2.submit(None, None, [txt, txt2], _js=js_code_clear)
-            # 基础功能区的回调函数注册
-            for k in functional:
-                if ("Visible" in functional[k]) and (not functional[k]["Visible"]): continue
-                click_handle = functional[k]["Button"].click(fn=ArgsGeneralWrapper(predict), inputs=[*input_combo, gr.State(True), gr.State(k)], outputs=output_combo)
-                cancel_handles.append(click_handle)
-            for btn in customize_btns.values():
-                click_handle = btn.click(fn=ArgsGeneralWrapper(predict), inputs=[*input_combo, gr.State(True), gr.State(btn.value)], outputs=output_combo)
-                cancel_handles.append(click_handle)
-            # 文件上传区，接收文件后与chatbot的互动
-            file_upload.upload(on_file_uploaded, [file_upload, chatbot, txt, txt2, checkboxes, cookies], [chatbot, txt, txt2, cookies]).then(None, None, None,   _js=r"()=>{toast_push('上传完毕 ...'); cancel_loading_status();}")
-            file_upload_2.upload(on_file_uploaded, [file_upload_2, chatbot, txt, txt2, checkboxes, cookies], [chatbot, txt, txt2, cookies]).then(None, None, None, _js=r"()=>{toast_push('上传完毕 ...'); cancel_loading_status();}")
-            # 函数插件-固定按钮区
-            for k in plugins:
-                if not plugins[k].get("AsButton", True): continue
-                click_handle = plugins[k]["Button"].click(ArgsGeneralWrapper(plugins[k]["Function"]), [*input_combo], output_combo)
-                click_handle.then(on_report_generated, [cookies, file_upload, chatbot], [cookies, file_upload, chatbot])
-                cancel_handles.append(click_handle)
-            # 函数插件-下拉菜单与随变按钮的互动
-            def on_dropdown_changed(k):
-                variant = plugins[k]["Color"] if "Color" in plugins else "secondary"
-                info = plugins[k].get("Info", k)
-                ret = {switchy_bt: gr.update(value=k, variant=variant, info_str=f'函数插件区: {info}')}
-                if plugins[k].get("AdvancedArgs", False): # 是否唤起高级插件参数区
-                    ret.update({plugin_advanced_arg: gr.update(visible=True,  label=f"插件[{k}]的高级参数说明：" + plugins[k].get("ArgsReminder", [f"没有提供高级参数功能说明"]))})
-                else:
-                    ret.update({plugin_advanced_arg: gr.update(visible=False, label=f"插件[{k}]不需要高级参数。")})
-                return ret
-            dropdown.select(on_dropdown_changed, [dropdown], [switchy_bt, plugin_advanced_arg] )
-
-            def on_md_dropdown_changed(k):
-                return {chatbot: gr.update(label="当前模型："+k)}
-            md_dropdown.select(on_md_dropdown_changed, [md_dropdown], [chatbot] )
-
-            def on_theme_dropdown_changed(theme, secret_css):
-                adjust_theme_func, css_part1, _, adjust_dynamic_theme = load_dynamic_theme(theme)
-                if adjust_dynamic_theme:
-                    css_part2 = adjust_dynamic_theme._get_theme_css()
-                else:
-                    css_part2 = adjust_theme_func()._get_theme_css()
-                return css_part2 + css_part1
-
-            theme_handle = theme_dropdown.select(on_theme_dropdown_changed, [theme_dropdown, secret_css], [secret_css])
-            theme_handle.then(
-                None,
-                [secret_css],
-                None,
-                _js=js_code_for_css_changing
-            )
-            # 随变按钮的回调函数注册
-            def route(k, *args, **kwargs):
-                if k in [r"打开插件列表", r"请先从插件列表中选择"]: return
-                return ArgsGeneralWrapper(plugins[k]["Function"])(*args, **kwargs)
-            click_handle = switchy_bt.click(route,[switchy_bt, *input_combo], output_combo)
+                                }
+                            )
+                            if clean_up:
+                                customize_fn_overwrite_ = {}
+                            cookies_.update(customize_fn_overwrite_)    # 更新cookie
+                            visible = (not clean_up) and (basic_fn_title != "")
+                            if basic_btn_dropdown_ in customize_btns:
+                                # 是自定义按钮，不是预定义按钮
+                                ret.update({customize_btns[basic_btn_dropdown_]: gr.update(visible=visible, value=basic_fn_title)})
+                            else:
+                                # 是预定义按钮
+                                ret.update({predefined_btns[basic_btn_dropdown_]: gr.update(visible=visible, value=basic_fn_title)})
+                            ret.update({cookies: cookies_})
+                            try: persistent_cookie_ = from_cookie_str(persistent_cookie_)   # persistent cookie to dict
+                            except: persistent_cookie_ = {}
+                            persistent_cookie_["custom_bnt"] = customize_fn_overwrite_      # dict update new value
+                            persistent_cookie_ = to_cookie_str(persistent_cookie_)          # persistent cookie to dict
+                            ret.update({py_pickle_cookie: persistent_cookie_})             # write persistent cookie
+                            return ret
+
+                        # update btn
+                        h = basic_fn_confirm.click(assign_btn, [py_pickle_cookie, cookies, basic_btn_dropdown, basic_fn_title, basic_fn_prefix, basic_fn_suffix],
+                                                   [py_pickle_cookie, cookies, *customize_btns.values(), *predefined_btns.values()])
+                        h.then(None, [py_pickle_cookie], None, _js="""(py_pickle_cookie)=>{setCookie("py_pickle_cookie", py_pickle_cookie, 365);}""")
+                        # clean up btn
+                        h2 = basic_fn_clean.click(assign_btn, [py_pickle_cookie, cookies, basic_btn_dropdown, basic_fn_title, basic_fn_prefix, basic_fn_suffix, gr.State(True)],
+                                                   [py_pickle_cookie, cookies, *customize_btns.values(), *predefined_btns.values()])
+                        h2.then(None, [py_pickle_cookie], None, _js="""(py_pickle_cookie)=>{setCookie("py_pickle_cookie", py_pickle_cookie, 365);}""")
+
+                        def persistent_cookie_reload(persistent_cookie_, cookies_):
+                            ret = {}
+                            for k in customize_btns:
+                                ret.update({customize_btns[k]: gr.update(visible=False, value="")})
+
+                            try: persistent_cookie_ = from_cookie_str(persistent_cookie_)    # persistent cookie to dict
+                            except: return ret
+
+                            customize_fn_overwrite_ = persistent_cookie_.get("custom_bnt", {})
+                            cookies_['customize_fn_overwrite'] = customize_fn_overwrite_
+                            ret.update({cookies: cookies_})
+
+                            for k,v in persistent_cookie_["custom_bnt"].items():
+                                if v['Title'] == "": continue
+                                if k in customize_btns: ret.update({customize_btns[k]: gr.update(visible=True, value=v['Title'])})
+                                else: ret.update({predefined_btns[k]: gr.update(visible=True, value=v['Title'])})
+                            return ret
+
+        # 功能区显示开关与功能区的互动
+        def fn_area_visibility(a):
+            ret = {}
+            ret.update({area_input_primary: gr.update(visible=("浮动输入区" not in a))})
+            ret.update({area_input_secondary: gr.update(visible=("浮动输入区" in a))})
+            ret.update({plugin_advanced_arg: gr.update(visible=("插件参数区" in a))})
+            if "浮动输入区" in a: ret.update({txt: gr.update(value="")})
+            return ret
+        checkboxes.select(fn_area_visibility, [checkboxes], [area_basic_fn, area_crazy_fn, area_input_primary, area_input_secondary, txt, txt2, plugin_advanced_arg] )
+        checkboxes.select(None, [checkboxes], None, _js=js_code_show_or_hide)
+
+        # 功能区显示开关与功能区的互动
+        def fn_area_visibility_2(a):
+            ret = {}
+            ret.update({area_customize: gr.update(visible=("自定义菜单" in a))})
+            return ret
+        checkboxes_2.select(fn_area_visibility_2, [checkboxes_2], [area_customize] )
+        checkboxes_2.select(None, [checkboxes_2], None, _js=js_code_show_or_hide_group2)
+
+        # 整理反复出现的控件句柄组合
+        input_combo = [cookies, max_length_sl, md_dropdown, txt, txt2, top_p, temperature, chatbot, history, system_prompt, plugin_advanced_arg]
+        output_combo = [cookies, chatbot, history, status]
+        predict_args = dict(fn=ArgsGeneralWrapper(predict), inputs=[*input_combo, gr.State(True)], outputs=output_combo)
+        # 提交按钮、重置按钮
+        cancel_handles.append(txt.submit(**predict_args))
+        cancel_handles.append(txt2.submit(**predict_args))
+        cancel_handles.append(submitBtn.click(**predict_args))
+        cancel_handles.append(submitBtn2.click(**predict_args))
+        resetBtn.click(None, None, [chatbot, history, status], _js=js_code_reset)   # 先在前端快速清除chatbot&status
+        resetBtn2.click(None, None, [chatbot, history, status], _js=js_code_reset)  # 先在前端快速清除chatbot&status
+        resetBtn.click(lambda: ([], [], "已重置"), None, [chatbot, history, status])    # 再在后端清除history
+        resetBtn2.click(lambda: ([], [], "已重置"), None, [chatbot, history, status])   # 再在后端清除history
+        clearBtn.click(None, None, [txt, txt2], _js=js_code_clear)
+        clearBtn2.click(None, None, [txt, txt2], _js=js_code_clear)
+        if AUTO_CLEAR_TXT:
+            submitBtn.click(None, None, [txt, txt2], _js=js_code_clear)
+            submitBtn2.click(None, None, [txt, txt2], _js=js_code_clear)
+            txt.submit(None, None, [txt, txt2], _js=js_code_clear)
+            txt2.submit(None, None, [txt, txt2], _js=js_code_clear)
+        # 基础功能区的回调函数注册
+        for k in functional:
+            if ("Visible" in functional[k]) and (not functional[k]["Visible"]): continue
+            click_handle = functional[k]["Button"].click(fn=ArgsGeneralWrapper(predict), inputs=[*input_combo, gr.State(True), gr.State(k)], outputs=output_combo)
+            cancel_handles.append(click_handle)
+        for btn in customize_btns.values():
+            click_handle = btn.click(fn=ArgsGeneralWrapper(predict), inputs=[*input_combo, gr.State(True), gr.State(btn.value)], outputs=output_combo)
+            cancel_handles.append(click_handle)
+        # 文件上传区，接收文件后与chatbot的互动
+        file_upload.upload(on_file_uploaded, [file_upload, chatbot, txt, txt2, checkboxes, cookies], [chatbot, txt, txt2, cookies]).then(None, None, None,   _js=r"()=>{toast_push('上传完毕 ...'); cancel_loading_status();}")
+        file_upload_2.upload(on_file_uploaded, [file_upload_2, chatbot, txt, txt2, checkboxes, cookies], [chatbot, txt, txt2, cookies]).then(None, None, None, _js=r"()=>{toast_push('上传完毕 ...'); cancel_loading_status();}")
+        # 函数插件-固定按钮区
+        for k in plugins:
+            if not plugins[k].get("AsButton", True): continue
+            click_handle = plugins[k]["Button"].click(ArgsGeneralWrapper(plugins[k]["Function"]), [*input_combo], output_combo)
             click_handle.then(on_report_generated, [cookies, file_upload, chatbot], [cookies, file_upload, chatbot])
             cancel_handles.append(click_handle)
-            # 终止按钮的回调函数注册
-            stopBtn.click(fn=None, inputs=None, outputs=None, cancels=cancel_handles)
-            stopBtn2.click(fn=None, inputs=None, outputs=None, cancels=cancel_handles)
-            plugins_as_btn = {name:plugin for name, plugin in plugins.items() if plugin.get('Button', None)}
-            def on_group_change(group_list):
-                btn_list = []
-                fns_list = []
-                if not group_list: # 处理特殊情况：没有选择任何插件组
-                    return [*[plugin['Button'].update(visible=False) for _, plugin in plugins_as_btn.items()], gr.Dropdown.update(choices=[])]
-                for k, plugin in plugins.items():
-                    if plugin.get("AsButton", True):
-                        btn_list.append(plugin['Button'].update(visible=match_group(plugin['Group'], group_list))) # 刷新按钮
-                        if plugin.get('AdvancedArgs', False): fns_list.append(k) # 对于需要高级参数的插件，亦在下拉菜单中显示
-                    elif match_group(plugin['Group'], group_list): fns_list.append(k) # 刷新下拉列表
-                return [*btn_list, gr.Dropdown.update(choices=fns_list)]
-            plugin_group_sel.select(fn=on_group_change, inputs=[plugin_group_sel], outputs=[*[plugin['Button'] for name, plugin in plugins_as_btn.items()], dropdown])
-            if ENABLE_AUDIO:
-                from crazy_functions.live_audio.audio_io import RealtimeAudioDistribution
-                rad = RealtimeAudioDistribution()
-                def deal_audio(audio, cookies):
-                    rad.feed(cookies['uuid'].hex, audio)
-                audio_mic.stream(deal_audio, inputs=[audio_mic, cookies])
-
-            # 定义从隐藏的 Textbox 中提取 base_url 和 api_key 的函数（已定义上面）
-            # 已在上面调用
-
-            # 启动 Gradio 的延迟任务
-            def run_delayed_tasks():
-                import threading, webbrowser, time
-                print(f"如果浏览器没有自动打开，请复制并转到以下URL：")
-                if DARK_MODE:   
-                    print(f"\t「暗色主题已启用（支持动态切换主题）」: http://localhost:{PORT}")
-                else:           
-                    print(f"\t「亮色主题已启用（支持动态切换主题）」: http://localhost:{PORT}")
-
-                def auto_updates(): 
-                    time.sleep(0)
-                    auto_update()
-                def open_browser(): 
-                    time.sleep(2)
-                    webbrowser.open_new_tab(f"http://localhost:{PORT}")
-                def warm_up_mods(): 
-                    time.sleep(6)
-                    warm_up_modules()
-
-                threading.Thread(target=auto_updates, name="self-upgrade", daemon=True).start() # 查看自动更新
-                threading.Thread(target=open_browser, name="open-browser", daemon=True).start() # 打开浏览器页面
-                threading.Thread(target=warm_up_mods, name="warm-up", daemon=True).start()      # 预热tiktoken模块
-
-            run_delayed_tasks()
-            demo.queue(concurrency_count=CONCURRENT_COUNT).launch(server_name="0.0.0.0", share=False, favicon_path="docs/logo.png", blocked_paths=["config.py","config_private.py","docker-compose.yml","Dockerfile"])
-
-            # 如果需要在二级路径下运行
-            # CUSTOM_PATH = get_conf('CUSTOM_PATH')
-            # if CUSTOM_PATH != "/":
-            #     from toolbox import run_gradio_in_subpath
-            #     run_gradio_in_subpath(demo, auth=AUTHENTICATION, port=PORT, custom_path=CUSTOM_PATH)
-            # else:
-            #     demo.launch(server_name="0.0.0.0", server_port=PORT, auth=AUTHENTICATION, favicon_path="docs/logo.png",
-            #                 blocked_paths=["config.py","config_private.py","docker-compose.yml","Dockerfile",f"{PATH_LOGGING}/admin"])
-
-    if __name__ == "__main__":
-        main()
+        # 函数插件-下拉菜单与随变按钮的互动
+        def on_dropdown_changed(k):
+            variant = plugins[k]["Color"] if "Color" in plugins[k] else "secondary"
+            info = plugins[k].get("Info", k)
+            ret = {switchy_bt: gr.update(value=k, variant=variant, info_str=f'函数插件区: {info}')}
+            if plugins[k].get("AdvancedArgs", False): # 是否唤起高级插件参数区
+                ret.update({plugin_advanced_arg: gr.update(visible=True,  label=f"插件[{k}]的高级参数说明：" + plugins[k].get("ArgsReminder", [f"没有提供高级参数功能说明"]))})
+            else:
+                ret.update({plugin_advanced_arg: gr.update(visible=False, label=f"插件[{k}]不需要高级参数。")})
+            return ret
+        dropdown.select(on_dropdown_changed, [dropdown], [switchy_bt, plugin_advanced_arg] )
+
+        def on_md_dropdown_changed(k):
+            return {chatbot: gr.update(label="当前模型："+k)}
+        md_dropdown.select(on_md_dropdown_changed, [md_dropdown], [chatbot] )
+
+        def on_theme_dropdown_changed(theme, secret_css):
+            adjust_theme, css_part1, _, adjust_dynamic_theme = load_dynamic_theme(theme)
+            if adjust_dynamic_theme:
+                css_part2 = adjust_dynamic_theme._get_theme_css()
+            else:
+                css_part2 = adjust_theme()._get_theme_css()
+            return css_part2 + css_part1
+
+        theme_handle = theme_dropdown.select(on_theme_dropdown_changed, [theme_dropdown, secret_css], [secret_css])
+        theme_handle.then(
+            None,
+            [secret_css],
+            None,
+            _js=js_code_for_css_changing
+        )
+        # 随变按钮的回调函数注册
+        def route(request: gr.Request, k, *args, **kwargs):
+            if k in [r"打开插件列表", r"请先从插件列表中选择"]: return
+            yield from ArgsGeneralWrapper(plugins[k]["Function"])(request, *args, **kwargs)
+        click_handle = switchy_bt.click(route,[switchy_bt, *input_combo], output_combo)
+        click_handle.then(on_report_generated, [cookies, file_upload, chatbot], [cookies, file_upload, chatbot])
+        cancel_handles.append(click_handle)
+        # 终止按钮的回调函数注册
+        stopBtn.click(fn=None, inputs=None, outputs=None, cancels=cancel_handles)
+        stopBtn2.click(fn=None, inputs=None, outputs=None, cancels=cancel_handles)
+        plugins_as_btn = {name:plugin for name, plugin in plugins.items() if plugin.get('Button', None)}
+        def on_group_change(group_list):
+            btn_list = []
+            fns_list = []
+            if not group_list: # 处理特殊情况：没有选择任何插件组
+                return [*[plugin['Button'].update(visible=False) for _, plugin in plugins_as_btn.items()], gr.Dropdown.update(choices=[])]
+            for k, plugin in plugins.items():
+                if plugin.get("AsButton", True):
+                    btn_list.append(plugin['Button'].update(visible=match_group(plugin['Group'], group_list))) # 刷新按钮
+                    if plugin.get('AdvancedArgs', False): dropdown_fn_list.append(k) # 对于需要高级参数的插件，亦在下拉菜单中显示
+                elif match_group(plugin['Group'], group_list): fns_list.append(k) # 刷新下拉列表
+            return [*btn_list, gr.Dropdown.update(choices=fns_list)]
+        plugin_group_sel.select(fn=on_group_change, inputs=[plugin_group_sel], outputs=[*[plugin['Button'] for name, plugin in plugins_as_btn.items()], dropdown])
+        if ENABLE_AUDIO:
+            from crazy_functions.live_audio.audio_io import RealtimeAudioDistribution
+            rad = RealtimeAudioDistribution()
+            def deal_audio(audio, cookies):
+                rad.feed(cookies['uuid'].hex, audio)
+            audio_mic.stream(deal_audio, inputs=[audio_mic, cookies])
+
+
+        demo.load(init_cookie, inputs=[cookies], outputs=[cookies])
+        demo.load(persistent_cookie_reload, inputs = [py_pickle_cookie, cookies],
+            outputs = [py_pickle_cookie, cookies, *customize_btns.values(), *predefined_btns.values()], _js=js_code_for_persistent_cookie_init)
+        demo.load(None, inputs=[dark_mode], outputs=None, _js="""(dark_mode)=>{apply_cookie_for_checkbox(dark_mode);}""")    # 配置暗色主题或亮色主题
+        demo.load(None, inputs=[gr.Textbox(LAYOUT, visible=False)], outputs=None, _js='(LAYOUT)=>{GptAcademicJavaScriptInit(LAYOUT);}')
+
+        # 启动 Gradio 的延迟任务
+        def run_delayed_tasks():
+            import threading, webbrowser, time
+            print(f"如果浏览器没有自动打开，请复制并转到以下URL：")
+            if DARK_MODE:   
+                print(f"\t「暗色主题已启用（支持动态切换主题）」: http://localhost:{PORT}")
+            else:           
+                print(f"\t「亮色主题已启用（支持动态切换主题）」: http://localhost:{PORT}")
+
+            def auto_updates(): 
+                time.sleep(0)
+                auto_update()
+            def open_browser(): 
+                time.sleep(2)
+                webbrowser.open_new_tab(f"http://localhost:{PORT}")
+            def warm_up_mods(): 
+                time.sleep(6)
+                warm_up_modules()
+
+            threading.Thread(target=auto_updates, name="self-upgrade", daemon=True).start() # 查看自动更新
+            threading.Thread(target=open_browser, name="open-browser", daemon=True).start() # 打开浏览器页面
+            threading.Thread(target=warm_up_mods, name="warm-up", daemon=True).start()      # 预热tiktoken模块
+
+        run_delayed_tasks()
+        demo.queue(concurrency_count=CONCURRENT_COUNT).launch(server_name="0.0.0.0", share=False, favicon_path="docs/logo.png", blocked_paths=["config.py","config_private.py","docker-compose.yml","Dockerfile"])
+
+        # 如果需要在二级路径下运行
+        # CUSTOM_PATH = get_conf('CUSTOM_PATH')
+        # if CUSTOM_PATH != "/":
+        #     from toolbox import run_gradio_in_subpath
+        #     run_gradio_in_subpath(demo, auth=AUTHENTICATION, port=PORT, custom_path=CUSTOM_PATH)
+        # else:
+        #     demo.launch(server_name="0.0.0.0", server_port=PORT, auth=AUTHENTICATION, favicon_path="docs/logo.png",
+        #                 blocked_paths=["config.py","config_private.py","docker-compose.yml","Dockerfile",f"{PATH_LOGGING}/admin"])
+
+    except Exception as e:
+        print("Unhandled exception during application startup:")
+        traceback.print_exc()
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main()