DiffuseAPI.py

@@ -87,7 +87,7 @@ class DiffuseAPI():
         "prompt": prompt,
         "styles": self.styles,
         "steps": self.num_steps,
-        "seed": self.seed,
+        "seed": self.ai_seed,
         "n_iter": 1,
         "height": self.height,
         "width": self.width,
@@ -112,10 +112,17 @@ class DiffuseAPI():
             alive.close()
             await sess.close()
             return None
-        async with aiohttp.ClientSession(self.url) as session:
-            async with session.head('/') as alive:
-                if alive.status != 200:
-                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]
+        request  = await sess.post("/sdapi/v1/txt2img", json=payload)
+        try:
+            req_json = await request.json()
+            request.close()
+            await sess.close()
+            return req_json["images", [0]]
+        except:
+            return None
+
+
+    
+
+    
+