3218485270
f493bde8a9
- 缓存热点批量流程（有采集缓存不触发 Google） - 简报不足直接从采集缓存生成（轻量补齐） - 三图合成（模特/印花/底图）+ 底图压缩 <2MB - 热点去重→风格去重自动切换 + 不适合类目 review 兜底 - 透明背景（background=transparent）+ 提示词清洗（敏感词/背景描述） - 任务前 basemap 校验 + 模板国家校验 + 模特任务级分配
145 lines
7.2 KiB
Python
145 lines
7.2 KiB
Python
"""固定提示词模板（规则写死，保证每条一致）+ 装配函数。
|
||
|
||
所有最终提示词都由四要素（motif / art_style / color_palette / composition）
|
||
用下面的模板确定性拼出，LLM 不再自由发挥，因此结构永远一致、可复用于 img2img。
|
||
|
||
v3：模板按国家区分（COUNTRY_TEMPLATES），每国有自己的设计风格引导段；
|
||
顶层 DEFAULT_TEMPLATES 作为兜底。文字规则统一：英文可加可不加、适配印花即可，
|
||
任何文字严禁政治/宗教/仇恨/暴力/性/品牌/商标/真实人物等敏感内容。
|
||
"""
|
||
from typing import Any, Dict, Optional
|
||
|
||
# —— 通用固定段（所有国家共用，保证结构一致）——
|
||
# 尺寸规则：最小约 15×18cm ~ 最大 26×32cm 之间自由选择（防模型默认出满幅大图）
|
||
SIZE_RULE = (
|
||
"size: choose freely between a MINIMUM print area of about 15x18 cm "
|
||
"and a MAXIMUM of 26x32 cm, any size in this range fits, "
|
||
"pick the one that best suits the design, keep proportions, "
|
||
"scale naturally to the content, do NOT stretch, "
|
||
"do NOT fill the entire canvas, do NOT force full-bleed, "
|
||
"leave balanced margins around the artwork"
|
||
)
|
||
# 排除段：无衣服/模特/场景/水印
|
||
NEG_FIXED = (
|
||
"no garment, no shirt, no model, no mannequin, no background scene, no watermark"
|
||
)
|
||
# 文字规则（v3）：可加可不加、适配印花即可；任何文字严禁敏感内容
|
||
# 注意：正向提示词不写敏感词（no politics/no hate/no violence/no sexual…会被图像审核误判），
|
||
# 负向约束统一由 negative_prompt 承担
|
||
TEXT_RULE = (
|
||
"text: optional - add short original English words or a small slogan ONLY if they fit "
|
||
"the print, or keep it text-free; any text must be safe, short and original; "
|
||
"no brand names, no logos, no trademarked phrases, no real people names"
|
||
)
|
||
|
||
# —— 国家风格引导段（每国不同：US 大胆高对比 / GB 英式自嘲与复古 / JP 卡哇伊极简 / AU 海滨户外）——
|
||
COUNTRY_STYLE_HINT: Dict[str, str] = {
|
||
"US": "US-market aesthetic: bold confident statement graphic, high contrast, "
|
||
"clean modern vector, sporty or humorous mood",
|
||
"GB": "UK-market aesthetic: witty understated British charm, heritage-inspired motifs, "
|
||
"retro sportswear or punk-zine mood",
|
||
"JP": "JP-market aesthetic: kawaii cute or clean minimal, soft pastel-friendly, "
|
||
"polished neat lines, small cute mascot mood",
|
||
"AU": "AU-market aesthetic: laid-back coastal and outdoor vibe, nature-inspired, "
|
||
"bright fresh energy",
|
||
"MX": "MX-market aesthetic: vibrant mexican folk art, sugar skull / loteria / aztec motifs, "
|
||
"fiesta colors, festive cultural pride mood",
|
||
}
|
||
|
||
|
||
def _image_prompt_for(country: str) -> str:
|
||
hint = COUNTRY_STYLE_HINT.get(country, COUNTRY_STYLE_HINT["US"])
|
||
return (
|
||
"{motif}, {art_style}, {color_palette}, {composition}, "
|
||
"standalone pure print design, print-ready artwork, "
|
||
"isolated on pure white background, flat vector-like graphic, "
|
||
"crisp clean edges, high resolution, ultra sharp, high contrast, "
|
||
f"{hint}, {SIZE_RULE}, "
|
||
f"{NEG_FIXED}, {TEXT_RULE}"
|
||
)
|
||
|
||
|
||
# 顶层默认（兜底：国家未配置时使用，内容等同 US 风格基线）
|
||
DEFAULT_TEMPLATES: Dict[str, str] = {
|
||
"image_prompt": _image_prompt_for("US"),
|
||
# 预览图：直接印在平铺白T上（无真人），用于快速看效果
|
||
"wearable_prompt": (
|
||
"{motif}, {art_style}, {color_palette}, {composition}, "
|
||
"printed centered on the chest of a flat-lay plain white t-shirt, "
|
||
"print sized freely between about 15x18 cm and a max of 26x32 cm, "
|
||
"scaled naturally to the artwork, not stretched, not full-bleed, "
|
||
"studio lighting, e-commerce product photo, no human model"
|
||
),
|
||
# 复合提示词（三图模特合成）：图1=模特 / 图2=纯印花设计稿 / 图3=平铺底图 → 模特穿着成品
|
||
"composite_prompt": (
|
||
"【图片角色,按提交顺序】图1=模特实拍图(基底)；图2=纯印花设计稿；"
|
||
"图3=平铺衣服底图(颜色/面料来源)。\n"
|
||
"TASK: 把图2的印花设计印到图3底色的衣服上,并让图1的模特穿上"
|
||
"“图3底色+图2印花”的衣服。\n"
|
||
"RULES:\n"
|
||
"1.底色锁定:从图3提取衣服底色与面料,最终合成中必须100%保持不变,严禁偏色。\n"
|
||
"2.印花提取:从图2精准提取纯印花图案(线条/色号/比例),叠加到图3底色上形成合成面料。\n"
|
||
"3.主体遮罩:识别图1模特服装穿着区域(忽略皮肤/头发/背景/配饰),"
|
||
"用合成面料完整覆盖,清除原衣服颜色与图案。\n"
|
||
"4.精准贴合:合成面料严格跟随图1衣服立体结构,褶皱/扭转处印花相应变形,"
|
||
"杜绝“贴纸感”与“平面涂色感”。\n"
|
||
"5.光影融合:按图1环境光方向调整亮度/对比度,印花受光影响产生明暗变化但色号不偏移。\n"
|
||
"6.纯净输出:仅输出一张最终合成图;图1背景/人物/构图/光影100%不变,"
|
||
"仅替换衣服印花与底色。\n"
|
||
"DESIGN CONTENT: {motif}, {art_style}, {color_palette}, {composition}."
|
||
),
|
||
# 复合负向（印图专用）
|
||
"composite_negative": (
|
||
"garment changed, wrong color, distorted print, blurry, low-res, "
|
||
"human model, body, extra objects, watermark, glow, 3d render, "
|
||
"text unless part of design"
|
||
),
|
||
}
|
||
|
||
# —— 按国家覆盖：目前仅 image_prompt 有国家差异化；wearable/composite 共用顶层默认 ——
|
||
COUNTRY_TEMPLATES: Dict[str, Dict[str, str]] = {
|
||
cc: {"image_prompt": _image_prompt_for(cc)} for cc in COUNTRY_STYLE_HINT
|
||
}
|
||
|
||
|
||
def resolve_templates(tpls: Optional[Dict[str, Any]], country: Optional[str] = None) -> Dict[str, str]:
|
||
"""解析最终模板：DEFAULT_TEMPLATES 兜底 → config 顶层覆盖 → config countries.<country> 覆盖。
|
||
|
||
config 结构示例:
|
||
prompt_templates:
|
||
image_prompt: "..." # 顶层默认
|
||
countries:
|
||
GB:
|
||
image_prompt: "..." # 国家专属
|
||
"""
|
||
t = dict(DEFAULT_TEMPLATES)
|
||
if tpls:
|
||
t.update({k: v for k, v in tpls.items() if k in DEFAULT_TEMPLATES})
|
||
countries = tpls.get("countries") or {}
|
||
if country and isinstance(countries, dict):
|
||
cc_tpls = countries.get(country) or {}
|
||
if isinstance(cc_tpls, dict):
|
||
t.update({k: v for k, v in cc_tpls.items() if k in DEFAULT_TEMPLATES})
|
||
elif country and country in COUNTRY_TEMPLATES:
|
||
t.update(COUNTRY_TEMPLATES[country])
|
||
return t
|
||
|
||
|
||
def assemble_prompts(
|
||
motif: str,
|
||
art_style: str,
|
||
palette: str,
|
||
composition: str,
|
||
tpls: Optional[Dict[str, Any]] = None,
|
||
country: Optional[str] = None,
|
||
) -> Dict[str, str]:
|
||
"""用固定模板确定性装配三种提示词。tpls 可来自 config 覆盖（按国家优先）。"""
|
||
t = resolve_templates(tpls, country)
|
||
out = {}
|
||
for key in ("image_prompt", "wearable_prompt", "composite_prompt"):
|
||
out[key] = t[key].format(
|
||
motif=motif, art_style=art_style, color_palette=palette, composition=composition
|
||
)
|
||
out["composite_negative"] = t["composite_negative"]
|
||
return out
|