import requests
from piapi_ai import client

if __name__ == "__main__":
    # data = {
    #     "task_type": "txt2img",
    #     "input": {
    #         "prompt": "DC Extended Universe photo of a beautiful girl. Superheroes, grim, darker tones, highly detailed yet ordinary people, cheerful, brighter tones, undetailed, mask, masked.",
    #         "width": 1024,
    #         "height": 1024
    #     }
    # }
    # flux = client.flux.create_task(data=data)
    flux = client.flux.get_task("d7bfaec8-5cae-41fb-b324-1fd86d8a09dd")
    print(flux)