minor fix

src/api/nto_api.py

@@ -463,10 +463,11 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
         # Upload both images concurrently
         upload_tasks = [
             supabase_upload_and_return_url(prefix="necklace_try_on", image=result),
+            supabase_upload_and_return_url(prefix="necklace_try_on_mask", image=mask)
         ]
-        result_url = await asyncio.gather(*upload_tasks)
+        result_url, mask_url = await asyncio.gather(*upload_tasks)
 
-        if not result_url:
+        if not result_url or not mask_url:
             raise Exception("Failed to upload one or both images")
 
         logger.info(f">>> RESULT IMAGES SAVED IN {round((time.time() - start_time_saving), 2)}s <<<")
@@ -480,6 +481,7 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
             response = {
                 "code": 200,
                 "output": f"{result_url}",
+                "mask": f"{mask_url}",
                 "inference_time": total_backend_time
             }