#!/usr/bin/env python
import subprocess
import os
import pkg_resources

from setuptools import setup, find_packages

ProjectDir = os.path.dirname(__file__)
result = subprocess.run(["pip", "install", "basicsr"], capture_output=True, text=True)
result = subprocess.run(
    ["pip", "install", "--no-cache-dir", "-U", "openmim"],
    capture_output=True,
    text=True,
)
result = subprocess.run(["mim", "install", "mmengine"], capture_output=True, text=True)
result = subprocess.run(
    ["mim", "install", "mmcv>=2.0.1"], capture_output=True, text=True
)
result = subprocess.run(
    ["mim", "install", "mmdet>=3.1.0"], capture_output=True, text=True
)
result = subprocess.run(
    ["mim", "install", "mmpose>=1.1.0"], capture_output=True, text=True
)

with open(os.path.join(ProjectDir, "requirements.txt"), "r") as f:
    requirements = f.read().splitlines()
requirements = [x for x in requirements if x and not x.startswith("#")]
requirements = [x.split(" ")[0] if "index-url" in x else x for x in requirements]

setup(
    name="musev",  # used in pip install
    version="1.0.0",
    author="anchorxia, zkangchen",
    author_email="anchorxia@tencent.com, zkangchen@tencent.com",
    description="Package about human video creation",
    # long_description=long_description,
    # long_description_content_type="text/markdown",
    url="https://github.com/TMElyralab/MuseV",
    packages=find_packages("musev"),
    package_dir={"": "musev"},
    # include_package_data=True, # please edit MANIFEST.in
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
    ],
    install_requires=requirements,
    # dependency_links=["https://download.pytorch.org/whl/cu118],
)