图像提示词图像上传支持9张

models.py

@@ -277,6 +277,8 @@ class PromptCreate(BaseModel):
     cover_image: str                        # 封面图
     media_type: str = "image"               # "image" 或 "video"（示例媒体类型）
     video_url: Optional[str] = None         # 示例视频 URL（仅 media_type=video）
+    images: Optional[List[str]] = None      # 图像提示词图片列表（最多9张，首张为封面）
+    is_original: bool = False               # 原创作品标记
     price: int = 0                          # 定价（0=免费）
     tags: Optional[List[str]] = None        # 自定义标签（最多10个，每个≤12字）
 
@@ -289,6 +291,8 @@ class PromptUpdate(BaseModel):
     cover_image: Optional[str] = None
     media_type: Optional[str] = None
     video_url: Optional[str] = None
+    images: Optional[List[str]] = None      # 图像提示词图片列表（最多9张，首张为封面）
+    is_original: Optional[bool] = None      # 原创作品标记
     price: Optional[int] = None
     tags: Optional[List[str]] = None
 

router_prompts.py

@@ -54,6 +54,23 @@ MAX_CATEGORY_LEN = 8
 MAX_CATEGORY_PER_TYPE = 30
 MAX_TAGS = 10          # 单个提示词最多标签数
 MAX_TAG_LEN = 12       # 单个标签最大字数
+MAX_IMAGES = 9         # 图像提示词最多图片数（与讨论区一致）
+
+
+def _sanitize_images(images) -> list:
+    """清洗图片列表：过滤非法元素，最多 MAX_IMAGES 张"""
+    if not isinstance(images, list):
+        return []
+    cleaned = []
+    for url in images:
+        if not isinstance(url, str):
+            continue
+        u = url.strip()
+        if u:
+            cleaned.append(u)
+        if len(cleaned) >= MAX_IMAGES:
+            break
+    return cleaned
 
 
 def calculate_tx_hash(tx_id, account, tx_type, amount, prev_hash):
@@ -352,6 +369,9 @@ async def create_prompt(prompt: PromptCreate, current_user: str = Depends(requir
         # 未上传示例视频时降级为图片展示
         media_type = "image"
 
+    # 图像提示词的图片列表（首张即封面）；视频提示词不保留图片列表
+    images = _sanitize_images(prompt.images) if media_type == "image" else []
+
     new_prompt = {
         "id": f"prompt_{int(time.time())}_{uuid.uuid4().hex[:6]}",
         "prompt_type": prompt.prompt_type,
@@ -362,6 +382,8 @@ async def create_prompt(prompt: PromptCreate, current_user: str = Depends(requir
         "cover_image": prompt.cover_image,
         "media_type": media_type,
         "video_url": video_url,
+        "images": images,
+        "is_original": bool(prompt.is_original),
         "price": price,
         "author": current_user,
         "created_at": int(time.time()),
@@ -441,11 +463,21 @@ async def update_prompt(prompt_id: str, update_data: PromptUpdate, current_user:
                         return False
                     prompt["media_type"] = update_data.media_type
                     if update_data.media_type == "image":
+                        # 图像类型不保留视频（与 create 逻辑一致）
                         prompt["video_url"] = None
                 if update_data.video_url is not None:
                     prompt["video_url"] = update_data.video_url
                     if update_data.video_url:
                         prompt["media_type"] = "video"
+                # 类型变更后补正关联字段（pydantic 无法区分未传与 null，统一在此收尾）
+                if prompt.get("media_type") == "image":
+                    prompt["video_url"] = None
+                else:
+                    prompt["images"] = []
+                if update_data.images is not None:
+                    prompt["images"] = _sanitize_images(update_data.images)
+                if update_data.is_original is not None:
+                    prompt["is_original"] = bool(update_data.is_original)
                 if update_data.tags is not None:
                     prompt["tags"] = _sanitize_tags(update_data.tags)
                 result_holder["prompt"] = prompt