ga tau nama commit apa

main.py

@@ -62,6 +62,7 @@ async def predict_image(file: UploadFile = File(...)):
         # change the mask to base64 and calculate the score
         for i in range(len(results)):
             mask_area = calculate_mask_area(results[i]["mask"])
+            print(results[i]["mask"].size)
             processed_image = process_image(results[i]["mask"])
             results[i]["mask"] = image_to_base64(processed_image, "PNG")
             if results[i]["label"] == "background":
@@ -90,4 +91,4 @@ async def predict_image(file: UploadFile = File(...)):
     except Exception as e:
         # Mendapatkan stack trace
         error_traceback = traceback.format_exc()
-        raise HTTPException(status_code=500, detail=f"An error occurred: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"An error occurred: {error_traceback}")

utils/helpers.py

@@ -38,5 +38,6 @@ def combine_images(original_image: Image.Image, masks: list) -> Image.Image:
         mask_image = Image.open(io.BytesIO(base64.b64decode(mask['mask'])))
         mask_image = mask_image.convert("L")  # Convert mask to grayscale
         mask_image = ImageEnhance.Brightness(mask_image).enhance(0.5)  # Adjust the brightness to make it more visible
-        combined = ImageChops.add(combined, mask_image)
+        color_mask = Image.new("RGBA", original_image.size, (255, 0, 0, 128))  # Red color with transparency
+        combined.paste(color_mask, (0, 0), mask_image)
     return combined