from fastapi import APIRouter, HTTPException, Depends, Query, Path, BackgroundTasks from sqlalchemy.orm import Session from typing import List, Optional import httpx import logging import os from dotenv import load_dotenv from app.database import get_db from app.models.tryon import TryOn from app.schemas.tryon import ( TryOnCreate, TryOnUpdate, TryOnResponse, AiTryonRequest, AiTryonResponse, TaskInfo ) from app.services.dashscope_service import DashScopeService from app.utils.response import APIResponse # 加载环境变量 load_dotenv() logger = logging.getLogger(__name__) router = APIRouter() # 从环境变量获取API密钥 DASHSCOPE_API_KEY = os.getenv("DASHSCOPE_API_KEY") DASHSCOPE_API_URL = "https://dashscope.aliyuncs.com/api/v1/services/aigc/image2image/image-synthesis" @router.post("", response_model=TryOnResponse, status_code=201) async def create_tryon( tryon_data: TryOnCreate, background_tasks: BackgroundTasks, db: Session = Depends(get_db) ): """ 创建一个试穿记录并发送到阿里百炼平台 - **top_garment_url**: 上衣图片URL（可选） - **bottom_garment_url**: 下衣图片URL（可选） - **person_image_url**: 人物图片URL（必填） 注意：上衣和下衣至少需要提供一个 """ if not tryon_data.top_garment_url and not tryon_data.bottom_garment_url: raise HTTPException(status_code=400, detail="上衣和下衣图片至少需要提供一个") try: # 创建试穿记录 db_tryon = TryOn( top_garment_url=tryon_data.top_garment_url, bottom_garment_url=tryon_data.bottom_garment_url, person_image_url=tryon_data.person_image_url ) db.add(db_tryon) db.commit() db.refresh(db_tryon) # 在后台发送请求到阿里百炼平台 background_tasks.add_task( send_tryon_request, db=db, tryon_id=db_tryon.id, top_garment_url=tryon_data.top_garment_url, bottom_garment_url=tryon_data.bottom_garment_url, person_image_url=tryon_data.person_image_url ) return db_tryon except Exception as e: logger.error(f"创建试穿记录失败: {str(e)}") db.rollback() raise HTTPException(status_code=500, detail=f"创建试穿记录失败: {str(e)}") async def send_tryon_request( db: Session, tryon_id: int, top_garment_url: Optional[str], bottom_garment_url: Optional[str], person_image_url: str ): """发送试穿请求到阿里百炼平台""" try: # 创建DashScopeService实例 dashscope_service = DashScopeService() # 调用服务发送试穿请求 response = await dashscope_service.generate_tryon( person_image_url=person_image_url, top_garment_url=top_garment_url, bottom_garment_url=bottom_garment_url ) # 更新数据库记录 db_tryon = db.query(TryOn).filter(TryOn.id == tryon_id).first() if db_tryon: db_tryon.request_id = response.get("request_id") db_tryon.task_id = response.get("task_id") db_tryon.task_status = response.get("status") db.commit() logger.info(f"试穿请求发送成功，任务ID: {db_tryon.task_id}") except Exception as e: logger.error(f"发送试穿请求异常: {str(e)}") # 更新数据库记录状态 db_tryon = db.query(TryOn).filter(TryOn.id == tryon_id).first() if db_tryon: db_tryon.task_status = "ERROR" db.commit() @router.get("", response_model=List[TryOnResponse]) async def get_tryons( skip: int = Query(0, description="跳过的记录数量"), limit: int = Query(100, description="返回的最大记录数量"), db: Session = Depends(get_db) ): """ 获取试穿记录列表，支持分页 - **skip**: 跳过的记录数量，用于分页 - **limit**: 返回的最大记录数量，用于分页 """ try: tryons = db.query(TryOn).order_by(TryOn.id.desc()).offset(skip).limit(limit).all() return tryons except Exception as e: logger.error(f"获取试穿记录列表失败: {str(e)}") raise HTTPException(status_code=500, detail=f"获取试穿记录列表失败: {str(e)}") @router.get("/{tryon_id}", response_model=TryOnResponse) async def get_tryon( tryon_id: int = Path(..., description="试穿记录ID"), db: Session = Depends(get_db) ): """ 根据ID获取试穿记录详情 - **tryon_id**: 试穿记录ID """ try: tryon = db.query(TryOn).filter(TryOn.id == tryon_id).first() if not tryon: raise HTTPException(status_code=404, detail=f"未找到ID为{tryon_id}的试穿记录") return tryon except HTTPException: raise except Exception as e: logger.error(f"获取试穿记录详情失败: {str(e)}") raise HTTPException(status_code=500, detail=f"获取试穿记录详情失败: {str(e)}") @router.put("/{tryon_id}", response_model=TryOnResponse) async def update_tryon( tryon_id: int = Path(..., description="试穿记录ID"), tryon_data: TryOnUpdate = None, db: Session = Depends(get_db) ): """ 更新试穿记录信息 - **tryon_id**: 试穿记录ID - **task_status**: 任务状态（可选） - **completion_url**: 生成图片URL（可选） """ try: db_tryon = db.query(TryOn).filter(TryOn.id == tryon_id).first() if not db_tryon: raise HTTPException(status_code=404, detail=f"未找到ID为{tryon_id}的试穿记录") # 更新提供的字段 if tryon_data.task_status is not None: db_tryon.task_status = tryon_data.task_status if tryon_data.completion_url is not None: db_tryon.completion_url = tryon_data.completion_url db.commit() db.refresh(db_tryon) return db_tryon except HTTPException: raise except Exception as e: logger.error(f"更新试穿记录失败: {str(e)}") db.rollback() raise HTTPException(status_code=500, detail=f"更新试穿记录失败: {str(e)}") @router.post("/{tryon_id}/check", response_model=TryOnResponse) async def check_tryon_status( tryon_id: int = Path(..., description="试穿记录ID"), db: Session = Depends(get_db), dashscope_service: DashScopeService = Depends(lambda: DashScopeService()) ): """ 检查试穿任务状态 - **tryon_id**: 试穿记录ID """ try: db_tryon = db.query(TryOn).filter(TryOn.id == tryon_id).first() if not db_tryon: raise HTTPException(status_code=404, detail=f"未找到ID为{tryon_id}的试穿记录") if not db_tryon.task_id: raise HTTPException(status_code=400, detail=f"试穿记录未包含任务ID") # 调用DashScopeService检查任务状态 try: status_response = await dashscope_service.check_tryon_status(db_tryon.task_id) # 更新数据库记录 db_tryon.task_status = status_response.get("status") # 如果任务完成，保存结果URL if db_tryon.task_status == "SUCCEEDED": db_tryon.completion_url = status_response.get("image_url") db.commit() db.refresh(db_tryon) logger.info(f"试穿任务状态更新: {db_tryon.task_status}") except Exception as e: logger.error(f"调用DashScope API检查任务状态失败: {str(e)}") return db_tryon except HTTPException: raise except Exception as e: logger.error(f"检查试穿任务状态失败: {str(e)}") raise HTTPException(status_code=500, detail=f"检查试穿任务状态失败: {str(e)}")