body = {"preset_name": "preset-1", "seed": None }
headers = {"Token": "<YOUR_API_KEY>"}

# ------------------------------------------------------------------------------------

preset_name = body.get("preset_name")

import time
import requests

if not preset_name:
    print("Please, enter a 'preset_name' variable to start the generation process ;)")
    exit()

url = "https://stdiff.io/api/preset2image"
session = requests.Session()
response = session.post(url, json=body, headers=headers).json()
# {
#     "success": true,
#     "task_id": "7133745436839378945",
#     "preset_name": "preset-1",
#     "seed": null
# }
task_id = response.get("task_id")
if task_id:
    print(f"Request to generate AI image based on the preset `{preset_name}` "
          f"was created successfully. Task ID: `{task_id}`")
    new_response = {}
else:
    new_response = response

status = None
while not status == "GENERATED" and task_id:
    print("Getting latest image generating status from the server...")
    new_response = session.get(url, params={"task_id": task_id}, headers=headers).json()
    status = new_response.get("status")
    print(f"...status: {status}")
    if status == "ERROR" or not status:
        break
    time.sleep(5)

if status == "GENERATED":
    print(f"Yohoo! Image was successfully generated. You can download it from here:"
          f" {new_response.get('image_url')}")
else:
    print(f"Error occurred when trying to generate image. Response: {new_response}")

# {
#     "success": true,
#     "task_id": "7133745436839378945",
#     "status": "GENERATED",
#     "image_url": "https://stdiff.io/media/generated_images/preset-3_1163882438_9c716427b787f3202e708a572ef38f7d.png"
# }