feat: 初始化项目，并且接近完成 ozon 部分

server/api/generate.py

@@ -0,0 +1,126 @@
+"""无状态套图生成：请求自带采集数据，不落商品库。"""
+from __future__ import annotations
+
+from fastapi import APIRouter, BackgroundTasks, Depends, HTTPException
+
+from config import get_settings
+from db import get_db
+from models import Suite
+from schemas import (
+    GenerateRequest, PLATFORM_SPECS, SUPPORTED_TYPES, SuiteCreateResponse, TextMaterial,
+    PlanRequest, PlanResponse, PlanItemOut,
+)
+from services.generator import run_suite
+from services.planner import generate_plan
+from services.prompt import type_name
+
+router = APIRouter(prefix="/api", tags=["generate"])
+
+
+def texts_to_raw(texts: list[TextMaterial]) -> dict:
+    """插件上传的文本素材 → prompt 上下文用的 raw dict（后写的覆盖先写的）。"""
+    raw: dict = {}
+    for t in texts:
+        if t.kind == "title" and t.content:
+            raw["title"] = t.content
+        elif t.kind == "price" and t.content:
+            raw["price"] = t.content
+        elif t.kind == "brand" and t.content:
+            raw["brand"] = t.content
+        elif t.kind == "params" and t.pairs:
+            merged = {p["key"]: p["value"] for p in (raw.get("params") or [])}
+            for p in t.pairs:
+                merged.setdefault(p["key"], p["value"])
+            raw["params"] = [{"key": k, "value": v} for k, v in merged.items()]
+        elif t.kind == "selling_point" and t.content:
+            raw["sellingPoints"] = t.content
+        elif t.kind == "desc" and t.content:
+            raw["desc"] = t.content
+    return raw
+
+
+@router.post("/generate", response_model=SuiteCreateResponse)
+async def generate_suite(
+    req: GenerateRequest,
+    background: BackgroundTasks,
+    db=Depends(get_db),
+) -> SuiteCreateResponse:
+    if not req.images:
+        raise HTTPException(status_code=400, detail="未勾选任何图片，无法生成")
+
+    # 生成任务列表：方案优先；无方案时按 types（空则默认四种）
+    if req.plan:
+        jobs: list[dict] = []
+        for item in req.plan:
+            if item.count <= 0:
+                continue
+            if item.kind not in SUPPORTED_TYPES:
+                raise HTTPException(status_code=400, detail=f"方案项「{item.title}」的图类型不支持: {item.kind}")
+            # 同一项多张 → 展开为多任务，第二张起在标题上加序号
+            for n in range(item.count):
+                jobs.append({
+                    "kind": item.kind,
+                    "title": item.title if item.count == 1 else f"{item.title}{n + 1}",
+                    "detail": item.detail,
+                    "prompt_hint": item.prompt_hint,
+                    "variant_name": item.variant_name,
+                })
+        if not jobs:
+            raise HTTPException(status_code=400, detail="方案中所有项的数量都是 0")
+        types = list(dict.fromkeys(j["kind"] for j in jobs))
+    else:
+        types = req.types or ["white_bg", "key_features", "lifestyle", "multi_scene"]
+        bad = [t for t in types if t not in SUPPORTED_TYPES]
+        if bad:
+            raise HTTPException(status_code=400, detail=f"不支持的图类型: {bad}")
+        jobs = [{"kind": t, "title": type_name(t), "detail": "", "prompt_hint": "", "variant_name": None} for t in types]
+
+    if req.platform not in PLATFORM_SPECS:
+        raise HTTPException(status_code=400, detail=f"不支持的目标平台: {req.platform}（ozon | wb | cn）")
+    spec = PLATFORM_SPECS[req.platform]
+
+    settings = get_settings()
+    suite = Suite(
+        product_id=None,
+        style_set=req.style_set,
+        platform=req.platform,
+        lang=spec["lang"],
+        ratio=spec["ratio"],
+        types=types,
+        plan=jobs,
+        provider=req.provider or settings.image_provider,
+        context=texts_to_raw(req.texts),
+        # 参考图池：main 组优先，其余组按序补充（variant 绑定靠 variant_name 匹配）
+        ref_images=[
+            {
+                "url": i.url,
+                "group_key": i.group_key,
+                "variant_name": i.variant_name,
+            }
+            for i in sorted(req.images, key=lambda x: 0 if x.group_key == "main" else 1)
+        ],
+    )
+    db.add(suite)
+    await db.commit()
+    await db.refresh(suite)
+
+    background.add_task(run_suite, str(suite.id))
+    return SuiteCreateResponse(suite_id=str(suite.id))
+
+
+@router.post("/plan", response_model=PlanResponse)
+async def plan_suite(req: PlanRequest) -> PlanResponse:
+    """DeepSeek 根据商品信息生成出图方案。"""
+    product_info = texts_to_raw(req.texts)
+    if not product_info.get("title"):
+        raise HTTPException(status_code=400, detail="缺少商品标题，无法规划")
+    try:
+        result = await generate_plan(product_info, req.sku_variants, req.image_stats, req.platform)
+    except RuntimeError as exc:
+        raise HTTPException(status_code=502, detail=str(exc)) from exc
+    except Exception as exc:  # noqa: BLE001
+        raise HTTPException(status_code=502, detail=f"规划失败: {exc}") from exc
+    return PlanResponse(
+        summary=result["summary"],
+        items=[PlanItemOut(**i) for i in result["items"]],
+    )