DiffuseAPI.py

@@ -1,4 +1,5 @@
 import aiohttp
+from base64 import b64decode
 from sys import exit
 
 class DiffuseAPI():
@@ -82,6 +83,18 @@ class DiffuseAPI():
         return self.nsfw_enabled
 
     async def generate_image(self, prompt, neg_prompt=""):
+        payload = {
+        "prompt": prompt,
+        "styles": self.styles,
+        "steps": self.num_steps,
+        "seed": self.seed,
+        "n_iter": 1,
+        "height": self.height,
+        "width": self.width,
+        "negative_prompts": neg_prompt,
+        "cfg_scale": self.cfg_scale
+        }
+
         settings = {
         "filter_nsfw": not self.nsfw_enabled
         }
@@ -90,9 +103,7 @@ class DiffuseAPI():
             "override_settings": settings
         }
 
-        payload = {"prompt": prompt, "styles": self.styles, "steps": self.num_steps, "seed": self.seed,
+        payload.update(override_payload)
-                    "n_iter": 1, "height": self.height, "width": self.width, "negative_prompts": neg_prompt,
-                    "cfg_scale": self.cfg_scale} | override_payload
 
         sess = aiohttp.ClientSession(self.url)
         alive = await sess.head('/')
@@ -107,4 +118,4 @@ class DiffuseAPI():
                     return None
             async with session.post("/sdapi/v1/txt2img", json=payload) as image_json:
                 image_data = await image_json.json()
-                return image_data["images"][0]
+                return image_data["images"][0]