dall-e-3添加 'style' 风格参数
dall-e-3添加 'style' 风格参数(参考 platform.openai.com/doc/api-reference),修改dall-e-3作图时的参数判断逻辑
This commit is contained in:
parent
ef12d4f754
commit
6126024f2c
@ -2,7 +2,7 @@ from toolbox import CatchException, update_ui, get_conf, select_api_key, get_log
|
|||||||
from crazy_functions.multi_stage.multi_stage_utils import GptAcademicState
|
from crazy_functions.multi_stage.multi_stage_utils import GptAcademicState
|
||||||
|
|
||||||
|
|
||||||
def gen_image(llm_kwargs, prompt, resolution="1024x1024", model="dall-e-2", quality=None):
|
def gen_image(llm_kwargs, prompt, resolution="1024x1024", model="dall-e-2", quality=None, style=None):
|
||||||
import requests, json, time, os
|
import requests, json, time, os
|
||||||
from request_llms.bridge_all import model_info
|
from request_llms.bridge_all import model_info
|
||||||
|
|
||||||
@ -25,7 +25,10 @@ def gen_image(llm_kwargs, prompt, resolution="1024x1024", model="dall-e-2", qual
|
|||||||
'model': model,
|
'model': model,
|
||||||
'response_format': 'url'
|
'response_format': 'url'
|
||||||
}
|
}
|
||||||
if quality is not None: data.update({'quality': quality})
|
if quality is not None:
|
||||||
|
data['quality'] = quality
|
||||||
|
if style is not None:
|
||||||
|
data['style'] = style
|
||||||
response = requests.post(url, headers=headers, json=data, proxies=proxies)
|
response = requests.post(url, headers=headers, json=data, proxies=proxies)
|
||||||
print(response.content)
|
print(response.content)
|
||||||
try:
|
try:
|
||||||
@ -115,13 +118,18 @@ def 图片生成_DALLE3(prompt, llm_kwargs, plugin_kwargs, chatbot, history, sys
|
|||||||
chatbot.append(("您正在调用“图像生成”插件。", "[Local Message] 生成图像, 请先把模型切换至gpt-*或者api2d-*。如果中文Prompt效果不理想, 请尝试英文Prompt。正在处理中 ....."))
|
chatbot.append(("您正在调用“图像生成”插件。", "[Local Message] 生成图像, 请先把模型切换至gpt-*或者api2d-*。如果中文Prompt效果不理想, 请尝试英文Prompt。正在处理中 ....."))
|
||||||
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面 由于请求gpt需要一段时间,我们先及时地做一次界面更新
|
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面 由于请求gpt需要一段时间,我们先及时地做一次界面更新
|
||||||
if ("advanced_arg" in plugin_kwargs) and (plugin_kwargs["advanced_arg"] == ""): plugin_kwargs.pop("advanced_arg")
|
if ("advanced_arg" in plugin_kwargs) and (plugin_kwargs["advanced_arg"] == ""): plugin_kwargs.pop("advanced_arg")
|
||||||
resolution = plugin_kwargs.get("advanced_arg", '1024x1024').lower()
|
resolution_arg = plugin_kwargs.get("advanced_arg", '1024x1024-standard-vivid').lower()
|
||||||
if resolution.endswith('-hd'):
|
parts = resolution_arg.split('-')
|
||||||
resolution = resolution.replace('-hd', '')
|
resolution = parts[0] # 解析分辨率
|
||||||
quality = 'hd'
|
quality = 'standard' # 质量与风格默认值
|
||||||
else:
|
style = 'vivid'
|
||||||
quality = 'standard'
|
# 遍历检查是否有额外参数
|
||||||
image_url, image_path = gen_image(llm_kwargs, prompt, resolution, model="dall-e-3", quality=quality)
|
for part in parts[1:]:
|
||||||
|
if part in ['hd', 'standard']:
|
||||||
|
quality = part
|
||||||
|
elif part in ['vivid', 'natural']:
|
||||||
|
style = part
|
||||||
|
image_url, image_path = gen_image(llm_kwargs, prompt, resolution, model="dall-e-3", quality=quality, style=style)
|
||||||
chatbot.append([prompt,
|
chatbot.append([prompt,
|
||||||
f'图像中转网址: <br/>`{image_url}`<br/>'+
|
f'图像中转网址: <br/>`{image_url}`<br/>'+
|
||||||
f'中转网址预览: <br/><div align="center"><img src="{image_url}"></div>'
|
f'中转网址预览: <br/><div align="center"><img src="{image_url}"></div>'
|
||||||
@ -201,4 +209,3 @@ def 图片修改_DALLE2(prompt, llm_kwargs, plugin_kwargs, chatbot, history, sys
|
|||||||
f'本地文件预览: <br/><div align="center"><img src="file={image_path}"></div>'
|
f'本地文件预览: <br/><div align="center"><img src="file={image_path}"></div>'
|
||||||
])
|
])
|
||||||
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面 界面更新
|
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面 界面更新
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user