update response with presigend url (parsed url)

Author: swwho96

services/detected_service.py

@@ -6,6 +6,7 @@
 from core.s3_utils import download_image, upload_image
 import core.load_model as load_model
 from schemas.common import ErrorCode
+from urllib.parse import urlparse, urlunparse
 
 async def detect_cloth(download_url:str, upload_urls:List[str]):
 
@@ -60,7 +61,13 @@ async def worker(idx, img):
             *[worker(idx, item) for idx, item in enumerate(crop_items)],
             return_exceptions=True
         )
-        uploaded_urls = [upload_urls[idx] for idx, r in enumerate(results) if r["success"]]
+        uploaded_urls = []
+        for idx, r in enumerate(results):
+            if r["success"]:
+                parsed = urlparse(upload_urls[idx])
+                clean_url = urlunparse((parsed.scheme, parsed.netloc, parsed.path, '', '', ''))
+                uploaded_urls.append(clean_url)
+
         if not uploaded_urls:
             return {
                 "success": False,

services/predict_services.py

@@ -8,6 +8,7 @@
 from core.s3_utils import download_image, upload_image
 import core.load_model as load_model
 from rembg import remove
+from urllib.parse import urlparse, urlunparse
 from core.config import CATEGORY_EN_TO_ID, CATEGORY_EN_TO_KO, SITUATION_EN_TO_ID, SITUATION_EN_TO_KO, STYLE_EN_TO_ID, STYLE_EN_TO_KO, SEASON_EN_TO_ID, SEASON_EN_TO_KO
 
 
@@ -57,7 +58,10 @@ async def classify_info_with_session(session: aiohttp.ClientSession, download_ur
             labels[name] = [{"id":CATEGORY_EN_TO_ID[texts[i]], "name":CATEGORY_EN_TO_KO[texts[i]]} for i in idx.squeeze(0).tolist()]
         if name == "seasons":
             labels[name] = [{"id":SEASON_EN_TO_ID[texts[i]], "name":SEASON_EN_TO_KO[texts[i]]} for i in idx.squeeze(0).tolist()]
-
+    
+    parsed = urlparse(upload_url)
+    clean_url = urlunparse((parsed.scheme, parsed.netloc, parsed.path, '', '', ''))
+    labels["uploaded_url"] = clean_url
     return {
         "isSuccess":True,
         "result": labels