Upload 5 files

models.py

@@ -269,7 +269,7 @@ class PostUpdate(BaseModel):
 
 class PromptCreate(BaseModel):
     """ 发布提示词 """
-    prompt_type: str                        # "image" 或 "video"（所属子界面）
+    prompt_type: str                        # "image"、"video" 或 "music"（所属子界面）
     category: str                           # 所属分类（人物/动作等）
     title: str                              # 标题
     content: str                            # 公开介绍文案
@@ -298,5 +298,5 @@ class PromptUpdate(BaseModel):
 
 class PromptCategoryAdd(BaseModel):
     """ 添加提示词分类 """
-    prompt_type: str                        # "image" 或 "video"
+    prompt_type: str                        # "image"、"video" 或 "music"
     name: str                               # 分类名称（≤8字，去重）

router_prompts.py

@@ -40,10 +40,11 @@ logger = logging.getLogger("ComfyUI-Ranking.Prompts")
 # 内置默认分类（首次读取 prompt_categories.json 不存在时自动初始化）
 DEFAULT_CATEGORIES = {
     "image": ["人物", "动物", "植物", "道具", "建筑"],
-    "video": ["动作", "表情变化", "天气变化", "光效变化", "运镜", "视觉风格"]
+    "video": ["动作", "表情变化", "天气变化", "光效变化", "运镜", "视觉风格"],
+    "music": ["中国风", "流行", "说唱", "摇滚", "电音", "纯音乐"]
 }
 
-VALID_PROMPT_TYPES = ("image", "video")
+VALID_PROMPT_TYPES = ("image", "video", "music")
 
 # 内容长度上限
 MAX_TITLE_LEN = 50
@@ -103,6 +104,10 @@ def _load_categories() -> dict:
             "video": list(DEFAULT_CATEGORIES["video"])
         }
         db.save_data("prompt_categories.json", categories)
+    # 存量分类文件补全 music 分组（历史数据只有 image/video 键）
+    if "music" not in categories:
+        categories["music"] = list(DEFAULT_CATEGORIES["music"])
+        db.save_data("prompt_categories.json", categories)
     return categories
 
 
@@ -139,7 +144,7 @@ def _mask_prompt_text(prompt_data: dict, can_view: bool) -> dict:
 async def get_prompt_categories(prompt_type: str = None):
     """
     获取提示词分类列表
-    - prompt_type: image / video，不传则返回全部
+    - prompt_type: image / video / music，不传则返回全部
     """
     categories = _load_categories()
     if prompt_type in VALID_PROMPT_TYPES:
@@ -154,7 +159,7 @@ async def add_prompt_category(request: Request, body: PromptCategoryAdd, current
     添加提示词分类（登录用户均可，重名去重）
     """
     if body.prompt_type not in VALID_PROMPT_TYPES:
-        raise HTTPException(status_code=400, detail="分类所属类型必须是 image 或 video")
+        raise HTTPException(status_code=400, detail="分类所属类型必须是 image、video 或 music")
 
     name = (body.name or "").strip()
     if not name:
@@ -216,7 +221,7 @@ async def get_prompts(prompt_type: str = None, category: str = None,
                       page: int = 1, limit: int = 20, sort: str = "latest"):
     """
     获取提示词列表（分页，支持类型/分类筛选与多种排序）
-    - prompt_type: image / video（子界面筛选）
+    - prompt_type: image / video / music（子界面筛选）
     - category: 分类筛选
     - sort: latest / likes / favorites / views / daily_views / tips
     ⚠️ 列表接口永不下发 prompt_text（付费内容脱敏）
@@ -338,7 +343,7 @@ async def create_prompt(prompt: PromptCreate, current_user: str = Depends(requir
     """
     # 参数校验
     if prompt.prompt_type not in VALID_PROMPT_TYPES:
-        raise HTTPException(status_code=400, detail="提示词类型必须是 image 或 video")
+        raise HTTPException(status_code=400, detail="提示词类型必须是 image、video 或 music")
     title = (prompt.title or "").strip()
     content = (prompt.content or "").strip()
     prompt_text = (prompt.prompt_text or "").strip()