Upload 7 files

models.py

@@ -278,6 +278,7 @@ class PromptCreate(BaseModel):
     media_type: str = "image"               # "image" 或 "video"（示例媒体类型）
     video_url: Optional[str] = None         # 示例视频 URL（仅 media_type=video）
     price: int = 0                          # 定价（0=免费）
+    tags: Optional[List[str]] = None        # 自定义标签（最多10个，每个≤12字）
 
 class PromptUpdate(BaseModel):
     """ 更新提示词 """
@@ -289,6 +290,7 @@ class PromptUpdate(BaseModel):
     media_type: Optional[str] = None
     video_url: Optional[str] = None
     price: Optional[int] = None
+    tags: Optional[List[str]] = None
 
 class PromptCategoryAdd(BaseModel):
     """ 添加提示词分类 """

router_prompts.py

@@ -52,6 +52,8 @@ MAX_PROMPT_TEXT_LEN = 20000
 MAX_PRICE = 10000
 MAX_CATEGORY_LEN = 8
 MAX_CATEGORY_PER_TYPE = 30
+MAX_TAGS = 10          # 单个提示词最多标签数
+MAX_TAG_LEN = 12       # 单个标签最大字数
 
 
 def calculate_tx_hash(tx_id, account, tx_type, amount, prev_hash):
@@ -59,6 +61,22 @@ def calculate_tx_hash(tx_id, account, tx_type, amount, prev_hash):
     return hashlib.sha256(data.encode()).hexdigest()
 
 
+def _sanitize_tags(tags) -> list:
+    """清洗标签：去空白/去重/截断，最多 MAX_TAGS 个，每个≤ MAX_TAG_LEN 字"""
+    if not isinstance(tags, list):
+        return []
+    cleaned = []
+    for tag in tags:
+        if not isinstance(tag, str):
+            continue
+        name = tag.strip()[:MAX_TAG_LEN]
+        if name and name not in cleaned:
+            cleaned.append(name)
+        if len(cleaned) >= MAX_TAGS:
+            break
+    return cleaned
+
+
 def _load_categories() -> dict:
     """读取分类数据，不存在时用内置默认值初始化并持久化"""
     categories = db.load_data("prompt_categories.json", default_data=None)
@@ -161,6 +179,21 @@ async def add_prompt_category(request: Request, body: PromptCategoryAdd, current
 # 📝 提示词CRUD接口
 # ==========================================
 
+@router.get("/api/prompt-tags")
+async def get_prompt_tags():
+    """
+    🏷️ 聚合全站标签（发布表单推荐用，按使用次数降序，最多返回100个）
+    """
+    prompts_db = db.load_data("prompts.json", default_data=[])
+    counter = {}
+    for prompt in prompts_db:
+        for tag in (prompt.get("tags") or []):
+            if isinstance(tag, str) and tag.strip():
+                counter[tag] = counter.get(tag, 0) + 1
+    ranked = sorted(counter.items(), key=lambda kv: (-kv[1], kv[0]))[:100]
+    return {"status": "success", "data": [{"name": n, "count": c} for n, c in ranked]}
+
+
 @router.get("/api/prompts")
 async def get_prompts(prompt_type: str = None, category: str = None,
                       page: int = 1, limit: int = 20, sort: str = "latest"):
@@ -342,7 +375,8 @@ async def create_prompt(prompt: PromptCreate, current_user: str = Depends(requir
         "views": 0,
         "daily_views": 0,
         "viewed_by": [],
-        "daily_views_date": ""
+        "daily_views_date": "",
+        "tags": _sanitize_tags(prompt.tags)
     }
 
     prompts_db = db.load_data("prompts.json", default_data=[])
@@ -412,6 +446,8 @@ async def update_prompt(prompt_id: str, update_data: PromptUpdate, current_user:
                     prompt["video_url"] = update_data.video_url
                     if update_data.video_url:
                         prompt["media_type"] = "video"
+                if update_data.tags is not None:
+                    prompt["tags"] = _sanitize_tags(update_data.tags)
                 result_holder["prompt"] = prompt
                 return True
         result_holder["error"] = "not_found"