102 lines
3.7 KiB
Python
102 lines
3.7 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
海报API模型定义
|
||
"""
|
||
|
||
from typing import List, Dict, Any, Optional, Tuple
|
||
from pydantic import BaseModel, Field
|
||
|
||
|
||
class PosterRequest(BaseModel):
|
||
"""海报生成请求模型"""
|
||
content: Dict[str, Any] = Field(..., description="内容数据，包含标题、正文等")
|
||
topic_index: str = Field(..., description="主题索引，用于文件命名")
|
||
template_name: Optional[str] = Field(None, description="模板名称，如果为None则根据配置选择")
|
||
|
||
class Config:
|
||
schema_extra = {
|
||
"example": {
|
||
"content": {
|
||
"title": "【北京故宫】避开人潮的秘密路线，90%的人都不知道！",
|
||
"content": "故宫，作为中国最著名的文化遗产之一...",
|
||
"tag": ["北京旅游", "故宫", "旅游攻略", "避暑胜地"]
|
||
},
|
||
"topic_index": "1",
|
||
"template_name": "vibrant"
|
||
}
|
||
}
|
||
|
||
|
||
class PosterResponse(BaseModel):
|
||
"""海报生成响应模型"""
|
||
request_id: str = Field(..., description="请求ID")
|
||
topic_index: str = Field(..., description="主题索引")
|
||
poster_path: str = Field(..., description="生成的海报文件路径")
|
||
template_name: str = Field(..., description="使用的模板名称")
|
||
|
||
class Config:
|
||
schema_extra = {
|
||
"example": {
|
||
"request_id": "poster-20240715-123456-a1b2c3d4",
|
||
"topic_index": "1",
|
||
"poster_path": "/result/run_20230715_123456/topic_1/poster_vibrant.png",
|
||
"template_name": "vibrant"
|
||
}
|
||
}
|
||
|
||
|
||
class TemplateListResponse(BaseModel):
|
||
"""模板列表响应模型"""
|
||
templates: List[str] = Field(..., description="可用的模板列表")
|
||
default_template: str = Field(..., description="默认模板")
|
||
|
||
class Config:
|
||
schema_extra = {
|
||
"example": {
|
||
"templates": ["vibrant", "business", "collage"],
|
||
"default_template": "vibrant"
|
||
}
|
||
}
|
||
|
||
|
||
class PosterTextRequest(BaseModel):
|
||
"""海报文案生成请求模型"""
|
||
system_prompt: str = Field(..., description="系统提示词")
|
||
user_prompt: str = Field(..., description="用户提示词")
|
||
context_data: Optional[Dict[str, Any]] = Field(None, description="上下文数据，用于填充提示词中的占位符")
|
||
temperature: Optional[float] = Field(0.3, description="生成温度参数")
|
||
top_p: Optional[float] = Field(0.4, description="top_p参数")
|
||
|
||
class Config:
|
||
schema_extra = {
|
||
"example": {
|
||
"system_prompt": "你是一位专业的旅游海报文案撰写专家...",
|
||
"user_prompt": "请为{location}的{attraction}创作一段简短有力的海报文案...",
|
||
"context_data": {
|
||
"location": "北京",
|
||
"attraction": "故宫"
|
||
},
|
||
"temperature": 0.3,
|
||
"top_p": 0.4
|
||
}
|
||
}
|
||
|
||
|
||
class PosterTextResponse(BaseModel):
|
||
"""海报文案生成响应模型"""
|
||
request_id: str = Field(..., description="请求ID")
|
||
text_content: Dict[str, Any] = Field(..., description="生成的文案内容")
|
||
|
||
class Config:
|
||
schema_extra = {
|
||
"example": {
|
||
"request_id": "text-20240715-123456-a1b2c3d4",
|
||
"text_content": {
|
||
"title": "紫禁城的秘密",
|
||
"subtitle": "600年历史，等你探索",
|
||
"description": "穿越时光，触摸历史，感受帝王的荣耀与辉煌"
|
||
}
|
||
}
|
||
} |