from setuptools import setup, find_packages

setup(
    name='chatbug',
    version='0.1.0',
    description='A conversational AI chatbot',
    author='Florin Tobler',
    author_email='florin.tobler@hotmail.com',
    packages=find_packages(exclude=["tests"]),
    install_requires=[
        'transformers',
        'accelerate',
        'bitsandbytes',
        'pytest',
        'pywebview',
    ],
    entry_points={
        'console_scripts': [
            'chatbug=chatbug.llama:main_func',
            # a^     b^       c^   d^
            # a => the command line argument
            # b => the package name
            # c => the file name in the package (same as imports)
            # d => the function to call
            'chatbugui=chatbug.ui.__main__:start_ui',
        ],
    },
)