import re
import os
import codecs
from os import path
from io import open
from setuptools import setup, find_packages

def read_requirements(path):
    with open(path, "r") as f:
        requirements = f.read().splitlines()
    return requirements

requirements = read_requirements("requirements.txt")
here = path.abspath(path.dirname(__file__))

with open(path.join(here, "README.md"), encoding="utf-8") as f:
    long_description = f.read()

# Default version if __init__.py not found
version_string = "0.1.0"

# Try to load version from __init__.py if it exists
init_file_path = os.path.join(here, "template/__init__.py")
if os.path.exists(init_file_path):
    with codecs.open(init_file_path, encoding="utf-8") as init_file:
        version_match = re.search(
            r"^__version__ = ['\"]([^'\"]*)['\"]", init_file.read(), re.M
        )
        if version_match:
            version_string = version_match.group(1)

setup(
    name="moshi-hf",
    version=version_string,
    description="Moshi HuggingFace inference server",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/omegalabsinc/omegalabs-anytoany-bittensor",
    author="OMEGA Labs, Inc.",
    packages=find_packages(),
    include_package_data=True,
    author_email="salman@omega-labs.ai",
    license="MIT",
    python_requires=">=3.8",
    install_requires=requirements,
    classifiers=[
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Developers",
        "Topic :: Software Development :: Build Tools",
        "License :: OSI Approved :: MIT License",
        "Programming Language :: Python :: 3 :: Only",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Topic :: Scientific/Engineering",
        "Topic :: Scientific/Engineering :: Mathematics",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
        "Topic :: Software Development",
        "Topic :: Software Development :: Libraries",
        "Topic :: Software Development :: Libraries :: Python Modules",
    ],
)