setup.py

"""
setup.py

Package configuration for QueryTwin AI.
Enables `pip install -e .` for local development.
"""

from setuptools import setup, find_packages
from pathlib import Path

README = (Path(__file__).parent / "README.md").read_text(encoding="utf-8", errors="ignore")

setup(
    name="querytwin-ai",
    version="2.0.0",
    description="Intelligent Duplicate Question Detection powered by XGBoost, "
                "NLP Feature Engineering, and Transformer Embeddings.",
    long_description=README,
    long_description_content_type="text/markdown",
    author="Satvik",
    python_requires=">=3.9",
    packages=find_packages(exclude=["tests", "tests.*"]),
    install_requires=[
        "streamlit>=1.30",
        "numpy>=1.24",
        "pandas>=2.0",
        "scikit-learn>=1.3",
        "xgboost>=2.0",
        "joblib>=1.3",
        "nltk>=3.8",
        "beautifulsoup4>=4.12",
        "lxml>=4.9",
        "scipy>=1.11",
        "fuzzywuzzy>=0.18",
        "python-Levenshtein>=0.21",
        "distance>=0.1.3",
        "matplotlib>=3.7",
        "seaborn>=0.13",
        "plotly>=5.18",
        "shap>=0.43",
        "optuna>=3.4",
        "tqdm>=4.66",
    ],
    extras_require={
        "transformers": ["sentence-transformers>=2.2"],
        "dev": ["pytest>=7.4", "pytest-cov>=4.1"],
    },
    entry_points={
        "console_scripts": [
            "querytwin-train=src.train:main",
        ],
    },
    classifiers=[
        "Programming Language :: Python :: 3",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
        "Topic :: Text Processing :: Linguistic",
    ],
)