Update utils.py

utils.py

@@ -5,7 +5,7 @@ import requests
 import json
 import base64
 
-from PIL import Image, ImageChops
+from PIL import Image
 
 
 jwt_token = ''
@@ -91,7 +91,7 @@ def generate_novelai_image(
     if selection == 'inp' and inp_img['background'].getextrema()[3][1] > 0:
         payload['action'] = "infill"
         payload['model'] = 'nai-diffusion-3-inpainting'
-        payload['parameters']['mask'] = image2base64(ImageChops.invert(inp_img['layers'][0]))
+        payload['parameters']['mask'] = image2base64(convert_mask(inp_img['layers'][0]))
         payload['parameters']['image'] = image2base64(inp_img['background'])
         payload['parameters']['extra_noise_seed'] = seed
     if i2i_image is not None and selection == 'i2i':
@@ -131,4 +131,10 @@ def image2base64(img):
     img.save(output_buffer, format='PNG' if img.mode=='RGBA' else 'JPEG')
     byte_data = output_buffer.getvalue()
     base64_str = base64.b64encode(byte_data).decode()
-    return base64_str
+    return base64_str
+
+def convert_mask(img):
+    a = img.split()[3]
+    base = Image.new('RGB', img.size, 'white')
+    base.putalpha(a)
+    return base