from setuptools import setup, Extension
from Cython.Build import cythonize
import numpy as np

extensions = [
    Extension('constants', ['constants.pyx']),
    Extension('annotation', ['annotation.pyx']),
    Extension('credentials', ['credentials.pyx']),
    Extension('file_data', ['file_data.pyx']),
    Extension('hardware_service', ['hardware_service.pyx'], extra_compile_args=["-g"], extra_link_args=["-g"]),
    Extension('security', ['security.pyx']),
    Extension('remote_command', ['remote_command.pyx']),
    Extension('remote_command_handler', ['remote_command_handler.pyx']),
    Extension('user', ['user.pyx']),
    Extension('api_client', ['api_client.pyx']),
    Extension('ai_config', ['ai_config.pyx']),
    Extension('inference', ['inference.pyx'], include_dirs=[np.get_include()]),
    Extension('main', ['main.pyx']),
]

setup(
    name="azaion.ai",
    ext_modules=cythonize(
        extensions,
        compiler_directives={
            "language_level": 3,
            "emit_code_comments" : False,
            "binding": True,
            'boundscheck': False,
            'wraparound': False
        }
    ),
    install_requires=[
        'ultralytics>=8.0.0',
        'pywin32; platform_system=="Windows"'
    ],
    zip_safe=False
)