tests.py

import asyncio

from AsyncKandinsky import FusionBrainApi, ApiApi, ApiWeb

model = FusionBrainApi(ApiApi("Сюда свой api_key", "Сюда свой secret_key"))
# Любой способ на выбор
model = FusionBrainApi(ApiWeb("Ваша почта", "Ваш пароль"))


async def generate():
    try:
        result = await model.text2image("котик", style="ANIME", art_gpt=True)
    except ValueError as e:
        print(f"Error:\t{e}")
    else:
        with open("cat_anime_img.png", "wb") as f:
            f.write(result.getvalue())
        print("Done!")

    # _______________________________________________________________________________________
    try:
        result = await model.text2video("котик бежит по полю")
    except ValueError as e:
        print(f"Error:\t{e}")
    else:
        with open("cat_anime_video.mp4", "wb") as f:
            f.write(result.getvalue())
        print("Done!")

    # _______________________________________________________________________________________
    try:
        result = await model.text2animation(["котик бежит по полю", "котик пьёт воду из речки"])
    except ValueError as e:
        print(f"Error:\t{e}")
    else:
        with open("cat_anime_animation.mp4", "wb") as f:
            f.write(result.getvalue())
        print("Done!")


async def read_styles():
    for style in await model.get_styles():
        print(style)


if __name__ == '__main__':
    asyncio.run(read_styles())
    asyncio.run(generate())