feat: 实现插件系统与默认插件

.gitignore

@@ -149,6 +149,8 @@ files/
 tmp/
 
 scripts/
+!extensions/**/scripts/
+!extensions/**/scripts/*.py
 include/
 pyvenv.cfg
 

ChuanhuChatbot.py

@@ -14,12 +14,14 @@
 from modules.utils import *
 from modules.config import *
 from modules import config
+from modules import extensions
 import gradio as gr
 import colorama
 
 logging.getLogger("httpx").setLevel(logging.WARNING)
 
 patch_gradio()
+extensions.load_extensions(disabled_extensions=disabled_extensions)
 
 # with open("web_assets/css/ChuanhuChat.css", "r", encoding="utf-8") as f:
 #     ChuanhuChatCSS = f.read()
@@ -344,9 +346,7 @@ def create_new_model():
                                 value=user_name.value,
                                 lines=1,
                             )
-                    with gr.Tab(label=i18n("拓展")):
-                        gr.Markdown(
-                            "Will be here soon...\n(We hope)\n\nAnd we hope you can help us to make more extensions!")
+                    extensions.render_extension_tabs()
 
                     # changeAPIURLBtn = gr.Button(i18n("🔄 切换API地址"))
 
@@ -357,6 +357,9 @@ def create_new_model():
                     gr.Markdown("## "+i18n("设置"))
                     gr.HTML(get_html("close_btn.html").format(
                         obj="box"), elem_classes="close-btn")
+                gr.HTML(
+                    f'<span id="extension-settings-label-source" style="display:none">{i18n("插件设置")}</span>'
+                )
                 with gr.Tabs(elem_id="chuanhu-setting-tabs"):
                     # with gr.Tab(label=i18n("模型")):
 
@@ -425,6 +428,9 @@ def create_new_model():
                             elem_classes="view-only-textbox no-container",
                         )
 
+                    with gr.Tab(label=i18n("插件")):
+                        extensions.render_extension_manager()
+
                     with gr.Tab(label=i18n("关于"), elem_id="about-tab"):
                         gr.Markdown(
                             '<img alt="Chuanhu Chat logo" src="file=web_assets/icon/any-icon-512.png" style="max-width: 144px;">')
@@ -433,6 +439,8 @@ def create_new_model():
                             versions=versions_html()), elem_id="footer")
                         gr.Markdown(CHUANHU_DESCRIPTION, elem_id="description")
 
+                    extensions.render_extension_settings()
+
             with gr.Group(elem_id="chuanhu-training"):
                 with gr.Row():
                     gr.Markdown("## "+i18n("训练"))
@@ -803,6 +811,7 @@ def create_greeting(request: gr.Request):
     reload_javascript()
     setup_wizard()
     _allowed_paths = ["web_assets"]
+    _allowed_paths.append("extensions")
     if config.midjourney_temp_folder:
         _allowed_paths.append(config.midjourney_temp_folder)
     demo.queue().launch(

config_example.json

@@ -69,6 +69,7 @@
     // 是否多个API Key轮换使用
     "multi_api_key": false,
     "hide_my_key": false, // 如果你想在UI中隐藏 API 密钥输入框，将此值设置为 true
+    "disabled_extensions": [], // 禁用的插件 ID 列表，例如 ["prompt_prefix_demo"]
     // "available_models": ["GPT3.5 Turbo", "GPT4 Turbo", "GPT4 Vision"], // 可用的模型列表，将覆盖默认的可用模型列表
     // "extra_models": ["模型名称3", "模型名称4", ...], // 额外的模型，将添加到可用的模型列表之后
     // "extra_model_metadata": {