feat: 插件开发 ozon 端主体完成

server/services/prompt.py

@@ -12,38 +12,53 @@ from __future__ import annotations
 
 import re
 
-# ── 风格模板（与插件端 STYLE_SET_OPTIONS 对应）─────────────────────────────
+# ── 风格模板（与插件端 STYLE_SET_OPTIONS 对应；提示词可被用户在插件里改写覆盖）───
+# 提示词用中文：生图 provider（通义万相/豆包）均为国产模型，中文理解一流，且便于用户自行改写。
 
 STYLE_SETS: dict[int, dict] = {
     1: {
-        "name": "经典商拍",
-        "tone": "premium commercial e-commerce photography, clean soft studio lighting, "
-                "gentle gradient background, catalog-grade presentation, refined and trustworthy",
-        "bg": "light neutral studio backdrop with soft vignette",
+        "name": "高级质感大片",
+        "tone": "高端电商大片质感，柔和的方向性棚拍光，背景带细腻的浅渐变，材质纹理清晰可见，"
+                "色彩层次高级克制，商业画册级品质，构图干净、留白充足",
+        "bg": "",
     },
     2: {
-        "name": "生活杂志",
-        "tone": "editorial lifestyle magazine aesthetic, natural window light, "
-                "cozy lived-in atmosphere, muted film tones, candid storytelling",
-        "bg": "warm lifestyle home setting with plants and textured fabrics",
+        "name": "清新生活场景",
+        "tone": "明亮通透的生活场景摄影，自然窗光，柔和的低饱和居家环境，浅景深虚化，"
+                "真实自然的氛围感，绿植与暖色织物点缀，温馨有人气",
+        "bg": "",
     },
     3: {
-        "name": "极简高冷",
-        "tone": "minimalist high-end aesthetic, vast negative space, single directional light, "
-                "cool grey palette, architectural calm, quiet luxury",
-        "bg": "seamless light grey studio background with subtle shadow",
+        "name": "极简白底规范",
+        "tone": "极简棚拍风格，纯净无缝的浅色背景，柔和均匀的无影布光，以商品为中心的严谨构图，"
+                "安静的高级感，画面只保留轻微的自然接触投影",
+        "bg": "",
     },
     4: {
-        "name": "活力爆款",
-        "tone": "vibrant high-conversion e-commerce style, punchy saturated accents, "
-                "energetic composition, bold contrast, promotional poster energy",
-        "bg": "bright colorful gradient backdrop with dynamic geometric shapes",
+        "name": "炫彩促销风",
+        "tone": "高能量促销风格，高饱和度色块背景搭配动感几何图形，强对比，节日大促海报氛围，"
+                "构图抢眼、视觉冲击力强",
+        "bg": "",
     },
     5: {
-        "name": "暗调质感",
-        "tone": "dark moody premium product photography, dramatic rim lighting, "
-                "deep charcoal background, rich texture detail, luxurious atmosphere",
-        "bg": "matte black background with soft spotlight and subtle smoke haze",
+        "name": "暗调轻奢",
+        "tone": "暗调轻奢质感，深炭灰色背景，轮廓光勾勒商品边缘，材质细节丰富，带轻微雾感，"
+                "如美术馆展陈般的呈现",
+        "bg": "",
+    },
+    6: {
+        "name": "俄式风情",
+        "tone": "俄式风情电商大片，浓郁温暖的色调，红与金的传统配色点缀，冬日节庆氛围，"
+                "深色木质与毛毡织物背景，如暖炉烛光般的柔和光晕，厚重扎实的质感，"
+                "带一丝巴洛克式的华丽细节，适合俄语区市场",
+        "bg": "",
+    },
+    7: {
+        "name": "北欧极简",
+        "tone": "北欧极简风格，白色与浅灰的原木空间，大量自然漫射光，干净利落的线条，"
+                "浅色木质背景点缀少量绿植，克制的中性配色，画面通透轻盈，"
+                "舒适宁静的氛围",
+        "bg": "",
     },
 }
 
@@ -192,9 +207,10 @@ def _prompt_material(ctx: dict, style: dict, lang: str) -> str:
     )
 
 def _prompt_lifestyle(ctx: dict, style: dict, lang: str) -> str:
+    bg = f" ({style['bg']})" if style.get("bg") else ""
     return (
         f"Lifestyle in-context scene for product \"{ctx['title']}\": the product is naturally "
-        f"used / placed in a real environment ({style['bg']}), realistic human-scale surroundings, "
+        f"used / placed in a real environment{bg}, realistic human-scale surroundings, "
         f"soft daylight, authentic candid mood, product remains the clear visual focus. "
         f"{style['tone']}. {TEXT_RENDER[lang]} {QUALITY} {PRODUCT_REF_LOCK}"
     )
@@ -242,6 +258,7 @@ def _prompt_custom(ctx: dict, style: dict, lang: str, extra: dict) -> str:
     hint = (extra.get("prompt_hint") or "").strip()
     purpose = extra.get("title") or ""
     detail = extra.get("detail") or ""
+    bg = f" {style['bg']} as environment." if style.get("bg") else ""
     composed = (
         f"E-commerce marketing image for product \"{ctx['title']}\""
         + (f" — {purpose}" if purpose else "")
@@ -250,7 +267,7 @@ def _prompt_custom(ctx: dict, style: dict, lang: str, extra: dict) -> str:
     )
     if hint:
         composed += f" Composition: {hint}."
-    return f"{composed} {style['tone']}. {style['bg']} as environment. {TEXT_RENDER[lang]} {QUALITY} {PRODUCT_REF_LOCK}"
+    return f"{composed}{bg} {style['tone']}. {TEXT_RENDER[lang]} {QUALITY} {PRODUCT_REF_LOCK}"
 
 _PROMPT_BUILDERS = {
     "white_bg": _prompt_white_bg,
@@ -265,13 +282,18 @@ _PROMPT_BUILDERS = {
 }
 
 
-def build_prompt(type_id: str, ctx: dict, style_set: int, lang: str, extra: dict | None = None) -> str:
+def build_prompt(type_id: str, ctx: dict, style_set: int, lang: str, extra: dict | None = None,
+                 style_prompt: str | None = None) -> str:
     """构造指定图类型的完整生图 prompt。
 
     extra: 方案项信息 {title, detail, prompt_hint}——custom 类型必需，
     预设类型也会把 prompt_hint 作为构图补充注入。
+    style_prompt: 用户改写的风格提示词，覆盖 style_set 内置模板（tone/bg 整体替换）。
     """
-    style = STYLE_SETS.get(style_set, STYLE_SETS[1])
+    if style_prompt and style_prompt.strip():
+        style = {"name": "custom", "tone": style_prompt.strip(), "bg": ""}
+    else:
+        style = STYLE_SETS.get(style_set, STYLE_SETS[1])
     extra = extra or {}
     if type_id == "custom":
         prompt = _prompt_custom(ctx, style, lang, extra)