From adca35f7bfcde08571ea7248e1dc6e761867951c Mon Sep 17 00:00:00 2001
From: aaron <>
Date: Tue, 25 Mar 2025 12:05:48 +0800
Subject: [PATCH] update

---
 app/routers/tryon_router.py | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

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)