[feat] 支持自定义dalle接口prompt前缀

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`

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",

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')