diff --git a/crazy_functional.py b/crazy_functional.py
index bf435be0..df6a2e32 100644
--- a/crazy_functional.py
+++ b/crazy_functional.py
@@ -45,6 +45,8 @@ def get_crazy_functions():
from crazy_functions.Latex_Function_Wrap import PDF_Localize
from crazy_functions.Internet_GPT import 连接网络回答问题
from crazy_functions.Internet_GPT_Wrap import NetworkGPT_Wrap
+ from crazy_functions.Image_Generate import 图片生成_DALLE2, 图片生成_DALLE3, 图片修改_DALLE2
+ from crazy_functions.Image_Generate_Wrap import ImageGen_Wrap
function_plugins = {
"虚空终端": {
@@ -324,7 +326,7 @@ def get_crazy_functions():
"ArgsReminder": "如果有必要, 请在此处追加更细致的矫错指令（使用英文）。",
"Function": HotReload(Latex英文纠错加PDF对比),
},
- "Arxiv论文精细翻译（输入arxivID）[需Latex]": {
+ "📚Arxiv论文精细翻译（输入arxivID）[需Latex]": {
"Group": "学术",
"Color": "stop",
"AsButton": False,
@@ -336,7 +338,7 @@ def get_crazy_functions():
"Function": HotReload(Latex翻译中文并重新编译PDF), # 当注册Class后，Function旧接口仅会在“虚空终端”中起作用
"Class": Arxiv_Localize, # 新一代插件需要注册Class
},
- "本地Latex论文精细翻译（上传Latex项目）[需Latex]": {
+ "📚本地Latex论文精细翻译（上传Latex项目）[需Latex]": {
"Group": "学术",
"Color": "stop",
"AsButton": False,
@@ -361,6 +363,39 @@ def get_crazy_functions():
}
}
+ function_plugins.update(
+ {
+ "🎨图片生成（DALLE2/DALLE3, 使用前切换到GPT系列模型）": {
+ "Group": "对话",
+ "Color": "stop",
+ "AsButton": False,
+ "Info": "使用 DALLE2/DALLE3 生成图片 | 输入参数字符串，提供图像的内容",
+ "Function": HotReload(图片生成_DALLE2), # 当注册Class后，Function旧接口仅会在“虚空终端”中起作用
+ "Class": ImageGen_Wrap # 新一代插件需要注册Class
+ },
+ }
+ )
+
+ function_plugins.update(
+ {
+ "🎨图片修改_DALLE2 （使用前请切换模型到GPT系列）": {
+ "Group": "对话",
+ "Color": "stop",
+ "AsButton": False,
+ "AdvancedArgs": False, # 调用时，唤起高级参数输入区（默认False）
+ # "Info": "使用DALLE2修改图片 | 输入参数字符串，提供图像的内容",
+ "Function": HotReload(图片修改_DALLE2),
+ },
+ }
+ )
+
+
+
+
+
+
+
+
# -=--=- 尚未充分测试的实验性插件 & 需要额外依赖的插件 -=--=-
try:
@@ -450,50 +485,7 @@ def get_crazy_functions():
print(trimmed_format_exc())
print("Load function plugin failed")
- try:
- from crazy_functions.图片生成 import 图片生成_DALLE2, 图片生成_DALLE3, 图片修改_DALLE2
- function_plugins.update(
- {
- "图片生成_DALLE2 （先切换模型到gpt-*）": {
- "Group": "对话",
- "Color": "stop",
- "AsButton": False,
- "AdvancedArgs": True, # 调用时，唤起高级参数输入区（默认False）
- "ArgsReminder": "在这里输入分辨率, 如1024x1024（默认），支持 256x256, 512x512, 1024x1024", # 高级参数输入区的显示提示
- "Info": "使用DALLE2生成图片 | 输入参数字符串，提供图像的内容",
- "Function": HotReload(图片生成_DALLE2),
- },
- }
- )
- function_plugins.update(
- {
- "图片生成_DALLE3 （先切换模型到gpt-*）": {
- "Group": "对话",
- "Color": "stop",
- "AsButton": False,
- "AdvancedArgs": True, # 调用时，唤起高级参数输入区（默认False）
- "ArgsReminder": "在这里输入自定义参数「分辨率-质量(可选)-风格(可选)」, 参数示例「1024x1024-hd-vivid」 || 分辨率支持 「1024x1024」(默认) /「1792x1024」/「1024x1792」 || 质量支持 「-standard」(默认) /「-hd」 || 风格支持 「-vivid」(默认) /「-natural」", # 高级参数输入区的显示提示
- "Info": "使用DALLE3生成图片 | 输入参数字符串，提供图像的内容",
- "Function": HotReload(图片生成_DALLE3),
- },
- }
- )
- function_plugins.update(
- {
- "图片修改_DALLE2 （先切换模型到gpt-*）": {
- "Group": "对话",
- "Color": "stop",
- "AsButton": False,
- "AdvancedArgs": False, # 调用时，唤起高级参数输入区（默认False）
- # "Info": "使用DALLE2修改图片 | 输入参数字符串，提供图像的内容",
- "Function": HotReload(图片修改_DALLE2),
- },
- }
- )
- except:
- print(trimmed_format_exc())
- print("Load function plugin failed")
try:
from crazy_functions.总结音视频 import 总结音视频
diff --git a/crazy_functions/图片生成.py b/crazy_functions/Image_Generate.py
similarity index 96%
rename from crazy_functions/图片生成.py
rename to crazy_functions/Image_Generate.py
index 24d35630..70beec81 100644
--- a/crazy_functions/图片生成.py
+++ b/crazy_functions/Image_Generate.py
@@ -108,7 +108,7 @@ def 图片生成_DALLE2(prompt, llm_kwargs, plugin_kwargs, chatbot, history, sys
chatbot.append((prompt, "[Local Message] 图像生成提示为空白，请在“输入区”输入图像生成提示。"))
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面 界面更新
return
- chatbot.append(("您正在调用“图像生成”插件。", "[Local Message] 生成图像, 请先把模型切换至gpt-*。如果中文Prompt效果不理想, 请尝试英文Prompt。正在处理中 ....."))
+ chatbot.append(("您正在调用“图像生成”插件。", "[Local Message] 生成图像, 使用前请切换模型到GPT系列。如果中文Prompt效果不理想, 请尝试英文Prompt。正在处理中 ....."))
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")
resolution = plugin_kwargs.get("advanced_arg", '1024x1024')
@@ -129,7 +129,7 @@ def 图片生成_DALLE3(prompt, llm_kwargs, plugin_kwargs, chatbot, history, sys
chatbot.append((prompt, "[Local Message] 图像生成提示为空白，请在“输入区”输入图像生成提示。"))
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面 界面更新
return
- chatbot.append(("您正在调用“图像生成”插件。", "[Local Message] 生成图像, 请先把模型切换至gpt-*。如果中文Prompt效果不理想, 请尝试英文Prompt。正在处理中 ....."))
+ chatbot.append(("您正在调用“图像生成”插件。", "[Local Message] 生成图像, 使用前请切换模型到GPT系列。如果中文Prompt效果不理想, 请尝试英文Prompt。正在处理中 ....."))
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")
resolution_arg = plugin_kwargs.get("advanced_arg", '1024x1024-standard-vivid').lower()
@@ -166,7 +166,7 @@ class ImageEditState(GptAcademicState):
return confirm, file
def lock_plugin(self, chatbot):
- chatbot._cookies['lock_plugin'] = 'crazy_functions.图片生成->图片修改_DALLE2'
+ chatbot._cookies['lock_plugin'] = 'crazy_functions.Image_Generate->图片修改_DALLE2'
self.dump_state(chatbot)
def unlock_plugin(self, chatbot):
diff --git a/crazy_functions/Image_Generate_Wrap.py b/crazy_functions/Image_Generate_Wrap.py
new file mode 100644
index 00000000..8cd71efc
--- /dev/null
+++ b/crazy_functions/Image_Generate_Wrap.py
@@ -0,0 +1,56 @@
+
+from toolbox import get_conf, update_ui
+from crazy_functions.Image_Generate import 图片生成_DALLE2, 图片生成_DALLE3, 图片修改_DALLE2
+from crazy_functions.plugin_template.plugin_class_template import GptAcademicPluginTemplate, ArgProperty
+
+
+class ImageGen_Wrap(GptAcademicPluginTemplate):
+ def __init__(self):
+ """
+ 请注意`execute`会执行在不同的线程中，因此您在定义和使用类变量时，应当慎之又慎！
+ """
+ pass
+
+ def define_arg_selection_menu(self):
+ """
+ 定义插件的二级选项菜单
+
+ 第一个参数，名称`main_input`，参数`type`声明这是一个文本框，文本框上方显示`title`，文本框内部显示`description`，`default_value`为默认值；
+ 第二个参数，名称`advanced_arg`，参数`type`声明这是一个文本框，文本框上方显示`title`，文本框内部显示`description`，`default_value`为默认值；
+
+ """
+ gui_definition = {
+ "main_input":
+ ArgProperty(title="输入图片描述", description="需要生成图像的文本描述，尽量使用英文", default_value="", type="string").model_dump_json(), # 主输入，自动从输入框同步
+ "model_name":
+ ArgProperty(title="模型", options=["DALLE2", "DALLE3"], default_value="DALLE3", description="无", type="dropdown").model_dump_json(),
+ "resolution":
+ ArgProperty(title="分辨率", options=["256x256(限DALLE2)", "512x512(限DALLE2)", "1024x1024", "1792x1024(限DALLE3)", "1024x1792(限DALLE3)"], default_value="1024x1024", description="无", type="dropdown").model_dump_json(),
+ "quality (仅DALLE3生效)":
+ ArgProperty(title="质量", options=["standard", "hd"], default_value="standard", description="无", type="dropdown").model_dump_json(),
+ "style (仅DALLE3生效)":
+ ArgProperty(title="风格", options=["vivid", "natural"], default_value="vivid", description="无", type="dropdown").model_dump_json(),
+
+ }
+ return gui_definition
+
+ def execute(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, user_request):
+ """
+ 执行插件
+ """
+ # 分辨率
+ resolution = plugin_kwargs["resolution"].replace("(限DALLE2)", "").replace("(限DALLE3)", "")
+
+ if plugin_kwargs["model_name"] == "DALLE2":
+ plugin_kwargs["advanced_arg"] = resolution
+ yield from 图片生成_DALLE2(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, user_request)
+
+ elif plugin_kwargs["model_name"] == "DALLE3":
+ quality = plugin_kwargs["quality (仅DALLE3生效)"]
+ style = plugin_kwargs["style (仅DALLE3生效)"]
+ plugin_kwargs["advanced_arg"] = f"{resolution}-{quality}-{style}"
+ yield from 图片生成_DALLE3(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, user_request)
+
+ else:
+ chatbot.append([None, "抱歉，找不到该模型"])
+ yield from update_ui(chatbot=chatbot, history=history) # 刷新界面
diff --git a/crazy_functions/game_fns/game_interactive_story.py b/crazy_functions/game_fns/game_interactive_story.py
index 6c528c39..759197dd 100644
--- a/crazy_functions/game_fns/game_interactive_story.py
+++ b/crazy_functions/game_fns/game_interactive_story.py
@@ -92,7 +92,7 @@ class MiniGame_ResumeStory(GptAcademicGameBaseState):
def generate_story_image(self, story_paragraph):
try:
- from crazy_functions.图片生成 import gen_image
+ from crazy_functions.Image_Generate import gen_image
prompt_ = predict_no_ui_long_connection(inputs=story_paragraph, llm_kwargs=self.llm_kwargs, history=[], sys_prompt='你需要根据用户给出的小说段落，进行简短的环境描写。要求：80字以内。')
image_url, image_path = gen_image(self.llm_kwargs, prompt_, '512x512', model="dall-e-2", quality='standard', style='natural')
return f'
'