小鱼开发
267 lines
9.1 KiB
Python
267 lines
9.1 KiB
Python
"""
|
|
脚本生成 API
|
|
============
|
|
|
|
提供脚本分类查询、润色、标题生成等功能。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import logging
|
|
import time
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.ai.model_router import get_model_router
|
|
from app.ai.prompts import list_categories, list_prompt_files, load_prompt, render_template
|
|
from app.api.deps import get_current_user
|
|
from app.core.exceptions import (
|
|
AITimeoutException,
|
|
InsufficientPointsException,
|
|
PlatformError,
|
|
PlatformErrorType,
|
|
)
|
|
from app.db.session import get_db
|
|
from app.models.user import User
|
|
from app.schemas.common import ApiResponse, success_response
|
|
from app.schemas.script import (
|
|
CategoryItem,
|
|
GenerateTitleRequest,
|
|
GenerateTitleResponse,
|
|
PolishRequest,
|
|
)
|
|
from app.services import point_service as ps
|
|
from app.services.script_service import get_script_service
|
|
|
|
router = APIRouter()
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _map_platform_error(e: PlatformError) -> HTTPException:
|
|
"""把第三方平台错误映射为用户友好的 HTTP 异常（带标准 error_code）。"""
|
|
if e.error_type == PlatformErrorType.CONTENT_VIOLATION:
|
|
return HTTPException(
|
|
status_code=400,
|
|
detail={
|
|
"message": "人物分镜台词未通过安全审核，请修改后重试",
|
|
"error_code": "content_violation",
|
|
},
|
|
)
|
|
if e.error_type == PlatformErrorType.RATE_LIMIT:
|
|
return HTTPException(
|
|
status_code=429,
|
|
detail={
|
|
"message": "当前请求过于频繁，请稍后再试",
|
|
"error_code": "rate_limit",
|
|
},
|
|
)
|
|
if e.error_type == PlatformErrorType.TIMEOUT:
|
|
return AITimeoutException("服务响应超时，请稍后重试")
|
|
if e.error_type == PlatformErrorType.AUTH_FAILED:
|
|
return HTTPException(
|
|
status_code=401,
|
|
detail={
|
|
"message": "第三方服务认证失败，请稍后重试或联系客服",
|
|
"error_code": "auth_failed",
|
|
},
|
|
)
|
|
if e.error_type == PlatformErrorType.SERVER_ERROR:
|
|
return HTTPException(
|
|
status_code=503,
|
|
detail={
|
|
"message": "第三方服务繁忙，请稍后重试",
|
|
"error_code": "server_error",
|
|
},
|
|
)
|
|
return HTTPException(
|
|
status_code=400,
|
|
detail={
|
|
"message": "请求失败，请检查后重试",
|
|
"error_code": e.error_type or "unknown",
|
|
},
|
|
)
|
|
|
|
|
|
@router.get("/categories", response_model=ApiResponse[list[CategoryItem]])
|
|
async def get_categories():
|
|
"""
|
|
获取提示词分类列表
|
|
|
|
返回所有大类及其下的提示词文件列表，供前端选择。
|
|
"""
|
|
categories = list_categories()
|
|
for cat in categories:
|
|
cat["files"] = list_prompt_files(cat["code"])
|
|
return success_response(
|
|
data=categories,
|
|
message="获取分类列表成功",
|
|
)
|
|
|
|
|
|
@router.post("/polish", response_model=ApiResponse[str])
|
|
async def polish_content(
|
|
request: PolishRequest,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
"""
|
|
AI 润色文案/画面描述
|
|
|
|
- `polishType=scene`: 润色画面描述（根据 shot_type 自动区分分镜/空镜）
|
|
- `polishType=voiceover`: 润色配音文本
|
|
|
|
参数:
|
|
- `shot_type`: "segment"（分镜）或 "empty_shot"（空镜），画面润色时必填
|
|
"""
|
|
service = get_script_service()
|
|
type_name = "画面" if request.polish_type == "scene" else "文案"
|
|
|
|
# 前置积分检查
|
|
required_points = ps._calculate_cost("polish")
|
|
check = await ps.check_balance(db, current_user.id, required_points)
|
|
if not check["sufficient"]:
|
|
raise InsufficientPointsException(
|
|
f"积分不足，需要 {required_points} 积分，当前余额 {check['balance']}"
|
|
)
|
|
|
|
try:
|
|
polished = await service.polish_content(
|
|
content=request.content,
|
|
polish_type=request.polish_type,
|
|
shot_type=request.shot_type or "segment",
|
|
)
|
|
|
|
# 扣费
|
|
points = ps._calculate_cost("polish")
|
|
await ps.consume(
|
|
db,
|
|
user_id=current_user.id,
|
|
points=points,
|
|
source_type="polish",
|
|
source_id=f"polish_{current_user.id}_{int(time.time() * 1000)}",
|
|
description="【文案润色】",
|
|
)
|
|
await db.commit()
|
|
|
|
return success_response(
|
|
data=polished,
|
|
message=f"{type_name}润色完成",
|
|
)
|
|
except InsufficientPointsException:
|
|
raise
|
|
except HTTPException:
|
|
raise
|
|
except PlatformError as e:
|
|
raise _map_platform_error(e)
|
|
except ValueError as e:
|
|
logger.warning(f"[Polish] 润色失败: {e}")
|
|
raise HTTPException(status_code=500, detail="润色失败，请检查输入内容后重试")
|
|
except Exception as e:
|
|
logger.error(f"[Polish] 润色异常: {e}")
|
|
raise HTTPException(status_code=500, detail=f"{type_name}润色失败，请稍后重试")
|
|
|
|
|
|
@router.post("/generate-title", response_model=ApiResponse[GenerateTitleResponse])
|
|
async def generate_title(
|
|
request: GenerateTitleRequest,
|
|
db: AsyncSession = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
"""
|
|
根据脚本内容智能生成标题
|
|
|
|
调用 LLM 根据脚本内容生成大标题或小标题。
|
|
提示词从文件加载，支持热更新。
|
|
"""
|
|
model_router = await get_model_router()
|
|
|
|
# 加载提示词
|
|
system_prompt = load_prompt("user/title_system")
|
|
user_template = load_prompt("user/title")
|
|
|
|
if not system_prompt or not user_template:
|
|
raise HTTPException(status_code=500, detail="标题生成提示词文件缺失")
|
|
|
|
# 根据使用场景确定描述
|
|
if request.usage == "cover":
|
|
usage_desc = "视频封面标题——用于封面图设计，是决定用户是否点击的第一要素"
|
|
style_requirement = "极具冲击力、抓眼球，适合静态封面大图展示，善用爆款句式"
|
|
usage_note = "- 封面主标题必须极度吸睛，让用户一眼就想点进去，善用数字、疑问、痛点、冲突\n- 封面副标题要补充悬念或细节，激发点击欲望"
|
|
else:
|
|
usage_desc = "视频画面标题——直接叠加在视频画面上，与动态视频内容共存"
|
|
style_requirement = "口语化、精炼有力，适合视频内展示，避免遮挡画面主体"
|
|
usage_note = "- 视频画面上的标题需要精炼，聚焦核心关键词\n- 副标题与主标题形成呼应，补充说明但不喧宾夺主"
|
|
|
|
# 渲染用户提示词
|
|
title_type_desc = (
|
|
"大标题（主标题，提炼核心卖点，吸睛）"
|
|
if request.title_type == "main"
|
|
else "小标题（副标题，补充说明或制造悬念）"
|
|
)
|
|
user_prompt = render_template(
|
|
user_template,
|
|
title_type=request.title_type,
|
|
title_type_desc=title_type_desc,
|
|
script_content=request.script_content,
|
|
max_length=request.max_length,
|
|
usage=request.usage,
|
|
usage_desc=usage_desc,
|
|
style_requirement=style_requirement,
|
|
usage_note=usage_note,
|
|
)
|
|
|
|
# 前置积分检查
|
|
required_points = ps._calculate_cost("title")
|
|
check = await ps.check_balance(db, current_user.id, required_points)
|
|
if not check["sufficient"]:
|
|
raise InsufficientPointsException(
|
|
f"积分不足，需要 {required_points} 积分，当前余额 {check['balance']}"
|
|
)
|
|
|
|
try:
|
|
async with asyncio.timeout(15):
|
|
result = await model_router.generate(
|
|
prompt=user_prompt,
|
|
system_prompt=system_prompt,
|
|
task_type="script",
|
|
max_tokens=64,
|
|
)
|
|
|
|
title = result.content.strip() if result.content else ""
|
|
# 去除可能的引号
|
|
title = title.strip('"').strip("'").strip("「」").strip("『』").strip("《》")
|
|
# 截断到最大长度
|
|
if len(title) > request.max_length:
|
|
title = title[: request.max_length]
|
|
|
|
# 扣费
|
|
points = ps._calculate_cost("title")
|
|
await ps.consume(
|
|
db,
|
|
user_id=current_user.id,
|
|
points=points,
|
|
source_type="title",
|
|
source_id=f"title_{current_user.id}_{int(time.time() * 1000)}",
|
|
description="【标题生成】",
|
|
)
|
|
await db.commit()
|
|
|
|
return success_response(
|
|
data=GenerateTitleResponse(title=title),
|
|
message="标题生成成功",
|
|
)
|
|
except InsufficientPointsException:
|
|
raise
|
|
except HTTPException:
|
|
raise
|
|
except PlatformError as e:
|
|
raise _map_platform_error(e)
|
|
except TimeoutError:
|
|
logger.warning("[generate_title] 标题生成超时")
|
|
raise AITimeoutException("标题生成超时，请稍后重试")
|
|
except Exception as e:
|
|
logger.error(f"[generate_title] 标题生成失败: {e}")
|
|
raise HTTPException(status_code=500, detail=f"标题生成失败: {str(e)}")
|