import requests
from piapi_ai import client as piapi_client

if __name__ == "__main__":
    # endpoint = 'https://api.piapi.ai/api/v1/task'
    # headers = {
    #     'X-API-Key': '80d3e864cb1c0a74a728d85f627d8aacf4fc334a43c489392bbd4e43f378e583',
    # }
    #
    # response = requests.get("https://api.piapi.ai/api/v1/task/{task_id}".format(task_id="df422ee1-2ac8-4f70-9687-74e44d6518f5"), headers=headers)
    #
    # print(response.status_code)
    # print(response.json())
    task = piapi_client.flux.get_task("62257693-3b97-4734-b898-27783abca405")
    print(task)