add: supabase upload url return necklacetryon

app.py

@@ -22,4 +22,7 @@ app.add_middleware(
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
-)
+)
+
+import uvicorn
+uvicorn.run(app)

src/api/nto_api.py

@@ -3,6 +3,8 @@ project @ NTO-TCP-HF
 created @ 2024-10-28
 author  @ github/ishworrsubedii
 """
+import secrets
+import tempfile
 import time
 import cv2
 import numpy as np
@@ -23,6 +25,9 @@ from pydantic import BaseModel
 import replicate
 import requests
 from src.utils.logger import logger
+import secrets
+
+from batch_half_model_generation.supabase_image_fetch import supabase_client
 
 pipeline = Pipeline()
 
@@ -36,8 +41,8 @@ supabase_storage: str = os.getenv("SUPABASE_STORAGE")
 cto_replicate: str = os.getenv(
     "CTO")
 
-supabase: Client = create_client(supabase_key=key, supabase_url=url)
-bucket = supabase.storage.from_(supabase_storage)
+supabase = create_client(supabase_url=url, supabase_key=key)
+bucket = supabase.storage.from_("JewelMirrorOutputs")
 
 
 def replicate_run_cto(input):
@@ -357,6 +362,33 @@ async def parse_necklace_try_on_id(necklaceImageId: str = Form(...),
     )
 
 
+def supabase_upload_and_return_url(prefix: str, image: Image.Image,quality: int = 85) :
+
+    try:
+        filename = f"{prefix}_{secrets.token_hex(16)}.webp"
+
+
+
+
+        buffer = BytesIO()
+        image.save(buffer, format='WEBP', quality=quality, optimize=True)
+        image_bytes = buffer.getvalue()
+
+        bucket.upload(
+            path=filename,
+            file=image_bytes,
+        )
+
+        return bucket.get_public_url(filename)
+
+    except Exception as e:
+        print(f"Failed to upload image: {str(e)}")
+        return None
+    finally:
+        if 'buffer' in locals():
+            buffer.close()
+
+
 @nto_cto_router.post("/necklaceTryOnID")
 async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends(parse_necklace_try_on_id),
                              image: UploadFile = File(...)):
@@ -403,12 +435,14 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
                             status_code=500)
 
     try:
-        inMemFile = BytesIO()
-        inMemFileMask = BytesIO()
-        result.save(inMemFile, format="WEBP", quality=85)
-        mask.save(inMemFileMask, format="WEBP", quality=85)
-        outputBytes = inMemFile.getvalue()
-        maskBytes = inMemFileMask.getvalue()
+        logger.info(">>> SAVING RESULT IMAGES <<<")
+        start_time_saving = time.time()
+
+        result_url = supabase_upload_and_return_url(prefix="necklace_try_on", image=result)
+        mask_url = supabase_upload_and_return_url(prefix="necklace_try_on_mask", image=mask)
+
+        logger.info(f">>> RESULT IMAGES SAVED IN {round((time.time() - start_time_saving), 2)}s <<<")
+
         logger.info(">>> RESULT IMAGES SAVED <<<")
     except Exception as e:
         logger.error(f">>> RESULT SAVING ERROR: {str(e)} <<<")
@@ -419,8 +453,8 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
         total_backend_time = round((time.time() - start_time), 2)
         response = {
             "code": 200,
-            "output": f"data:image/WEBP;base64,{base64.b64encode(outputBytes).decode('utf-8')}",
-            "mask": f"data:image/WEBP;base64,{base64.b64encode(maskBytes).decode('utf-8')}",
+            "output": f"{result_url}",
+            "mask": f"{mask_url}",
             "inference_time": total_backend_time
         }
         if creditResponse == "No Credits Available":