67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
|
"""项目封面生成与下载 API"""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
from fastapi import APIRouter, Depends, HTTPException, Request
|
||
|
|
from fastapi.responses import FileResponse
|
||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
|
|
||
|
|
from app.database import get_db
|
||
|
|
from app.services.cover_generation_service import cover_generation_service
|
||
|
|
|
||
|
|
router = APIRouter(prefix="/projects", tags=["项目封面"])
|
||
|
|
|
||
|
|
|
||
|
|
class CoverGenerateRequest(BaseModel):
|
||
|
|
overwrite: bool = Field(default=True, description="是否覆盖已有封面")
|
||
|
|
|
||
|
|
|
||
|
|
class CoverGenerateResponse(BaseModel):
|
||
|
|
project_id: str
|
||
|
|
cover_status: str
|
||
|
|
cover_image_url: str | None = None
|
||
|
|
cover_prompt: str | None = None
|
||
|
|
provider: str | None = None
|
||
|
|
model: str | None = None
|
||
|
|
message: str
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/{project_id}/cover/generate", response_model=CoverGenerateResponse, summary="生成项目封面")
|
||
|
|
async def generate_project_cover(
|
||
|
|
project_id: str,
|
||
|
|
payload: CoverGenerateRequest,
|
||
|
|
request: Request,
|
||
|
|
db: AsyncSession = Depends(get_db),
|
||
|
|
):
|
||
|
|
user_id = getattr(request.state, "user_id", None)
|
||
|
|
if not user_id:
|
||
|
|
raise HTTPException(status_code=401, detail="未登录")
|
||
|
|
|
||
|
|
result = await cover_generation_service.generate_cover(
|
||
|
|
db=db,
|
||
|
|
user_id=user_id,
|
||
|
|
project_id=project_id,
|
||
|
|
overwrite=payload.overwrite,
|
||
|
|
)
|
||
|
|
return CoverGenerateResponse(**result)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/{project_id}/cover/download", summary="下载项目封面")
|
||
|
|
async def download_project_cover(
|
||
|
|
project_id: str,
|
||
|
|
request: Request,
|
||
|
|
db: AsyncSession = Depends(get_db),
|
||
|
|
):
|
||
|
|
user_id = getattr(request.state, "user_id", None)
|
||
|
|
if not user_id:
|
||
|
|
raise HTTPException(status_code=401, detail="未登录")
|
||
|
|
|
||
|
|
project, file_path = await cover_generation_service.get_cover_download_path(
|
||
|
|
db=db,
|
||
|
|
user_id=user_id,
|
||
|
|
project_id=project_id,
|
||
|
|
)
|
||
|
|
suffix = file_path.suffix or ".png"
|
||
|
|
filename = f"{project.title}-cover{suffix}"
|
||
|
|
return FileResponse(path=file_path, filename=filename, media_type="application/octet-stream")
|