inkreachhub/pod_trend_agent
3
1
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

修复种草图生成：以三合一主图为参考 img2img 生成，按颜色分配（每色优先、超出随机补足）；新增 BR/CA/DE/ES/IT/PL/SA 七国配置与提示词；删除验证用测试脚本

Browse Source
This commit is contained in:
3218485270
2026-08-24 14:02:19 +08:00
parent f493bde8a9
commit 3c341d2e78
51 changed files with 2571 additions and 514 deletions
Download Patch File Download Diff File Expand all files Collapse all files
graph/nodes/product_node.py
+29 -23
View File
@@ -221,6 +221,7 @@ def _process_spu(
# 6) 模板选择按 SPU.mark 决定：
# mark==1 → 新三图合成模板 MODEL_WEAR_PROMPT（图1模特 + 图2印花设计 + 图3底图）
# mark!=1 → 旧两图合成模板 composite_prompt（底图 + 印花设计）
# mark==1 统一只做三合一：无模特图时跳过合成，不再回退两图合成（印花+底图）
if int(spu.get("mark") or 0) == 1:
print(f"{tag} SPU {spu['code']} mark=1 → 使用三图合成模板（图1模特+图2印花+图3底图）")
if model_img is not None:
@@ -231,13 +232,14 @@ def _process_spu(
result["model_folder"] = model_img.parent.name
print(f"{tag} 模特图（任务级分配，{model_img.parent.name}/）: {model_copy}")
else:
print(f"{tag} material_library 无模特图，回退两图合成（composite_prompt）")
print(f"{tag} material_library 无模特图，mark=1 统一只做三合一，跳过合成")
else:
print(f"{tag} SPU {spu['code']} mark={spu.get('mark')} → 使用两图合成模板 composite_prompt（底图+印花）")
# 7) 合成：
# 有模特图 → 三图合成（图1=模特 / 图2=印花设计 / 图3=底图）
# 无模特图 → 两图合成平铺服装图（图3=底图 + 图2=印花设计）
# mark=1 无模特图 → 跳过合成（统一只做三合一，不做印花+底图两图合成）
# mark!=1 无模特图 → 两图合成平铺服装图（图3=底图 + 图2=印花设计）
if "design_path" not in result:
print(f"{tag} 无设计稿，跳过合成")
elif model_img is not None:
@@ -267,28 +269,32 @@ def _process_spu(
print(f"{tag} 三图合成重试仍失败 → 跳过该产品（不生成标题/不写模板）: {e}")
return None
else:
printed_path = str(prod_dir / f"{img_code}_printed.png")
try:
# 两图合成（无模特）：用平铺印图文案（wearable_prompt），回退旧 composite_prompt
flat_prompt = (brief.get("wearable_prompt") or "").strip() or brief.get("composite_prompt", "")
ib.print(flat_prompt, str(basemap_img), printed_path,
brief.get("composite_negative", ""),
extra_images=[design_path], # 图2印花
size="1504x2000") # 合成统一 1504x2000
result["printed_path"] = printed_path
print(f"{tag} 平铺服装图已生成（无模特，底图+印花）: {printed_path}")
except Exception as e: # noqa: BLE001
print(f"{tag} 平铺服装图失败，退避重试…: {e}")
retried = _retry_image(ib.print, flat_prompt, str(basemap_img), printed_path,
brief.get("composite_negative", ""),
extra_images=[design_path], size="1504x2000")
if retried is not None:
# mark=1 无模特图 → 统一只做三合一，不做印花+底图两图合成
if int(spu.get("mark") or 0) == 1:
print(f"{tag} mark=1 无模特图，跳过合成（统一只做三合一）")
else:
printed_path = str(prod_dir / f"{img_code}_printed.png")
try:
# 两图合成（无模特，mark!=1）：用平铺印图文案（wearable_prompt），回退旧 composite_prompt
flat_prompt = (brief.get("wearable_prompt") or "").strip() or brief.get("composite_prompt", "")
ib.print(flat_prompt, str(basemap_img), printed_path,
brief.get("composite_negative", ""),
extra_images=[design_path], # 图2印花
size="1504x2000") # 合成统一 1504x2000
result["printed_path"] = printed_path
print(f"{tag} 平铺服装图重试成功: {printed_path}")
else:
errors.append({"node": "product", "type": type(e).__name__, "message": f"平铺服装图生成失败(重试仍失败): {e}", "trace": ""})
print(f"{tag} 平铺服装图重试仍失败 → 跳过该产品（不生成标题/不写模板）: {e}")
return None
print(f"{tag} 平铺服装图已生成（无模特，底图+印花）: {printed_path}")
except Exception as e: # noqa: BLE001
print(f"{tag} 平铺服装图失败，退避重试…: {e}")
retried = _retry_image(ib.print, flat_prompt, str(basemap_img), printed_path,
brief.get("composite_negative", ""),
extra_images=[design_path], size="1504x2000")
if retried is not None:
result["printed_path"] = printed_path
print(f"{tag} 平铺服装图重试成功: {printed_path}")
else:
errors.append({"node": "product", "type": type(e).__name__, "message": f"平铺服装图生成失败(重试仍失败): {e}", "trace": ""})
print(f"{tag} 平铺服装图重试仍失败 → 跳过该产品（不生成标题/不写模板）: {e}")
return None
# 7.2) 多色：单 SPU 多色时每个颜色再执行一次三合一（用各自底图），轮播图按颜色路由
color_composites: List[Dict[str, Any]] = []