diff --git a/app/routers/tryon_router.py b/app/routers/tryon_router.py index 3c9fcd5..d985d49 100644 --- a/app/routers/tryon_router.py +++ b/app/routers/tryon_router.py @@ -14,6 +14,8 @@ from app.schemas.tryon import ( ) from app.services.dashscope_service import DashScopeService from app.utils.response import APIResponse +from app.services.qcloud_service import QCloudCOSService + # 加载环境变量 load_dotenv() @@ -209,7 +211,20 @@ async def check_tryon_status( # 如果任务完成,保存结果URL if db_tryon.task_status == "SUCCEEDED": - db_tryon.completion_url = status_response.get("image_url") + url = status_response.get("image_url") + # 下载图片并上传到自己的腾讯云 + # 下载图片 + response = httpx.get(url) + image_data = response.content + + # 上传到腾讯云 + # 获取腾讯云的配置 + qcloud_service = QCloudCOSService() + # 上传图片到腾讯云 + upload_result = qcloud_service.upload_file(image_data, "tryon_results") + + + db_tryon.completion_url = upload_result.get("url") db.commit() db.refresh(db_tryon)