from setuptools import find_packages, setup

with open("README.md", "r", encoding="utf-8") as f:
    readme = f.read()

setup(
    name="imaginAIry",
    author="Bryce Drennan",
    # author_email="b r y p y d o t io",
    version="1.5.1",
    description="AI imagined images. Pythonic generation of stable diffusion images.",
    long_description=readme,
    long_description_content_type="text/markdown",
    project_urls={
        "Documentation": "https://github.com/brycedrennan/imaginAIry/blob/master/README.md",
        "Source": "https://github.com/brycedrennan/imaginAIry",
    },
    packages=find_packages(include=("imaginairy", "imaginairy.*")),
    entry_points={
        "console_scripts": [
            "imagine=imaginairy.cmds:imagine_cmd",
            "aimg=imaginairy.cmds:aimg",
        ],
    },
    package_data={
        "imaginairy": [
            "configs/*.yaml",
            "vendored/clip/*.txt.gz",
            "vendored/clipseg/*.pth",
        ]
    },
    install_requires=[
        "click",
        "protobuf != 3.20.2, != 3.19.5",
        "fairscale>=0.4.4",  # for vendored blip
        "ftfy",  # for vendored clip
        "torch>=1.2.0",
        "numpy",
        "tqdm",
        "diffusers",
        "imageio==2.9.0",
        "pytorch-lightning==1.4.2",
        "omegaconf==2.1.1",
        "einops==0.3.0",
        "timm>=0.4.12",  # for vendored blip
        "torchdiffeq",
        "transformers==4.19.2",
        "torchmetrics==0.6.0",
        "torchvision>=0.13.1",
        "kornia==0.6",
        "realesrgan",
        "gfpgan>=1.3.7",
    ],
)