update: custom error messages

src/api/nto_api.py

@@ -66,7 +66,7 @@ async def clothing_try_on_v2(image: UploadFile = File(...), clothing_type: str =
         logger.info(">>> IMAGE LOADED SUCCESSFULLY <<<")
     except Exception as e:
         logger.error(f">>> IMAGE LOADING ERROR: {str(e)} <<<")
-        return JSONResponse(status_code=500, content={"error": f"Error reading image: {str(e)}", "code": 500})
+        return JSONResponse(status_code=500, content={"error": f"Error reading image", "code": 500})
 
     try:
         mask = await pipeline.shoulderPointMaskGeneration_(image=image)
@@ -74,7 +74,7 @@ async def clothing_try_on_v2(image: UploadFile = File(...), clothing_type: str =
     except Exception as e:
         logger.error(f">>> MASK GENERATION ERROR: {str(e)} <<<")
         return JSONResponse(status_code=500,
-                          content={"error": f"Error generating mask: {str(e)}", "code": 500})
+                            content={"error": f"Error generating mask", "code": 500})
 
     try:
         mask_img_base_64, act_img_base_64 = BytesIO(), BytesIO()
@@ -89,7 +89,7 @@ async def clothing_try_on_v2(image: UploadFile = File(...), clothing_type: str =
     except Exception as e:
         logger.error(f">>> IMAGE ENCODING ERROR: {str(e)} <<<")
         return JSONResponse(status_code=500,
-                          content={"error": f"Error converting images to base64: {str(e)}", "code": 500})
+                            content={"error": f"Error converting images to base64", "code": 500})
 
     input = {
         "mask": mask_data_uri,
@@ -104,7 +104,7 @@ async def clothing_try_on_v2(image: UploadFile = File(...), clothing_type: str =
         logger.info(">>> REPLICATE PROCESSING COMPLETED <<<")
     except Exception as e:
         logger.error(f">>> REPLICATE PROCESSING ERROR: {str(e)} <<<")
-        return JSONResponse(content={"error": f"Error running CTO Replicate: {str(e)}", "code": 500}, status_code=500)
+        return JSONResponse(content={"error": f"Error running CTO Replicate", "code": 500}, status_code=500)
 
     total_inference_time = round((time.time() - start_time), 2)
     logger.info(f">>> TOTAL INFERENCE TIME: {total_inference_time}s <<<")
@@ -134,7 +134,7 @@ async def clothing_try_on(image: UploadFile = File(...),
         logger.info(">>> IMAGES LOADED SUCCESSFULLY <<<")
     except Exception as e:
         logger.error(f">>> IMAGE LOADING ERROR: {str(e)} <<<")
-        return JSONResponse(status_code=500, content={"error": f"Error reading image or mask: {str(e)}", "code": 500})
+        return JSONResponse(status_code=500, content={"error": f"Error reading image or mask", "code": 500})
 
     try:
         actual_image = image.copy()
@@ -153,7 +153,7 @@ async def clothing_try_on(image: UploadFile = File(...),
     except Exception as e:
         logger.error(f">>> IMAGE PROCESSING ERROR: {str(e)} <<<")
         return JSONResponse(status_code=500,
-                            content={"error": f"Error processing image or mask: {str(e)}", "code": 500})
+                            content={"error": f"Error processing image or mask", "code": 500})
 
     try:
         mask_img_base_64, act_img_base_64 = BytesIO(), BytesIO()
@@ -168,7 +168,7 @@ async def clothing_try_on(image: UploadFile = File(...),
     except Exception as e:
         logger.error(f">>> IMAGE ENCODING ERROR: {str(e)} <<<")
         return JSONResponse(status_code=500,
-                            content={"error": f"Error encoding images to base64: {str(e)}", "code": 500})
+                            content={"error": f"Error encoding images", "code": 500})
 
     input = {
         "mask": mask_data_uri,
@@ -183,7 +183,7 @@ async def clothing_try_on(image: UploadFile = File(...),
         logger.info(">>> REPLICATE PROCESSING COMPLETED <<<")
     except Exception as e:
         logger.error(f">>> REPLICATE PROCESSING ERROR: {str(e)} <<<")
-        return JSONResponse(content={"error": f"Error running CTO Replicate: {str(e)}", "code": 500}, status_code=500)
+        return JSONResponse(content={"error": f"Error running clothing try on", "code": 500}, status_code=500)
 
     try:
         response = requests.get(output[0])
@@ -205,7 +205,7 @@ async def clothing_try_on(image: UploadFile = File(...),
         }
     except Exception as e:
         logger.error(f">>> OUTPUT IMAGE PROCESSING ERROR: {str(e)} <<<")
-        return JSONResponse(status_code=500, content={"error": f"Error processing output image: {str(e)}", "code": 500})
+        return JSONResponse(status_code=500, content={"error": f"Error processing output image", "code": 500})
 
     logger.info(f">>> TOTAL INFERENCE TIME: {total_inference_time}s <<<")
     logger.info(">>> REQUEST COMPLETED SUCCESSFULLY <<<")
@@ -213,6 +213,7 @@ async def clothing_try_on(image: UploadFile = File(...),
 
     return JSONResponse(content=response, status_code=200)
 
+
 @nto_cto_router.post("/productData/{storeId}")
 async def product_data(
         storeId: str,
@@ -373,7 +374,7 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
         logger.info(">>> API KEY VALIDATION SUCCESSFUL <<<")
     except Exception as e:
         logger.error(f">>> API KEY VALIDATION ERROR: {str(e)} <<<")
-        return JSONResponse(content={"error": f"Error validating API key: {str(e)}", "code": 500}, status_code=500)
+        return JSONResponse(content={"error": f"Error validating API key", "code": 500}, status_code=500)
 
     try:
         imageBytes = await image.read()
@@ -383,16 +384,22 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
     except Exception as e:
         logger.error(f">>> IMAGE LOADING ERROR: {str(e)} <<<")
         return JSONResponse(content={
-            "error": f"The requested resource (Image, necklace category, or store) is not available. Please verify the availability and try again. Error: {str(e)}",
+            "error": f"The requested resource (Image, necklace category, or store) is not available. Please verify the availability and try again",
             "code": 404}, status_code=404)
 
     try:
         result, headetText, mask = await pipeline.necklaceTryOn_(image=image, jewellery=jewellery,
                                                                  storename=necklace_try_on_id.storename)
+
+        if result is None:
+            logger.error(">>> NO FACE DETECTED IN THE IMAGE <<<")
+            return JSONResponse(
+                content={"error": "No face detected in the image please try again with a different image",
+                         "code": 400}, status_code=400)
         logger.info(">>> NECKLACE TRY ON PROCESSING COMPLETED <<<")
     except Exception as e:
         logger.error(f">>> NECKLACE TRY ON PROCESSING ERROR: {str(e)} <<<")
-        return JSONResponse(content={"error": f"Error during necklace try-on process: {str(e)}", "code": 500},
+        return JSONResponse(content={"error": f"Error during necklace try-on process", "code": 500},
                             status_code=500)
 
     try:
@@ -405,7 +412,7 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
         logger.info(">>> RESULT IMAGES SAVED <<<")
     except Exception as e:
         logger.error(f">>> RESULT SAVING ERROR: {str(e)} <<<")
-        return JSONResponse(content={"error": f"Error saving result images: {str(e)}", "code": 500}, status_code=500)
+        return JSONResponse(content={"error": f"Error saving result images", "code": 500}, status_code=500)
 
     try:
         creditResponse = deductAndTrackCredit(storename=necklace_try_on_id.storename, endpoint="/necklaceTryOnID")
@@ -423,7 +430,7 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
         logger.info(">>> CREDITS DEDUCTED SUCCESSFULLY <<<")
     except Exception as e:
         logger.error(f">>> CREDIT DEDUCTION ERROR: {str(e)} <<<")
-        return JSONResponse(content={"error": f"Error deducting credits: {str(e)}", "code": 500}, status_code=500)
+        return JSONResponse(content={"error": f"Error deducting credits", "code": 500}, status_code=500)
 
     logger.info(f">>> TOTAL INFERENCE TIME: {total_backend_time}s <<<")
     logger.info(f">>> NECKLACE TRY ON COMPLETED :: {necklace_try_on_id.storename} <<<")
@@ -431,6 +438,7 @@ async def necklace_try_on_id(necklace_try_on_id: NecklaceTryOnIDEntity = Depends
 
     return JSONResponse(content=response, status_code=200)
 
+
 @nto_cto_router.post("/canvasPoints")
 async def canvas_points(necklace_try_on_id: NecklaceTryOnIDEntity = Depends(parse_necklace_try_on_id),
                         image: UploadFile = File(...)):
@@ -446,7 +454,7 @@ async def canvas_points(necklace_try_on_id: NecklaceTryOnIDEntity = Depends(pars
     except Exception as e:
         logger.error(f">>> IMAGE LOADING ERROR: {str(e)} <<<")
         return JSONResponse(content={
-            "error": f"The requested resource (Image, necklace category, or store) is not available. Please verify the availability and try again. Error: {str(e)}",
+            "error": f"The requested resource (Image, necklace category, or store) is not available. Please verify the availability and try again. Error",
             "code": 404}, status_code=404)
 
     try:
@@ -455,7 +463,7 @@ async def canvas_points(necklace_try_on_id: NecklaceTryOnIDEntity = Depends(pars
         logger.info(">>> CANVAS POINTS PROCESSING COMPLETED <<<")
     except Exception as e:
         logger.error(f">>> CANVAS POINTS PROCESSING ERROR: {str(e)} <<<")
-        return JSONResponse(content={"error": f"Error during canvas point process: {str(e)}", "code": 500},
+        return JSONResponse(content={"error": f"Error during canvas point process", "code": 500},
                             status_code=500)
 
     try:
@@ -466,7 +474,7 @@ async def canvas_points(necklace_try_on_id: NecklaceTryOnIDEntity = Depends(pars
         logger.info(">>> CREDITS DEDUCTED SUCCESSFULLY <<<")
     except Exception as e:
         logger.error(f">>> CREDIT DEDUCTION ERROR: {str(e)} <<<")
-        return JSONResponse(content={"error": f"Error deducting credits: {str(e)}", "code": 500}, status_code=500)
+        return JSONResponse(content={"error": f"Error deducting credits", "code": 500}, status_code=500)
 
     total_inference_time = round((time.time() - start_time), 2)
     logger.info(f">>> TOTAL INFERENCE TIME: {total_inference_time}s <<<")
@@ -506,7 +514,7 @@ async def necklace_try_on_with_points(necklace_try_on_id: NecklaceTryOnIDEntity
         logger.info(">>> NECKLACE TRY ON PROCESSING COMPLETED <<<")
     except Exception as e:
         logger.error(f">>> NECKLACE TRY ON PROCESSING ERROR: {str(e)} <<<")
-        return JSONResponse(content={"error": f"Error during necklace try-on process: {str(e)}", "code": 500},
+        return JSONResponse(content={"error": f"Error during necklace try-on process", "code": 500},
                             status_code=500)
 
     try:
@@ -519,7 +527,7 @@ async def necklace_try_on_with_points(necklace_try_on_id: NecklaceTryOnIDEntity
         logger.info(">>> RESULT IMAGES SAVED <<<")
     except Exception as e:
         logger.error(f">>> RESULT SAVING ERROR: {str(e)} <<<")
-        return JSONResponse(content={"error": f"Error saving result images: {str(e)}", "code": 500}, status_code=500)
+        return JSONResponse(content={"error": f"Error saving result images", "code": 500}, status_code=500)
 
     try:
         creditResponse = deductAndTrackCredit(storename=necklace_try_on_id.storename, endpoint="/necklaceTryOnID")
@@ -537,7 +545,7 @@ async def necklace_try_on_with_points(necklace_try_on_id: NecklaceTryOnIDEntity
         logger.info(">>> CREDITS DEDUCTED SUCCESSFULLY <<<")
     except Exception as e:
         logger.error(f">>> CREDIT DEDUCTION ERROR: {str(e)} <<<")
-        return JSONResponse(content={"error": f"Error deducting credits: {str(e)}", "code": 500}, status_code=500)
+        return JSONResponse(content={"error": f"Error deducting credits", "code": 500}, status_code=500)
 
     logger.info(f">>> TOTAL INFERENCE TIME: {total_inference_time}s <<<")
     logger.info(f">>> NECKLACE TRY ON WITH POINTS COMPLETED :: {necklace_try_on_id.storename} <<<")

src/components/necklaceTryOn.py

@@ -92,8 +92,9 @@ class NecklaceTryOn:
             return [result, headerText, mask]
 
         except Exception as e:
+            logger.error(f">>> NECKLACE TRY ON ERROR: {str(e)} <<<")
             logger.error(f"{CustomException(e)}:: {storename}")
-            raise CustomException(e)
+            return [None, "error", None]
 
     def necklaceTryOnV3(self, image: Image.Image, jewellery: Image.Image, storename: str) -> list[
         Union[Image.Image, str]]: