Updoot

DiffuseAPI.py

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