Refactor

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,20 @@ class DiffuseAPI():
             alive.close()
             await sess.close()
             return None
-        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
-
-
-    
+        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]
             
             
             
+            # try:
+            #     req_json = await request.json()
+            #     request.close()
+            #     await sess.close()
+            #     return req_json["images"][0]
+            # except:
+            #     return None