Update utils.py

utils.py

@@ -44,7 +44,10 @@ def generate_novelai_image(
     ref_str=0.6,
     i2i_image=None,
     i2i_str=0.7,
-    i2i_noise=0
+    i2i_noise=0,
+    overlay=True,
+    inp_img=None,
+    selection='i2i'
 ):
     # Assign a random seed if seed is -1
     if seed == -1:
@@ -80,20 +83,18 @@ def generate_novelai_image(
             "sm": smea,
             "sm_dyn": dyn,
             "uncond_scale": 1,
+            "overlay": True
         }
     }
     if ref_image is not None:
         payload['parameters']['reference_image'] = image2base64(ref_image)
-    '''
-    if use_inp:
+    if selection == 'inp' and inp_img is not None:
         payload['action'] = "infill"
         payload['model'] = 'nai-diffusion-3-inpainting'
         payload['parameters']['mask'] = image2base64(inp_img['layers'][0])
         payload['parameters']['image'] = image2base64(inp_img['background'])
         payload['parameters']['extra_noise_seed'] = seed
-        payload['parameters']['strength'] = inp_str
-    '''
-    if i2i_image is not None:
+    if i2i_image is not None and selection == 'i2i':
         payload['action'] = "img2img"
         payload['parameters']['image'] = image2base64(i2i_image)
         payload['parameters']['strength'] = i2i_str