Update utils.py

utils.py

@@ -42,10 +42,9 @@ def generate_novelai_image(
     ref_image=None,
     info_extract=1,
     ref_str=0.6,
-    inp_img=None,
-    overlay=False,
-    use_inp=False,
-    inp_str=0.7
+    i2i_image=None,
+    i2i_str=0.7,
+    i2i_noise=0
 ):
     # Assign a random seed if seed is -1
     if seed == -1:
@@ -64,7 +63,7 @@ def generate_novelai_image(
             "steps": steps,
             "n_samples": 1,
             "ucPreset": 0,
-            "add_original_image": overlay,
+            "add_original_image": True,
             "cfg_rescale": cfg_rescale,
             "controlnet_strength": 1,
             "dynamic_thresholding": dyn_threshold,
@@ -72,7 +71,7 @@ def generate_novelai_image(
             "legacy": False,
             "legacy_v3_extend": False,
             "negative_prompt": negative_prompt,
-            "noise": 0,
+            "noise": i2i_noise,
             "noise_schedule": schedule,
             "qualityToggle": True,
             "reference_information_extracted": info_extract,
@@ -85,6 +84,7 @@ def generate_novelai_image(
     }
     if ref_image is not None:
         payload['parameters']['reference_image'] = image2base64(ref_image)
+    '''
     if use_inp:
         payload['action'] = "infill"
         payload['model'] = 'nai-diffusion-3-inpainting'
@@ -92,6 +92,12 @@ def generate_novelai_image(
         payload['parameters']['image'] = image2base64(inp_img['background'])
         payload['parameters']['extra_noise_seed'] = seed
         payload['parameters']['strength'] = inp_str
+    '''
+    if i2i_image is not None:
+        payload['action'] = "img2img"
+        payload['parameters']['image'] = image2base64(i2i_image)
+        payload['parameters']['strength'] = i2i_str
+        payload['parameters']['extra_noise_seed'] = seed
     # Send the POST request
     response = requests.post(url, json=payload, headers=headers)