# -*- coding: utf-8 -*-
# Copyright (c) XiMing Xing. All rights reserved.
# Author: XiMing Xing

from setuptools import setup, find_packages

setup(
    name='PyTorch-SVGRender',
    packages=find_packages(exclude=["test*", "docs", "examples"]),
    version='1.0.0',
    license='Mozilla Public License Version 2.0',
    description='SVG Differentiable Rendering: Generating vector graphics using neural networks.',
    author='XiMing Xing, Juncheng Hu et al.',
    author_email='ximingxing@gmail.com',
    url='https://github.com/ximinng/PyTorch-SVGRender',
    long_description_content_type='text/markdown',
    keywords=[
        'artificial intelligence',
        'AIGC',
        'generative models',
        'SVG generation',
    ],
    install_requires=[
        'hydra-core==1.3.2',  # configuration processor
        'omegaconf==2.3.0',  # YAML processor
        'accelerate==0.20.3',  # Hugging Face - pytorch distributed configuration
        'diffusers==0.20.2',  # Hugging Face - diffusion models
        'transformers==4.30.2',  # Hugging Face - transformers
        'datasets==2.13.1',
        'safetensors==0.3.1',
        'xformers',
        'einops==0.6.1',
        'pillow',
        'imageio-ffmpeg==0.4.8'
        'torch>=1.13.1',
        'torchvision>=0.14.1',
        'tensorboard==2.14.0',
        'triton==2.0.0.post1',
        'numba==0.57.1',
        'tqdm',  # progress bar
        'ftfy==6.1.1',
        'regex==2023.6.3',
        'timm==0.6.13',  # computer vision models
        "numpy==1.24.4",
        'scikit-learn==1.3.2',
        'scikit-fmm==2023.4.2',
        'scipy==1.10.1',
        'scikit-image==0.20.0',
        'Pillow',  # keep the PIL.Image.Resampling deprecation away,
        'pytorch-lightning==2.1.0',
        'matplotlib==3.7.1',
        'visdom=0.2.4',
        'wandb==0.15.8',  # weights & Biases
        'opencv-python==4.8.0.74',  # cv2
        'BeautifulSoup4==4.12.2',
        'freetype-py',  # font
        'shapely==2.0.1',  # SVG
        'svgwrite==1.4.3',
        'svgutils==0.3.4',
        'svgpathtools==1.6.1',
        'fairscale=0.4.13'  # ImageReward
    ],
    dependency_links=[
        "clip @ git+https://github.com/openai/CLIP.git",
    ],
    classifiers=[
        'Development Status :: 4 - Beta',
        'Intended Audience :: Developers',
        'Topic :: Scientific/Engineering :: Artificial Intelligence',
        'License :: OSI Approved :: MIT License',
        'Programming Language :: Python :: 3.10',
    ],
)