from setuptools import setup, find_packages

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

setup(
    name="imgshape",
    version="2.1.3",
    description="Smart image shape, dataset analysis, preprocessing & augmentation recommendations for ML workflows",
    long_description=long_description,
    long_description_content_type="text/markdown",
    author="Stifler",
    author_email="stiflerxd.ai@cudabit.live",
    packages=find_packages(where="src"),
    package_dir={"": "src"},
    install_requires=[
        "Pillow>=9.0.0",
        "numpy>=1.21.0",
        "matplotlib>=3.4.0",
        "scikit-image>=0.19.0",
        "gradio>=3.0.0",
        # seaborn kept optional in pyproject, but safe to leave if you always want it
        "seaborn>=0.11.0",
    ],
    extras_require={
        "torch": ["torch>=1.12.0", "torchvision>=0.13.0"],
        "pdf": ["weasyprint>=53.0"],
        "dev": ["pytest>=7.0", "black>=23.0", "flake8>=3.9", "pre-commit>=2.20"],
    },
    entry_points={
        "console_scripts": [
            "imgshape=imgshape.cli:main",
        ]
    },
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
        "Topic :: Scientific/Engineering :: Image Recognition",
    ],
    python_requires=">=3.8",
    keywords="images dataset visualization augmentation pytorch ml",
    url="https://github.com/STiFLeR7/imgshape",
    project_urls={
        "Source": "https://github.com/STiFLeR7/imgshape",
        "Issues": "https://github.com/STiFLeR7/imgshape/issues",
    },
)