Create generate.py

generate.py

@@ -0,0 +1,65 @@
+import httpx
+
+async def generate(prompt, seed, model, nsfw, sampler, steps):
+    headers = {
+        'Accept': '*/*',
+        'Accept-Language': 'en-US,en;q=0.9,it;q=0.8',
+        'Connection': 'keep-alive',
+        'Content-Type': 'application/json',
+        'Origin': 'https://aqualxx.github.io',
+        'Referer': 'https://aqualxx.github.io/',
+        'Sec-Fetch-Dest': 'empty',
+        'Sec-Fetch-Mode': 'cors',
+        'Sec-Fetch-Site': 'cross-site',
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36',
+        'apikey': '0000000000',
+        'sec-ch-ua': '"Google Chrome";v="111", "Not(A:Brand";v="8", "Chromium";v="111"',
+        'sec-ch-ua-mobile': '?0',
+        'sec-ch-ua-platform': '"Windows"',
+    }
+
+    json_data = {
+        'prompt': prompt,
+        'params': {
+            'steps': steps,
+            'n': 1,
+            'sampler_name': sampler,
+            'width': 576,
+            'height': 576,
+            'cfg_scale': 24,
+            'seed_variation': 1000,
+            'seed': seed,
+            'karras': True,
+            'denoising_strength': 1,
+            'tiling': False,
+            'hires_fix': False,
+            'clip_skip': 1,
+            'post_processing': [],
+        },
+        'nsfw': False,
+        'censor_nsfw': False,
+        'trusted_workers': False,
+        'models': [
+            model,
+        ],
+        'shared': False,
+        'r2': True,
+        'jobId': '',
+        'index': 0,
+        'gathered': False,
+        'failed': False,
+    }
+
+    async with httpx.AsyncClient(timeout=300) as client:
+        response = await client.post('https://stablehorde.net/api/v2/generate/async', headers=headers, json=json_data)
+        image_id = response.json()["id"]
+        response = await client.get('https://stablehorde.net/api/v2/generate/check/'+image_id, headers=headers)
+        response = response.json()
+        while not response["finished"]:
+            response = await client.get('https://stablehorde.net/api/v2/generate/check/'+image_id, headers=headers)
+            response = response.json()
+        response = await client.get('https://stablehorde.net/api/v2/generate/status/'+image_id, headers=headers)
+        response = response.json()
+        image_url = response["generations"][0]["img"]
+        seed = response["generations"][0]["seed"]
+        return image_url, seed