Upload 2 files

config_private.py

@@ -0,0 +1,342 @@
+"""
+    以下所有配置也都支持利用环境变量覆写，环境变量配置格式见docker-compose.yml。
+    读取优先级：环境变量 > config_private.py > config.py
+    --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- --- ---
+    All the following configurations also support using environment variables to override,
+    and the environment variable configuration format can be seen in docker-compose.yml.
+    Configuration reading priority: environment variable > config_private.py > config.py
+"""
+
+# [step 1]>> API_KEY = "ak-SG7fqdmWfDQs5N99ejy0uLkvd6oG6YD3CiU681qWFP4IdY14"。极少数情况下，还需要填写组织（格式如org-123456789abcdefghijklmno的），请向下翻，找 API_ORG 设置项
+API_KEY = "ghu_zRDjm7nLr6xtDiIT22sRwHEPtepPgF3P0mSb"  # 可同时填写多个API-KEY，用英文逗号分割，例如API_KEY = "sk-openaikey1,sk-openaikey2,fkxxxx-api2dkey3,azure-apikey4"
+
+# [step 2]>> 改为True应用代理，如果直接在海外服务器部署，此处不修改；如果使用本地或无地域限制的大模型时，此处也不需要修改
+USE_PROXY = False
+if USE_PROXY:
+    """
+    代理网络的地址，打开你的代理软件查看代理协议(socks5h / http)、地址(localhost)和端口(11284)
+    填写格式是 [协议]://  [地址] :[端口]，填写之前不要忘记把USE_PROXY改成True，如果直接在海外服务器部署，此处不修改
+            <配置教程&视频教程> https://github.com/binary-husky/gpt_academic/issues/1>
+    [协议] 常见协议无非socks5h/http; 例如 v2**y 和 ss* 的默认本地协议是socks5h; 而cl**h 的默认本地协议是http
+    [地址] 填localhost或者127.0.0.1（localhost意思是代理软件安装在本机上）
+    [端口] 在代理软件的设置里找。虽然不同的代理软件界面不一样，但端口号都应该在最显眼的位置上
+    """
+    proxies = {
+        #          [协议]://  [地址]  :[端口]
+        "http":
+        "http://127.0.0.1:7890",  # 再例如  "http":  "http://127.0.0.1:7890",
+        "https":
+        "http://127.0.0.1:7890",  # 再例如  "https": "http://127.0.0.1:7890",
+    }
+else:
+    proxies = None
+
+# [step 3]>> 模型选择是 (注意: LLM_MODEL是默认选中的模型, 它*必须*被包含在AVAIL_LLM_MODELS列表中 )
+LLM_MODEL = "gpt-4"  # 可选 ↓↓↓
+AVAIL_LLM_MODELS = [
+    "gpt-4-1106-preview", "gpt-4-turbo-preview", "gpt-4-vision-preview",
+    "gpt-3.5-turbo-1106", "gpt-3.5-turbo-16k", "gpt-3.5-turbo",
+    "azure-gpt-3.5", "gpt-4", "gpt-4-32k", "azure-gpt-4", "glm-4",
+    "glm-3-turbo", "gemini-pro", "chatglm3"
+]
+# --- --- --- ---
+# P.S. 其他可用的模型还包括
+# AVAIL_LLM_MODELS = [
+#   "qianfan", "deepseekcoder",
+#   "spark", "sparkv2", "sparkv3", "sparkv3.5",
+#   "qwen-turbo", "qwen-plus", "qwen-max", "qwen-local",
+#   "moonshot-v1-128k", "moonshot-v1-32k", "moonshot-v1-8k",
+#   "gpt-3.5-turbo-0613", "gpt-3.5-turbo-16k-0613", "gpt-3.5-turbo-0125"
+#   "claude-3-sonnet-20240229","claude-3-opus-20240229", "claude-2.1", "claude-instant-1.2",
+#   "moss", "llama2", "chatglm_onnx", "internlm", "jittorllms_pangualpha", "jittorllms_llama",
+# ]
+# --- --- --- ---
+# 此外，为了更灵活地接入one-api多模型管理界面，您还可以在接入one-api时，
+# 使用"one-api-*"前缀直接使用非标准方式接入的模型，例如
+# AVAIL_LLM_MODELS = ["one-api-claude-3-sonnet-20240229(max_token=100000)"]
+# --- --- --- ---
+
+# --------------- 以下配置可以优化体验 ---------------
+
+# 重新URL重新定向，实现更换API_URL的作用（高危设置! 常规情况下不要修改! 通过修改此设置，您将把您的API-KEY和对话隐私完全暴露给您设定的中间人！）
+# 格式: API_URL_REDIRECT = {"https://api.openai.com/v1/chat/completions": "在这里填写重定向的api.openai.com的URL"}
+# 举例: API_URL_REDIRECT = {"https://api.openai.com/v1/chat/completions": "https://reverse-proxy-url/v1/chat/completions"}
+# http://localhost:8080/v1/chat/completions
+
+API_URL_REDIRECT = {}
+API_URL_REDIRECT = {
+    "https://api.openai.com/v1/chat/completions":
+    "https://proxy.cocopilot.org/v1/chat/completions"
+}
+
+# 多线程函数插件中，默认允许多少路线程同时访问OpenAI。Free trial users的限制是每分钟3次，Pay-as-you-go users的限制是每分钟3500次
+# 一言以蔽之：免费（5刀）用户填3，OpenAI绑了信用卡的用户可以填 16 或者更高。提高限制请查询：https://platform.openai.com/docs/guides/rate-limits/overview
+DEFAULT_WORKER_NUM = 3
+
+# 色彩主题, 可选 ["Default", "Chuanhu-Small-and-Beautiful", "High-Contrast"]
+# 更多主题, 请查阅Gradio主题商店: https://huggingface.co/spaces/gradio/theme-gallery 可选 ["Gstaff/Xkcd", "NoCrypt/Miku", ...]
+THEME = "Default"
+AVAIL_THEMES = [
+    "Default", "Chuanhu-Small-and-Beautiful", "High-Contrast", "Gstaff/Xkcd",
+    "NoCrypt/Miku"
+]
+
+# 默认的系统提示词（system prompt）
+INIT_SYS_PROMPT = "Serve me as a writing and programming assistant."
+
+# 对话窗的高度 （仅在LAYOUT="TOP-DOWN"时生效）
+CHATBOT_HEIGHT = 1115
+
+# 代码高亮
+CODE_HIGHLIGHT = True
+
+# 窗口布局
+LAYOUT = "LEFT-RIGHT"  # "LEFT-RIGHT"（左右布局） # "TOP-DOWN"（上下布局）
+
+# 暗色模式 / 亮色模式
+DARK_MODE = True
+
+# 发送请求到OpenAI后，等待多久判定为超时
+TIMEOUT_SECONDS = 30
+
+# 网页的端口, -1代表随机端口
+# WEB_PORT = -1
+WEB_PORT = 8520
+
+# 如果OpenAI不响应（网络卡顿、代理失败、KEY失效），重试的次数限制
+MAX_RETRY = 2
+
+# 插件分类默认选项
+DEFAULT_FN_GROUPS = ['对话', '编程', '学术', '智能体']
+
+# 定义界面上“询问多个GPT模型”插件应该使用哪些模型，请从AVAIL_LLM_MODELS中选择，并在不同模型之间用`&`间隔，例如"gpt-3.5-turbo&chatglm3&azure-gpt-4"
+MULTI_QUERY_LLM_MODELS = "gpt-3.5-turbo&chatglm3"
+
+# 选择本地模型变体（只有当AVAIL_LLM_MODELS包含了对应本地模型时，才会起作用）
+# 如果你选择Qwen系列的模型，那么请在下面的QWEN_MODEL_SELECTION中指定具体的模型
+# 也可以是具体的模型路径
+QWEN_LOCAL_MODEL_SELECTION = "Qwen/Qwen-1_8B-Chat-Int8"
+
+# 接入通义千问在线大模型 https://dashscope.console.aliyun.com/
+DASHSCOPE_API_KEY = ""  # 阿里灵积云API_KEY
+
+# 百度千帆（LLM_MODEL="qianfan"）
+BAIDU_CLOUD_API_KEY = ''
+BAIDU_CLOUD_SECRET_KEY = ''
+BAIDU_CLOUD_QIANFAN_MODEL = 'ERNIE-Bot'  # 可选 "ERNIE-Bot-4"(文心大模型4.0), "ERNIE-Bot"(文心一言), "ERNIE-Bot-turbo", "BLOOMZ-7B", "Llama-2-70B-Chat", "Llama-2-13B-Chat", "Llama-2-7B-Chat"
+
+# 如果使用ChatGLM2微调模型，请把 LLM_MODEL="chatglmft"，并在此处指定模型路径
+CHATGLM_PTUNING_CHECKPOINT = ""  # 例如"/home/hmp/ChatGLM2-6B/ptuning/output/6b-pt-128-1e-2/checkpoint-100"
+
+# 本地LLM模型如ChatGLM的执行方式 CPU/GPU
+LOCAL_MODEL_DEVICE = "cpu"  # 可选 "cuda"
+LOCAL_MODEL_QUANT = "FP16"  # 默认 "FP16" "INT4" 启用量化INT4版本 "INT8" 启用量化INT8版本
+
+# 设置gradio的并行线程数（不需要修改）
+CONCURRENT_COUNT = 100
+
+# 是否在提交时自动清空输入框
+AUTO_CLEAR_TXT = False
+
+# 加一个live2d装饰
+ADD_WAIFU = False
+
+# 设置用户名和密码（不需要修改）（相关功能不稳定，与gradio版本和网络都相关，如果本地使用不建议加这个）
+# [("username", "password"), ("username2", "password2"), ...]
+AUTHENTICATION = []
+
+# 如果需要在二级路径下运行（常规情况下，不要修改!!）（需要配合修改main.py才能生效!）
+CUSTOM_PATH = "/"
+
+# HTTPS 秘钥和证书（不需要修改）
+SSL_KEYFILE = ""
+SSL_CERTFILE = ""
+
+# 极少数情况下，openai的官方KEY需要伴随组织编码（格式如org-xxxxxxxxxxxxxxxxxxxxxxxx）使用
+API_ORG = ""
+
+# 如果需要使用Slack Claude，使用教程详情见 request_llms/README.md
+SLACK_CLAUDE_BOT_ID = ''
+SLACK_CLAUDE_USER_TOKEN = ''
+
+# 如果需要使用AZURE（方法一：单个azure模型部署）详情请见额外文档 docs\use_azure.md
+AZURE_ENDPOINT = "https://你亲手写的api名称.openai.azure.com/"
+AZURE_API_KEY = "填入azure openai api的密钥"  # 建议直接在API_KEY处填写，该选项即将被弃用
+AZURE_ENGINE = "填入你亲手写的部署名"  # 读 docs\use_azure.md
+
+# 如果需要使用AZURE（方法二：多个azure模型部署+动态切换）详情请见额外文档 docs\use_azure.md
+AZURE_CFG_ARRAY = {}
+
+# 阿里云实时语音识别 配置难度较高
+# 参考 https://github.com/binary-husky/gpt_academic/blob/master/docs/use_audio.md
+ENABLE_AUDIO = False
+ALIYUN_TOKEN = ""  # 例如 f37f30e0f9934c34a992f6f64f7eba4f
+ALIYUN_APPKEY = ""  # 例如 RoPlZrM88DnAFkZK
+ALIYUN_ACCESSKEY = ""  # （无需填写）
+ALIYUN_SECRET = ""  # （无需填写）
+
+# 接入讯飞星火大模型 https://console.xfyun.cn/services/iat
+XFYUN_APPID = "00000000"
+XFYUN_API_SECRET = "bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb"
+XFYUN_API_KEY = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
+
+# 接入智谱大模型
+ZHIPUAI_API_KEY = ""
+ZHIPUAI_MODEL = ""  # 此选项已废弃，不再需要填写
+
+# Claude API KEY
+ANTHROPIC_API_KEY = ""
+
+# 月之暗面 API KEY
+MOONSHOT_API_KEY = ""
+
+# Mathpix 拥有执行PDF的OCR功能，但是需要注册账号
+MATHPIX_APPID = ""
+MATHPIX_APPKEY = ""
+
+# 自定义API KEY格式
+CUSTOM_API_KEY_PATTERN = "[a-zA-Z0-9]+[_-][a-zA-Z0-9]+"
+
+# Google Gemini API-Key
+GEMINI_API_KEY = ''
+
+# HUGGINGFACE的TOKEN，下载LLAMA时起作用 https://huggingface.co/docs/hub/security-tokens
+HUGGINGFACE_ACCESS_TOKEN = "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"
+
+# GROBID服务器地址（填写多个可以均衡负载），用于高质量地读取PDF文档
+# 获取方法：复制以下空间https://huggingface.co/spaces/qingxu98/grobid，设为public，然后GROBID_URL = "https://(你的hf用户名如qingxu98)-(你的填写的空间名如grobid).hf.space"
+GROBID_URLS = [
+    "https://qingxu98-grobid.hf.space",
+    "https://qingxu98-grobid2.hf.space",
+    "https://qingxu98-grobid3.hf.space",
+    "https://qingxu98-grobid4.hf.space",
+    "https://qingxu98-grobid5.hf.space",
+    "https://qingxu98-grobid6.hf.space",
+    "https://qingxu98-grobid7.hf.space",
+    "https://qingxu98-grobid8.hf.space",
+]
+
+# 是否允许通过自然语言描述修改本页的配置，该功能具有一定的危险性，默认关闭
+ALLOW_RESET_CONFIG = False
+
+# 在使用AutoGen插件时，是否使用Docker容器运行代码
+AUTOGEN_USE_DOCKER = False
+
+# 临时的上传文件夹位置，请勿修改
+PATH_PRIVATE_UPLOAD = "private_upload"
+
+# 日志文件夹的位置，请勿修改
+PATH_LOGGING = "gpt_log"
+
+# 除了连接OpenAI之外，还有哪些场合允许使用代理，请勿修改
+WHEN_TO_USE_PROXY = [
+    "Download_LLM", "Download_Gradio_Theme", "Connect_Grobid",
+    "Warmup_Modules", "Nougat_Download", "AutoGen"
+]
+
+# *实验性功能*: 自动检测并屏蔽失效的KEY，请勿使用
+BLOCK_INVALID_APIKEY = False
+
+# 启用插件热加载
+PLUGIN_HOT_RELOAD = False
+
+# 自定义按钮的最大数量限制
+NUM_CUSTOM_BASIC_BTN = 4
+"""
+--------------- 配置关联关系说明 ---------------
+
+在线大模型配置关联关系示意图
+│
+├── "gpt-3.5-turbo" 等openai模型
+│   ├── API_KEY
+│   ├── CUSTOM_API_KEY_PATTERN（不常用）
+│   ├── API_ORG（不常用）
+│   └── API_URL_REDIRECT（不常用）
+│
+├── "azure-gpt-3.5" 等azure模型（单个azure模型，不需要动态切换）
+│   ├── API_KEY
+│   ├── AZURE_ENDPOINT
+│   ├── AZURE_API_KEY
+│   ├── AZURE_ENGINE
+│   └── API_URL_REDIRECT
+│
+├── "azure-gpt-3.5" 等azure模型（多个azure模型，需要动态切换，高优先级）
+│   └── AZURE_CFG_ARRAY
+│
+├── "spark" 星火认知大模型 spark & sparkv2
+│   ├── XFYUN_APPID
+│   ├── XFYUN_API_SECRET
+│   └── XFYUN_API_KEY
+│
+├── "claude-3-opus-20240229" 等claude模型
+│   └── ANTHROPIC_API_KEY
+│
+├── "stack-claude"
+│   ├── SLACK_CLAUDE_BOT_ID
+│   └── SLACK_CLAUDE_USER_TOKEN
+│
+├── "qianfan" 百度千帆大模型库
+│   ├── BAIDU_CLOUD_QIANFAN_MODEL
+│   ├── BAIDU_CLOUD_API_KEY
+│   └── BAIDU_CLOUD_SECRET_KEY
+│
+├── "glm-4", "glm-3-turbo", "zhipuai" 智谱AI大模型
+│   └── ZHIPUAI_API_KEY
+│
+├── "qwen-turbo" 等通义千问大模型
+│   └──  DASHSCOPE_API_KEY
+│
+├── "Gemini"
+│   └──  GEMINI_API_KEY
+│
+└── "one-api-...(max_token=...)" 用一种更方便的方式接入one-api多模型管理界面
+    ├── AVAIL_LLM_MODELS
+    ├── API_KEY
+    └── API_URL_REDIRECT
+
+
+本地大模型示意图
+│
+├── "chatglm3"
+├── "chatglm"
+├── "chatglm_onnx"
+├── "chatglmft"
+├── "internlm"
+├── "moss"
+├── "jittorllms_pangualpha"
+├── "jittorllms_llama"
+├── "deepseekcoder"
+├── "qwen-local"
+├──  RWKV的支持见Wiki
+└── "llama2"
+
+
+用户图形界面布局依赖关系示意图
+│
+├── CHATBOT_HEIGHT 对话窗的高度
+├── CODE_HIGHLIGHT 代码高亮
+├── LAYOUT 窗口布局
+├── DARK_MODE 暗色模式 / 亮色模式
+├── DEFAULT_FN_GROUPS 插件分类默认选项
+├── THEME 色彩主题
+├── AUTO_CLEAR_TXT 是否在提交时自动清空输入框
+├── ADD_WAIFU 加一个live2d装饰
+└── ALLOW_RESET_CONFIG 是否允许通过自然语言描述修改本页的配置，该功能具有一定的危险性
+
+
+插件在线服务配置依赖关系示意图
+│
+├── 语音功能
+│   ├── ENABLE_AUDIO
+│   ├── ALIYUN_TOKEN
+│   ├── ALIYUN_APPKEY
+│   ├── ALIYUN_ACCESSKEY
+│   └── ALIYUN_SECRET
+│
+└── PDF文档精准解析
+    ├── GROBID_URLS
+    ├── MATHPIX_APPID
+    └── MATHPIX_APPKEY
+
+
+"""

core_functional.py

@@ -7,43 +7,94 @@ from toolbox import apply_gpt_academic_string_mask_langbased
 from toolbox import build_gpt_academic_masked_string_langbased
 from textwrap import dedent
 
+
 def get_core_functions():
     return {
-
+        "中文学位论文润色": {
+            "Prefix":
+            r"请对以下的博士论文进行润色，使其符合中国博士论文写作的规范。请注意以下几点要求：" + "\n"
+            r"1. 保证用词符合学术规范，使用专业、精准的术语； " + "\n"
+            r"2. 避免使用第一人称词汇，如“我们”，应当使用被动语态或无主语句型以维持客观性；" + "\n"
+            r"3. 语句应严谨、逻辑清晰，避免产生歧义；" + "\n"
+            r"4. 对于技术术语及概念描述需准确无误；" + "\n"
+            r"5. 结构上要体现出论文的原创性、逻辑性和科学性；" + "\n"
+            r"6. 确保润色的内容具有条理性，既突出理论深度也强调技术创新；" + "\n"
+            r"7. 确保润色符合中国博士论文格式与学术道德规范。以下是我的论文：" + "\n\n",
+            "Suffix":
+            r"",
+            # [3] 按钮颜色 (可选参数，默认 secondary)
+            "Color":
+            r"secondary",
+            # [4] 按钮是否可见 (可选参数，默认 True，即可见)
+            "Visible":
+            True,
+            # [5] 是否在触发时清除历史 (可选参数，默认 False，即不处理之前的对话历史)
+            "AutoClearHistory":
+            False,
+            # [6] 文本预处理 （可选参数，默认 None，举例：写个函数移除所有的换行符）
+            "PreProcess":
+            None,
+        },
+        "中文学位论文降重": {
+            "Prefix":
+            r"尊敬的语言模型，我需要您的帮助来改写以下这段论文内容，以降低其与原文的相似度。请确保在保持原意不变的前提下，使用不同的词汇、语法结构和表达方式。"
+            + "\n\n",
+            "Suffix":
+            "\n\n" + r"请对我的原文进行深度改写，确保：" + "\n" + r"1. 保持原文的意思和论点不变。" + "\n" +
+            r"2. 尽可能使用不同的词汇和表达方式。" + "\n" + r"3. 优化句子结构，使其更加清晰和准确。" + "\n" +
+            r"4. 避免使用任何可能引起抄袭嫌疑的相似表述。",
+            # [3] 按钮颜色 (可选参数，默认 secondary)
+            "Color":
+            r"secondary",
+            # [4] 按钮是否可见 (可选参数，默认 True，即可见)
+            "Visible":
+            True,
+            # [5] 是否在触发时清除历史 (可选参数，默认 False，即不处理之前的对话历史)
+            "AutoClearHistory":
+            False,
+            # [6] 文本预处理 （可选参数，默认 None，举例：写个函数移除所有的换行符）
+            "PreProcess":
+            None,
+        },
         "学术语料润色": {
             # [1*] 前缀字符串，会被加在你的输入之前。例如，用来描述你的要求，例如翻译、解释代码、润色等等。
             #      这里填一个提示词字符串就行了，这里为了区分中英文情景搞复杂了一点
-            "Prefix":   build_gpt_academic_masked_string_langbased(
-                            text_show_english=
-                                r"Below is a paragraph from an academic paper. Polish the writing to meet the academic style, "
-                                r"improve the spelling, grammar, clarity, concision and overall readability. When necessary, rewrite the whole sentence. "
-                                r"Firstly, you should provide the polished paragraph. "
-                                r"Secondly, you should list all your modification and explain the reasons to do so in markdown table.",
-                            text_show_chinese=
-                                r"作为一名中文学术论文写作改进助理，你的任务是改进所提供文本的拼写、语法、清晰、简洁和整体可读性，"
-                                r"同时分解长句，减少重复，并提供改进建议。请先提供文本的更正版本，然后在markdown表格中列出修改的内容，并给出修改的理由:"
-                        ) + "\n\n",
+            "Prefix":
+            build_gpt_academic_masked_string_langbased(
+                text_show_english=
+                r"Below is a paragraph from an academic paper. Polish the writing to meet the academic style, "
+                r"improve the spelling, grammar, clarity, concision and overall readability. When necessary, rewrite the whole sentence. "
+                r"Firstly, you should provide the polished paragraph. "
+                r"Secondly, you should list all your modification and explain the reasons to do so in markdown table.",
+                text_show_chinese=
+                r"作为一名中文学术论文写作改进助理���你的任务是改进所提供文本的拼写、语法、清晰、简洁和整体可读性，"
+                r"同时分解长句，减少重复，并提供改进建议。请先提供文本的更正版本，然后在markdown表格中列出修改的内容，并给出修改的理由:"
+            ) + "\n\n",
             # [2*] 后缀字符串，会被加在你的输入之后。例如，配合前缀可以把你的输入内容用引号圈起来
-            "Suffix":   r"",
+            "Suffix":
+            r"",
             # [3] 按钮颜色 (可选参数，默认 secondary)
-            "Color":    r"secondary",
+            "Color":
+            r"secondary",
             # [4] 按钮是否可见 (可选参数，默认 True，即可见)
-            "Visible": True,
+            "Visible":
+            True,
             # [5] 是否在触发时清除历史 (可选参数，默认 False，即不处理之前的对话历史)
-            "AutoClearHistory": False,
+            "AutoClearHistory":
+            False,
             # [6] 文本预处理 （可选参数，默认 None，举例：写个函数移除所有的换行符）
-            "PreProcess": None,
+            "PreProcess":
+            None,
         },
-        
-        
         "总结绘制脑图": {
             # 前缀，会被加在你的输入之前。例如，用来描述你的要求，例如翻译、解释代码、润色等等
-            "Prefix":   r"",
+            "Prefix":
+            '''"""\n\n''',
             # 后缀，会被加在你的输入之后。例如，配合前缀可以把你的输入内容用引号圈起来
             "Suffix":
-                # dedent() 函数用于去除多行字符串的缩进
-                dedent("\n"+r'''
-                    ==============================
+            # dedent() 函数用于去除多行字符串的缩进
+            dedent("\n\n" + r'''
+                    """
 
                     使用mermaid flowchart对以上文本进行总结，概括上述段落的内容以及内在逻辑关系，例如：
 
@@ -57,117 +108,124 @@ def get_core_functions():
                         C --> |"箭头名2"| F["节点名6"]
                     ```
 
-                    警告：
+                    注意：
                     （1）使用中文
                     （2）节点名字使用引号包裹，如["Laptop"]
                     （3）`|` 和 `"`之间不要存在空格
                     （4）根据情况选择flowchart LR（从左到右）或者flowchart TD（从上到下）
                 '''),
         },
-        
-        
         "查找语法错误": {
-            "Prefix":   r"Help me ensure that the grammar and the spelling is correct. "
-                        r"Do not try to polish the text, if no mistake is found, tell me that this paragraph is good. "
-                        r"If you find grammar or spelling mistakes, please list mistakes you find in a two-column markdown table, "
-                        r"put the original text the first column, "
-                        r"put the corrected text in the second column and highlight the key words you fixed. "
-                        r"Finally, please provide the proofreaded text.""\n\n"
-                        r"Example:""\n"
-                        r"Paragraph: How is you? Do you knows what is it?""\n"
-                        r"| Original sentence | Corrected sentence |""\n"
-                        r"| :--- | :--- |""\n"
-                        r"| How **is** you? | How **are** you? |""\n"
-                        r"| Do you **knows** what **is** **it**? | Do you **know** what **it** **is** ? |""\n\n"
-                        r"Below is a paragraph from an academic paper. "
-                        r"You need to report all grammar and spelling mistakes as the example before."
-                        + "\n\n",
-            "Suffix":   r"",
-            "PreProcess": clear_line_break,    # 预处理：清除换行符
+            "Prefix":
+            r"Help me ensure that the grammar and the spelling is correct. "
+            r"Do not try to polish the text, if no mistake is found, tell me that this paragraph is good. "
+            r"If you find grammar or spelling mistakes, please list mistakes you find in a two-column markdown table, "
+            r"put the original text the first column, "
+            r"put the corrected text in the second column and highlight the key words you fixed. "
+            r"Finally, please provide the proofreaded text."
+            "\n\n"
+            r"Example:"
+            "\n"
+            r"Paragraph: How is you? Do you knows what is it?"
+            "\n"
+            r"| Original sentence | Corrected sentence |"
+            "\n"
+            r"| :--- | :--- |"
+            "\n"
+            r"| How **is** you? | How **are** you? |"
+            "\n"
+            r"| Do you **knows** what **is** **it**? | Do you **know** what **it** **is** ? |"
+            "\n\n"
+            r"Below is a paragraph from an academic paper. "
+            r"You need to report all grammar and spelling mistakes as the example before."
+            + "\n\n",
+            "Suffix":
+            r"",
+            "PreProcess":
+            clear_line_break,  # 预处理：清除换行符
         },
-        
-        
         "中译英": {
-            "Prefix":   r"Please translate following sentence to English:" + "\n\n",
-            "Suffix":   r"",
+            "Prefix":
+            r"Please translate following sentence to English:" + "\n\n",
+            "Suffix": r"",
         },
-        
-        
         "学术英中互译": {
-            "Prefix":   build_gpt_academic_masked_string_langbased(
-                            text_show_chinese=
-                                r"I want you to act as a scientific English-Chinese translator, "
-                                r"I will provide you with some paragraphs in one language "
-                                r"and your task is to accurately and academically translate the paragraphs only into the other language. "
-                                r"Do not repeat the original provided paragraphs after translation. "
-                                r"You should use artificial intelligence tools, "
-                                r"such as natural language processing, and rhetorical knowledge "
-                                r"and experience about effective writing techniques to reply. "
-                                r"I'll give you my paragraphs as follows, tell me what language it is written in, and then translate:",
-                            text_show_english=
-                                r"你是经验丰富的翻译，请把以下学术文章段落翻译成中文，"
-                                r"并同时充分考虑中文的语法、清晰、简洁和整体可读性，"
-                                r"必要时，你可以修改整个句子的顺序以确保翻译后的段落符合中文的语言习惯。"
-                                r"你需要翻译的文本如下："
-                        ) + "\n\n",
-            "Suffix":   r"",
+            "Prefix":
+            build_gpt_academic_masked_string_langbased(
+                text_show_chinese=
+                r"I want you to act as a scientific English-Chinese translator, "
+                r"I will provide you with some paragraphs in one language "
+                r"and your task is to accurately and academically translate the paragraphs only into the other language. "
+                r"Do not repeat the original provided paragraphs after translation. "
+                r"You should use artificial intelligence tools, "
+                r"such as natural language processing, and rhetorical knowledge "
+                r"and experience about effective writing techniques to reply. "
+                r"I'll give you my paragraphs as follows, tell me what language it is written in, and then translate:",
+                text_show_english=r"你是经验丰富的翻译，请把以下学术文章段落翻译成中文，"
+                r"并同时充分考虑中文的语法、清晰、简洁和整体可读性，"
+                r"必要时，你可以修改整个句子的顺序以确保翻译后的段落符合中文的语言习惯。"
+                r"你需要翻译的文本如下：") + "\n\n",
+            "Suffix":
+            r"",
         },
-        
-        
         "英译中": {
-            "Prefix":   r"翻译成地道的中文：" + "\n\n",
-            "Suffix":   r"",
-            "Visible":  False,
+            "Prefix": r"翻译成地道的中文：" + "\n\n",
+            "Suffix": r"",
+            "Visible": False,
         },
-        
-        
         "找图片": {
-            "Prefix":   r"我需要你找一张网络图片。使用Unsplash API(https://source.unsplash.com/960x640/?<英语关键词>)获取图片URL，"
-                        r"然后请使用Markdown格式封装，并且不要有反斜线，不要用代码块。现在，请按以下描述给我发送图片：" + "\n\n",
-            "Suffix":   r"",
-            "Visible":  False,
+            "Prefix":
+            r"我需要你找一张网络图片。使用Unsplash API(https://source.unsplash.com/960x640/?<英语关键词>)获取图片URL，"
+            r"然后请使用Markdown格式封装，并且不要有反斜线，不要用代码块。现在，请按以下描述给我发送图片：" + "\n\n",
+            "Suffix":
+            r"",
+            "Visible":
+            False,
         },
-        
-        
         "解释代码": {
-            "Prefix":   r"请解释以下代码：" + "\n```\n",
-            "Suffix":   "\n```\n",
+            "Prefix": r"请解释以下代码：" + "\n```\n",
+            "Suffix": "\n```\n",
         },
-        
-        
         "参考文献转Bib": {
-            "Prefix":   r"Here are some bibliography items, please transform them into bibtex style."
-                        r"Note that, reference styles maybe more than one kind, you should transform each item correctly."
-                        r"Items need to be transformed:" + "\n\n",
-            "Visible":  False,
-            "Suffix":   r"",
+            "Prefix":
+            r"Here are some bibliography items, please transform them into bibtex style."
+            r"Note that, reference styles maybe more than one kind, you should transform each item correctly."
+            r"Items need to be transformed:" + "\n\n",
+            "Visible":
+            False,
+            "Suffix":
+            r"",
         }
     }
 
 
 def handle_core_functionality(additional_fn, inputs, history, chatbot):
     import core_functional
-    importlib.reload(core_functional)    # 热更新prompt
+    importlib.reload(core_functional)  # 热更新prompt
     core_functional = core_functional.get_core_functions()
     addition = chatbot._cookies['customize_fn_overwrite']
     if additional_fn in addition:
         # 自定义功能
-        inputs = addition[additional_fn]["Prefix"] + inputs + addition[additional_fn]["Suffix"]
+        inputs = addition[additional_fn]["Prefix"] + inputs + addition[
+            additional_fn]["Suffix"]
         return inputs, history
     else:
         # 预制功能
         if "PreProcess" in core_functional[additional_fn]:
             if core_functional[additional_fn]["PreProcess"] is not None:
-                inputs = core_functional[additional_fn]["PreProcess"](inputs)  # 获取预处理函数（如果有的话）
+                inputs = core_functional[additional_fn]["PreProcess"](
+                    inputs)  # 获取预处理函数（如果有的话）
         # 为字符串加上上面定义的前缀和后缀。
         inputs = apply_gpt_academic_string_mask_langbased(
-            string = core_functional[additional_fn]["Prefix"] + inputs + core_functional[additional_fn]["Suffix"],
-            lang_reference = inputs,
+            string=core_functional[additional_fn]["Prefix"] + inputs +
+            core_functional[additional_fn]["Suffix"],
+            lang_reference=inputs,
         )
         if core_functional[additional_fn].get("AutoClearHistory", False):
             history = []
         return inputs, history
 
+
 if __name__ == "__main__":
     t = get_core_functions()["总结绘制脑图"]
-    print(t["Prefix"] + t["Suffix"])
+    print(t["Prefix"] + t["Suffix"])