biiig changes

modules/diffuseapi.py

@@ -18,7 +18,7 @@ class DiffuseAPI():
     def set_steps(self, steps):
         try:
             new_steps = int(steps)
-            if 0 > new_steps < 50:
+            if new_steps > 0 and new_steps <= 50:
                 self.num_steps = new_steps
                 return True
             return False
@@ -83,22 +83,26 @@ class DiffuseAPI():
     def get_nsfw_filter(self):
         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
-        }
+    async def generate_image(self, prompt, neg_prompt="", payload=None):
+        if payload is None:
+            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
+            }
+
+        payload['prompt'] = prompt
 
         settings = {
         "filter_nsfw": not self.nsfw_enabled,
-        "enable_pnginfo": False
+        "enable_pnginfo": False,
+#        "sd_hypernetwork": "anime_3"
         }
 
         override_payload = {
@@ -122,7 +126,7 @@ class DiffuseAPI():
             "gfpgan_visibility": 0,
             "codeformer_visibility": 0,
             "codeformer_weight": 0,
-            "upscaling_resize": 4,
+            "upscaling_resize": 2,
             "upscaling_resize_w": 512,
             "upscaling_resize_h": 1024,
             "upscaling_crop": True,
@@ -139,4 +143,4 @@ class DiffuseAPI():
                     return None
             async with session.post("/sdapi/v1/extra-single-image", json=payload) as image_json:
                 image_data = await image_json.json()
-                return image_data["image"]
+                return image_data["image"]