import os

from setuptools import find_packages, setup

pwd = os.path.dirname(__file__)
version_file = 'graphgen/version.py'


def readme():
    with open(os.path.join(pwd, 'README.md'), encoding='utf-8') as f:
        content = f.read()
    return content


def get_version():
    with open(os.path.join(pwd, version_file), 'r') as f:
        exec(compile(f.read(), version_file, 'exec'))
    return locals()['__version__']


def read_requirements():
    lines = []
    with open('requirements.txt', 'r') as f:
        for line in f.readlines():
            if line.startswith('#'):
                continue
            if 'textract' in line:
                continue
            if len(line) > 0:
                lines.append(line)
    return lines


install_packages = read_requirements()

if __name__ == '__main__':
    setup(
        name='graphg',
        version=get_version(),
        url='https://github.com/open-sciencelab/GraphGen',
        description=  # noqa E251
        'GraphGen: Enhancing Supervised Fine-Tuning for LLMs with Knowledge-Driven Synthetic Data Generation',  # noqa E501
        long_description=readme(),
        long_description_content_type='text/markdown',
        author='open-sciencelab',
        author_email='open-sciencelab@pjlab.org.cn',
        packages=find_packages(exclude=["models"]),
        package_data={
            'GraphGen': ['configs/*']
        },
        include_package_data=True,
        install_requires=install_packages,
        classifiers=[
            'Programming Language :: Python :: 3.8',
            'Programming Language :: Python :: 3.9',
            'Programming Language :: Python :: 3.10',
            'Programming Language :: Python :: 3.11',
            'Programming Language :: Python :: 3.12',
            'Intended Audience :: Developers',
            'Intended Audience :: Education',
            'Intended Audience :: Science/Research',
        ],
        entry_points={'console_scripts': ['graphgen=graphgen.generate:main']},
    )