v1.1.9: 升级LoadKontextPresets_UTK节点，新增用户提示输入功能，支持动态场景描述
This commit is contained in:
Cyber Dick Lang
+36
-1
@@ -8,13 +8,38 @@ A comprehensive toolkit for ComfyUI that provides various utility nodes for imag
|
||||
:license: MIT, see LICENSE for more details.
|
||||
"""
|
||||
|
||||
__version__ = "1.1.7"
|
||||
__version__ = "1.1.9"
|
||||
__author__ = "CyberDickLang"
|
||||
__email__ = "286878701@qq.com"
|
||||
__url__ = "https://github.com/whmc76"
|
||||
|
||||
# 更新日志
|
||||
CHANGELOG = {
|
||||
"1.1.9": [
|
||||
"升级 LoadKontextPresets_UTK 节点功能：",
|
||||
"- 新增用户提示输入参数（user_prompt），支持多行文本输入",
|
||||
"- 为所有27个预设添加$user_prompt$占位符，支持动态场景描述",
|
||||
"- 改造所有预设的brief描述，使其能够根据用户输入生成具体指令",
|
||||
"- 智能占位符替换：有用户输入时替换为具体需求，无输入时使用通用描述",
|
||||
"- 支持重新布光、场景传送、季节变换等预设的用户自定义场景",
|
||||
"- 支持人物变换、艺术风格、特殊效果等预设的个性化需求",
|
||||
"- 提升VLM模型生成指令的针对性和实用性",
|
||||
"- 保持原有27个预设的完整功能和分类结构"
|
||||
],
|
||||
"1.1.8": [
|
||||
"新增 LoadKontextPresets_UTK 节点（Kontext VLM System Presets）：",
|
||||
"- 基于 Kontext 项目集成，提供25种图像变换预设",
|
||||
"- 支持情境深度融合、无痕融合、场景传送等多种预设",
|
||||
"- 支持移动镜头、重新布光、专业产品图等专业预设",
|
||||
"- 支持画面缩放、图像上色、电影海报等创意预设",
|
||||
"- 支持卡通漫画化、移除文字、更换发型等功能预设",
|
||||
"- 支持肌肉猛男化、清空家具、室内设计等特殊预设",
|
||||
"- 支持季节变换、时光旅人、材质置换等艺术预设",
|
||||
"- 支持微缩世界、幻想领域、衣橱改造等创意预设",
|
||||
"- 支持艺术风格模仿、蓝图视角、添加倒影等效果预设",
|
||||
"- 支持像素艺术、铅笔手绘、油画风格等艺术风格预设",
|
||||
"- 所有预设均按照官方指南进行重写和升级，提供专业的图像变换指令"
|
||||
],
|
||||
"1.1.7": [
|
||||
"新增 ThinkRemover_UTK 节点：",
|
||||
"- 支持分离文本中的<think>内容和剩余内容，便于上下文处理和提示词优化"
|
||||
@@ -325,6 +350,13 @@ except ImportError as e:
|
||||
THINK_REMOVER_MAPPINGS = {}
|
||||
THINK_REMOVER_DISPLAY_MAPPINGS = {}
|
||||
|
||||
try:
|
||||
from .nodes.tools.kontext_presets import NODE_CLASS_MAPPINGS as KONTEXT_PRESETS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as KONTEXT_PRESETS_DISPLAY_MAPPINGS
|
||||
except ImportError as e:
|
||||
print(f"导入错误: {e}")
|
||||
KONTEXT_PRESETS_MAPPINGS = {}
|
||||
KONTEXT_PRESETS_DISPLAY_MAPPINGS = {}
|
||||
|
||||
# 合并所有节点映射
|
||||
NODE_CLASS_MAPPINGS = {}
|
||||
NODE_CLASS_MAPPINGS.update(EMPTY_UNIT_MAPPINGS)
|
||||
@@ -353,6 +385,7 @@ NODE_CLASS_MAPPINGS.update(TEXTBOX_MAPPINGS)
|
||||
NODE_CLASS_MAPPINGS.update(TEXT_CONCATENATE_MAPPINGS)
|
||||
NODE_CLASS_MAPPINGS.update(MATH_EXPRESSION_MAPPINGS)
|
||||
NODE_CLASS_MAPPINGS.update(THINK_REMOVER_MAPPINGS)
|
||||
NODE_CLASS_MAPPINGS.update(KONTEXT_PRESETS_MAPPINGS)
|
||||
|
||||
# 合并显示名称映射
|
||||
NODE_DISPLAY_NAME_MAPPINGS = {}
|
||||
@@ -382,6 +415,7 @@ NODE_DISPLAY_NAME_MAPPINGS.update(TEXTBOX_DISPLAY_MAPPINGS)
|
||||
NODE_DISPLAY_NAME_MAPPINGS.update(TEXT_CONCATENATE_DISPLAY_MAPPINGS)
|
||||
NODE_DISPLAY_NAME_MAPPINGS.update(MATH_EXPRESSION_DISPLAY_MAPPINGS)
|
||||
NODE_DISPLAY_NAME_MAPPINGS.update(THINK_REMOVER_DISPLAY_MAPPINGS)
|
||||
NODE_DISPLAY_NAME_MAPPINGS.update(KONTEXT_PRESETS_DISPLAY_MAPPINGS)
|
||||
|
||||
NODE_CATEGORIES = {
|
||||
"UniversalToolkit": [
|
||||
@@ -413,6 +447,7 @@ NODE_CATEGORIES = {
|
||||
"TextConcatenate_UTK",
|
||||
"MathExpression_UTK",
|
||||
"ThinkRemover_UTK",
|
||||
"LoadKontextPresets_UTK",
|
||||
]
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,181 @@
|
||||
import json
|
||||
|
||||
class LoadKontextPresets_UTK:
|
||||
data = {
|
||||
"prefix": "You are a creative prompt engineer. Your mission is to analyze the provided image and generate exactly 1 distinct image transformation *instructions*.",
|
||||
"presets": [
|
||||
# === 拼接/合成图片处理类 ===
|
||||
{
|
||||
"name": "Context Deep Fusion (情境深度融合)",
|
||||
"brief": "The provided image is a composite with a head and body from drastically different contexts (lighting, style, condition). Your mission is to generate instructions for a complete narrative and physical transformation of the head to flawlessly match the body and scene. The instructions must guide the AI to: 1. **Cinematic Re-Lighting**: Describe in vivid detail the scene's light sources (color, direction, hardness) and how this light should sculpt the head's features with new, appropriate shadows and highlights. 2. **Contextual Storytelling**: Instruct to add physical evidence of the scene's story onto the head, such as grime from a battle, sweat from exertion, or rain droplets from a storm. 3. **Color Grading Unification**: Detail how to apply the scene's specific color grade (e.g., cool desaturated tones, warm golden hour hues) to the head. 4. **Asset & Hair Adaptation**: Command the modification or removal of out-of-place elements (like clean jewelry in a gritty scene) and the restyling of the hair to fit the environment (e.g., messy, windblown, wet). 5. **Flawless Final Integration**: As the final step, describe the process of blending the neckline to be completely invisible, ensuring a uniform film grain and texture across the entire person."
|
||||
},
|
||||
{
|
||||
"name": "Seamless Integration (无痕融合)",
|
||||
"brief": "This image is a composite with minor inconsistencies between the head and body. Your task is to generate instructions for a subtle but master-level integration. Focus on creating a photorealistic and utterly convincing final image. The instructions should detail: 1. **Micro-Lighting Adjustment**: Fine-tune the lighting and shadows around the neck and jawline to create a perfect match. 2. **Skin Tone & Texture Unification**: Describe the process of unifying the skin tones for a seamless look, and more importantly, harmonizing the micro-textures like pores, fine hairs, and film grain across the blended area. 3. **Edge Blending Perfection**: Detail how to create an invisible transition at the neckline, making it appear as if it was never separate."
|
||||
},
|
||||
|
||||
# === 场景变换类 ===
|
||||
{
|
||||
"name": "Scene Teleportation (场景传送)",
|
||||
"brief": "Imagine the main subject of the image is suddenly teleported to a completely different and unexpected environment, while maintaining their exact pose. Based on the user's specific request: $user_prompt$, describe this new, richly detailed scene. The instruction must detail how the new environment's lighting and atmosphere should realistically affect the subject, including any necessary adjustments to clothing, accessories, or physical appearance to match the new setting."
|
||||
},
|
||||
{
|
||||
"name": "Season Change (季节变换)",
|
||||
"brief": "Transform the entire scene to be convincingly set in a different season. Based on the user's specific request: $user_prompt$, describe the seasonal transformation in detail. Include atmospheric effects like weather conditions, seasonal lighting, foliage changes, and appropriate clothing adjustments for the subject. Make sure the transformation feels natural and seasonally appropriate."
|
||||
},
|
||||
{
|
||||
"name": "Fantasy World (幻想领域)",
|
||||
"brief": "Transport the entire scene and its subject into a specific, richly detailed fantasy or sci-fi universe. Based on the user's specific request: $user_prompt$, describe the complete aesthetic overhaul. Replace modern elements with fantasy/sci-fi equivalents, transform clothing and accessories to match the new universe, and ensure the subject's appearance fits the magical or futuristic setting."
|
||||
},
|
||||
|
||||
# === 摄影技术类 ===
|
||||
{
|
||||
"name": "Camera Movement (移动镜头)",
|
||||
"brief": "Propose a dramatic and purposeful camera movement that reveals a new perspective or emotion in the scene. Based on the user's specific request: $user_prompt$, describe the *type* of shot and its *narrative purpose*. Explain how this camera movement enhances the story or emotion, and detail any necessary adjustments to composition, lighting, or focus to achieve the desired cinematic effect."
|
||||
},
|
||||
{
|
||||
"name": "Relighting (重新布光)",
|
||||
"brief": "Completely transform the mood and story of the image by proposing a new, cinematic lighting scheme. Based on the user's specific request: $user_prompt$, describe the lighting transformation in detail. Specify light sources, their positions, intensities, colors, and how they create the desired mood. Include any necessary adjustments to shadows, highlights, and overall atmosphere."
|
||||
},
|
||||
{
|
||||
"name": "Camera Zoom (画面缩放)",
|
||||
"brief": "Describe a specific zoom action that serves a narrative purpose. Based on the user's specific request: $user_prompt$, propose either a dramatic 'push-in' (zoom in) or a revealing 'pull-out' (zoom out). Explain the narrative purpose of this zoom, what new information or emotion it reveals, and any necessary adjustments to focus, depth of field, or composition."
|
||||
},
|
||||
{
|
||||
"name": "Professional Product Photography (专业产品图)",
|
||||
"brief": "Re-imagine this image as a high-end commercial product photograph. Based on the user's specific request: $user_prompt$, describe the professional photography setup. Specify the studio lighting arrangement, background treatment, composition style, and any props or lifestyle elements that enhance the product's appeal. Focus on creating an aspirational, commercial-quality image."
|
||||
},
|
||||
{
|
||||
"name": "Tilt-Shift Miniature (微缩世界)",
|
||||
"brief": "Convert the entire scene into a charming and highly detailed miniature model world. Based on the user's specific request: $user_prompt$, describe the tilt-shift miniature effect. Specify the depth of field adjustments, color saturation changes, and any modifications needed to enhance the toy-like, miniature appearance. Include details about focus areas and blur zones."
|
||||
},
|
||||
{
|
||||
"name": "Reflection Addition (添加倒影)",
|
||||
"brief": "Introduce a new, reflective surface into the scene to create a more dynamic composition. Based on the user's specific request: $user_prompt$, describe the reflective surface and its placement. Specify the type of reflection (mirror-like, water, glass, etc.), its quality and distortion, and how it enhances the overall composition and mood of the scene."
|
||||
},
|
||||
|
||||
# === 人物变换类 ===
|
||||
{
|
||||
"name": "Hair Style Change (更换发型)",
|
||||
"brief": "Describe a complete hair transformation for the subject. Based on the user's specific request: $user_prompt$, detail the new hairstyle, including cut, color, texture, and styling. Explain how this hair change reflects the desired persona or story, and include any necessary adjustments to accessories or clothing to complement the new look."
|
||||
},
|
||||
{
|
||||
"name": "Bodybuilding Transformation (肌肉猛男化)",
|
||||
"brief": "Dramatically transform the subject into a hyper-realistic, massively muscled bodybuilder. Based on the user's specific request: $user_prompt$, describe the bodybuilding transformation in detail. Specify muscle development, body proportions, skin texture changes, and any necessary clothing modifications to accommodate and showcase the new physique."
|
||||
},
|
||||
{
|
||||
"name": "Age Transformation (时光旅人)",
|
||||
"brief": "Visibly and realistically age or de-age the main subject. Based on the user's specific request: $user_prompt$, describe the age transformation in detail. Specify facial changes, hair modifications, skin texture adjustments, and any other age-related alterations. Ensure the transformation looks natural and appropriate for the target age."
|
||||
},
|
||||
{
|
||||
"name": "Fashion Makeover (衣橱改造)",
|
||||
"brief": "Give the subject a complete fashion makeover into a specific style. Based on the user's specific request: $user_prompt$, describe the entire outfit transformation. Specify clothing items, accessories, styling details, and how this fashion change reflects the desired aesthetic or persona. Include any necessary adjustments to hair or makeup to complement the new look."
|
||||
},
|
||||
|
||||
# === 环境变换类 ===
|
||||
{
|
||||
"name": "Furniture Removal (清空家具)",
|
||||
"brief": "Imagine the room in the image has been completely emptied for renovation. Based on the user's specific request: $user_prompt$, describe the furniture removal process. Specify what items need to be removed, how to realistically recreate the empty surfaces, and any necessary adjustments to lighting or architectural details to maintain the room's integrity."
|
||||
},
|
||||
{
|
||||
"name": "Interior Design (室内设计)",
|
||||
"brief": "Redesign this space in a specific, evocative style. Based on the user's specific request: $user_prompt$, describe the complete interior redesign. Specify furniture, color schemes, lighting, decor elements, and overall aesthetic while keeping the room's core structure intact. Create a cohesive design that reflects the desired style and mood."
|
||||
},
|
||||
|
||||
# === 艺术风格类 ===
|
||||
{
|
||||
"name": "Image Colorization (图像上色)",
|
||||
"brief": "Describe a specific artistic style for colorizing a black and white image. Based on the user's specific request: $user_prompt$, detail the colorization approach. Specify color palette choices, artistic style influences, mood considerations, and any special effects that enhance the colorization. Go beyond simple colorization to create an artistic interpretation."
|
||||
},
|
||||
{
|
||||
"name": "Cartoon/Anime Style (卡通漫画化)",
|
||||
"brief": "Redraw the entire image in a specific animated or illustrated style. Based on the user's specific request: $user_prompt$, describe the cartoon/anime transformation. Specify the artistic style, visual characteristics, color treatment, and any stylistic elements that define the chosen animation or illustration approach."
|
||||
},
|
||||
{
|
||||
"name": "Artistic Style Imitation (艺术风格模仿)",
|
||||
"brief": "Repaint the entire image in the style of a famous art movement. Based on the user's specific request: $user_prompt$, describe the artistic style transformation. Specify the art movement, its defining characteristics, brushwork techniques, color palette, and any other stylistic elements that capture the essence of the chosen artistic style."
|
||||
},
|
||||
{
|
||||
"name": "Pixel Art (像素艺术)",
|
||||
"brief": "Deconstruct the image into pixel art aesthetic. Based on the user's specific request: $user_prompt$, describe the pixel art transformation. Specify color palette limitations, pixel resolution, dithering techniques, and any retro gaming influences that create authentic pixel art appearance."
|
||||
},
|
||||
{
|
||||
"name": "Pencil Sketch (铅笔手绘)",
|
||||
"brief": "Transform the image into a pencil sketch style. Based on the user's specific request: $user_prompt$, describe the pencil sketch transformation. Specify line quality, shading techniques, paper texture effects, and any artistic considerations that create an authentic hand-drawn pencil sketch appearance."
|
||||
},
|
||||
{
|
||||
"name": "Oil Painting (油画风格)",
|
||||
"brief": "Transform the image into an oil painting style. Based on the user's specific request: $user_prompt$, describe the oil painting transformation. Specify brushwork techniques, color palette choices, texture effects, and any artistic elements that create an authentic oil painting aesthetic."
|
||||
},
|
||||
|
||||
# === 特殊效果类 ===
|
||||
{
|
||||
"name": "Material Transformation (材质置换)",
|
||||
"brief": "Re-imagine the main subject as a sculpture made from an unexpected material. Based on the user's specific request: $user_prompt$, describe the material transformation. Specify the new material's properties, how it affects the subject's appearance, lighting interactions, and any textural or reflective qualities that define the material."
|
||||
},
|
||||
{
|
||||
"name": "Movie Poster (电影海报)",
|
||||
"brief": "Transform the image into a compelling movie poster. Based on the user's specific request: $user_prompt$, describe the movie poster transformation. Specify the film genre, visual treatment, typography elements, and any cinematic effects that create an authentic movie poster appearance."
|
||||
},
|
||||
{
|
||||
"name": "Technical Blueprint (蓝图视角)",
|
||||
"brief": "Convert the image into a technical blueprint. Based on the user's specific request: $user_prompt$, describe the blueprint transformation. Specify the technical drawing style, measurement annotations, schematic elements, and any architectural or engineering details that create an authentic technical blueprint appearance."
|
||||
},
|
||||
|
||||
# === 实用功能类 ===
|
||||
{
|
||||
"name": "Text Removal (移除文字)",
|
||||
"brief": "Remove all text from the image as a meticulous restoration project. Based on the user's specific request: $user_prompt$, describe the text removal process. Specify which text elements need to be removed, how to reconstruct underlying surfaces, and any restoration techniques needed to create a seamless, text-free image."
|
||||
}
|
||||
],
|
||||
"suffix": "Your response must consist of concise instruction ready for the image editing AI. Do not add any conversational text, explanations, or deviations; only the instructions."
|
||||
}
|
||||
|
||||
@classmethod
|
||||
def INPUT_TYPES(cls):
|
||||
return {
|
||||
"required": {
|
||||
"preset": ([preset["name"] for preset in cls.data.get("presets", [])],),
|
||||
},
|
||||
"optional": {
|
||||
"user_prompt": ("STRING", {"default": "", "multiline": True}),
|
||||
}
|
||||
}
|
||||
RETURN_TYPES = ("STRING",)
|
||||
RETURN_NAMES = ("Prompt",)
|
||||
FUNCTION = "get_preset"
|
||||
CATEGORY = "utils"
|
||||
|
||||
@classmethod
|
||||
def get_brief_by_name(cls, name):
|
||||
for preset in cls.data.get("presets", []):
|
||||
if preset["name"] == name:
|
||||
return preset["brief"]
|
||||
return None
|
||||
|
||||
def get_preset(cls, preset, user_prompt=""):
|
||||
# Find the selected preset
|
||||
brief_text = ""
|
||||
for p in cls.data.get("presets", []):
|
||||
if p["name"] == preset:
|
||||
brief_text = p["brief"]
|
||||
break
|
||||
|
||||
# Replace the placeholder with user prompt
|
||||
if user_prompt and user_prompt.strip():
|
||||
brief_text = brief_text.replace("$user_prompt$", user_prompt.strip())
|
||||
else:
|
||||
# If no user prompt provided, remove the placeholder and provide a generic instruction
|
||||
brief_text = brief_text.replace("$user_prompt$", "the user's desired transformation")
|
||||
|
||||
brief = "The Brief:" + brief_text
|
||||
fullString = cls.data.get("prefix")+'\n'+brief+'\n'+cls.data.get("suffix")
|
||||
return (fullString,)
|
||||
|
||||
NODE_CLASS_MAPPINGS = {
|
||||
"LoadKontextPresets_UTK": LoadKontextPresets_UTK,
|
||||
}
|
||||
|
||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
"LoadKontextPresets_UTK": "Kontext VLM System Presets (UTK)",
|
||||
}
|
||||
Reference in New Issue
Block a user