From 2e0ef2bc3a5f56e36f45e9df3b1157ebe69ab7b6 Mon Sep 17 00:00:00 2001
From: Wizerd <Wizerd@recieved.email>
Date: Thu, 4 Jan 2024 22:42:00 +0800
Subject: [PATCH] =?UTF-8?q?[feat]=20=E6=94=AF=E6=8C=81=E8=87=AA=E5=AE=9A?=
 =?UTF-8?q?=E4=B9=89dalle=E6=8E=A5=E5=8F=A3prompt=E5=89=8D=E7=BC=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Readme.md        | 2 ++
 data/config.json | 1 +
 main.py          | 9 +++++++--
 3 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/Readme.md b/Readme.md
index 8c56d17..42f09ea 100644
--- a/Readme.md
+++ b/Readme.md
@@ -119,6 +119,8 @@
 
 - `arkose_urls`: Arkose Token获取地址，如果 `custom_arkose_url` 为 `false`，则该参数无效，如果 `custom_arkose_url` 为 `true`，则该参数必填，且需要填写为可以获取Arkose Token的地址列表，例如：`https://arkose-proxy-1.pandoranext.com/<proxy-prefix>,https://arkose-proxy-2.pandoranext.com/<proxy-prefix>`，支持同时设置多个Arkose Token获取地址，从前往后轮询调用，如果第一个获取失败则自动从第二个获取，以此类推。
 
+- `dalle_prompt_prefix`: 自定义的DALLE接口prompt前缀，可以引导gpt完成绘图任务。
+
 PS. 注意，arkose_urls中的地址需要支持PandoraNext的Arkose Token获取路径与参数，并与PandoraNext的Arkose Token获取接口的响应格式保持一致。
 
 - `bot_mode`
diff --git a/data/config.json b/data/config.json
index 13bdaad..0171cbb 100644
--- a/data/config.json
+++ b/data/config.json
@@ -15,6 +15,7 @@
     "use_oaiusercontent_url": "false",
     "custom_arkose_url": "false",
     "arkose_urls": "",
+    "dalle_prompt_prefix": "请严格根据我的以下要求完成绘图任务，如果我没有发出指定的绘画指令，则绘制出我发出的文字对应的图片：",
     "bot_mode": {
         "enabled": "false",
         "enabled_markdown_image_output": "false",
diff --git a/main.py b/main.py
index 9eb0755..82b8a91 100644
--- a/main.py
+++ b/main.py
@@ -62,6 +62,7 @@ CUSTOM_ARKOSE = CONFIG.get('custom_arkose_url', 'false').lower() == 'true'
 
 ARKOSE_URLS = CONFIG.get('arkose_urls', "")
 
+DALLE_PROMPT_PREFIX = CONFIG.get('dalle_prompt_prefix', '')
 
 # 设置日志级别
 log_level_dict = {
@@ -180,9 +181,9 @@ CORS(app, resources={r"/images/*": {"origins": "*"}})
 PANDORA_UPLOAD_URL = 'files.pandoranext.com'
 
 
-VERSION = '0.4.3'
+VERSION = '0.4.4'
 # VERSION = 'test'
-UPDATE_INFO = '修复各种文件生成的bug'
+UPDATE_INFO = '支持自定义DALLE绘图接口prompt前缀'
 # UPDATE_INFO = '【仅供临时测试使用】 '
 
 with app.app_context():
@@ -259,6 +260,8 @@ with app.app_context():
     if CUSTOM_ARKOSE:
         logger.info(f"arkose_urls: {ARKOSE_URLS}")
 
+    logger.info(f"DALLE_prompt_prefix: {DALLE_PROMPT_PREFIX}")
+
     logger.info(f"==========================================")
 
     
@@ -1583,6 +1586,8 @@ def images_generations():
     
     prompt = data.get('prompt', '')
 
+    prompt = DALLE_PROMPT_PREFIX + prompt
+
     # 获取请求中的response_format参数，默认为"url"
     response_format = data.get('response_format', 'url')