from setuptools import setup, find_packages

# It's a good practice to read long descriptions outside the setup function
with open('README.md', 'r', encoding='utf-8') as f:
    long_description = f.read()

setup(
    name='ScienceDiscovery',
    version='0.1.0',
    author='Markus J. Buehler',
    author_email='mbuehler@mit.edu',
    packages=find_packages(),
    install_requires=[
        'numpy',
        'networkx',
        'matplotlib',
        'pandas',
        'transformers>=4.39',
        'pyautogen>=0.2.28',
        'powerlaw',
        'markdown2',
        'pdfkit',
        'bitsandbytes',
        'peft',
        'accelerate',
        'torch',
        'torchvision',
        'torchaudio',
        'huggingface_hub',
        'langchain',
        'pyvis',
        'yachalk',
        'pytesseract',
        'llama-index',
        'tqdm',
        'ipython',
        'scikit-learn',
        'scipy',
        'seaborn',
        'uuid',
        'pdfminer.six',
        'community',
        'guidance',
        'python-louvain',
        'wkhtmltopdf',
        'weasyprint',
        'llama-index-embeddings-huggingface',
        'langchain-community',
    ],
    description='ScienceDiscovery: Use LLM-based multi-agent system to reason over graphs and generate novel research ideas.',
    long_description=long_description,
    long_description_content_type='text/markdown',
    url='https://github.com/lamm-mit/SciAgentsDiscovery',
    classifiers=[
        'License :: OSI Approved :: MIT License',
        'Programming Language :: Python :: 3.11'
    ],
    python_requires='>=3.10',
)