import setuptools

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

__version__ = "0.0.1"

REPO_NAME = "Chest-Xray-Pneumonia-ViT-MLflow-DVC"
AUTHOR_USER_NAME = "your-github-username" # CHANGE THIS
SRC_REPO = "vitClassifier"
AUTHOR_EMAIL = "your-email@example.com" # CHANGE THIS

setuptools.setup(
    name=SRC_REPO,
    version=__version__,
    author=AUTHOR_USER_NAME,
    author_email=AUTHOR_EMAIL,
    description="An end-to-end ML project for Chest X-ray Pneumonia classification using ViT.",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url=f"https://github.com/{AUTHOR_USER_NAME}/{REPO_NAME}",
    project_urls={
        "Bug Tracker": f"https://github.com/{AUTHOR_USER_NAME}/{REPO_NAME}/issues",
    },
    package_dir={"": "src"},
    packages=setuptools.find_packages(where="src")
)