Add macOS and Windows support with usage of prebuilt-lib

include/ten_vad.py

@@ -7,31 +7,68 @@
 from ctypes import c_int, c_int32, c_float, c_size_t, CDLL, c_void_p, POINTER
 import numpy as np
 import os
+import platform
 
 class TenVad:
     def __init__(self, hop_size: int = 256, threshold: float = 0.5):
         self.hop_size = hop_size
         self.threshold = threshold
-        if os.path.exists(
-            os.path.join(
+        if platform.system() == "Linux" and platform.machine() == "x86_64":
+            git_path = os.path.join(
                 os.path.dirname(os.path.relpath(__file__)),
-                "../lib/Linux/x64/libten_vad.so",
+                "../lib/Linux/x64/libten_vad.so"
             )
-        ):
-            self.vad_library = CDLL(
-                os.path.join(
-                    os.path.dirname(os.path.relpath(__file__)), 
-                    "../lib/Linux/x64/libten_vad.so",
+            if os.path.exists(git_path):
+                self.vad_library = CDLL(git_path)
+            else:
+                pip_path = os.path.join(
+                    os.path.dirname(os.path.relpath(__file__)),
+                    "./ten_vad_library/libten_vad.so"
                 )
+                self.vad_library = CDLL(pip_path)
+                
+        elif platform.system() == "Darwin":
+            git_path = os.path.join(
+                os.path.dirname(os.path.relpath(__file__)),
+                "../lib/macOS/ten_vad.framework/Versions/A/ten_vad"
             )
-        else:
-            self.vad_library = CDLL(
-                os.path.join(
-                    os.path.dirname(
-                        os.path.relpath(__file__)),
-                        "./ten_vad_library/libten_vad.so",
+            if os.path.exists(git_path):
+                self.vad_library = CDLL(git_path)
+            else:
+                pip_path = os.path.join(
+                    os.path.dirname(os.path.relpath(__file__)),
+                    "./ten_vad_library/libten_vad"
+                )
+                self.vad_library = CDLL(pip_path)
+        elif platform.system().upper() == 'WINDOWS':
+            if platform.machine().upper() in ['X64', 'X86_64', 'AMD64']:
+                git_path = os.path.join(
+                    os.path.dirname(os.path.realpath(__file__)),
+                    "../lib/Windows/x64/ten_vad.dll"
+                )
+                if os.path.exists(git_path):
+                    self.vad_library = CDLL(git_path)
+                else:
+                    pip_path = os.path.join(
+                        os.path.dirname(os.path.realpath(__file__)),
+                        "./ten_vad_library/ten_vad.dll"
                     )
+                    self.vad_library = CDLL(pip_path)
+            else:
+                git_path = os.path.join(
+                    os.path.dirname(os.path.realpath(__file__)),
+                    "../lib/Windows/x86/ten_vad.dll"
                 )
+                if os.path.exists(git_path):
+                    self.vad_library = CDLL(git_path)
+                else:
+                    pip_path = os.path.join(
+                        os.path.dirname(os.path.realpath(__file__)),
+                        "./ten_vad_library/ten_vad.dll"
+                    )
+                    self.vad_library = CDLL(pip_path)
+        else:
+            raise NotImplementedError(f"Unsupported platform: {platform.system()} {platform.machine()}")
         self.vad_handler = c_void_p(0)
         self.out_probability = c_float()
         self.out_flags = c_int32()

setup.py

@@ -5,7 +5,7 @@
 #  Refer to the "LICENSE" file in the root directory for more information.
 #
 from setuptools import setup
-import os, shutil
+import os, shutil, platform
 from setuptools.command.install import install
 
 class custom_install_command(install):
@@ -13,8 +13,25 @@ class custom_install_command(install):
         install.run(self)
         target_dir = os.path.join(self.install_lib, "ten_vad_library")
         os.makedirs(target_dir, exist_ok=True)
-        shutil.copy("lib/Linux/x64/libten_vad.so", target_dir)
-        print(f"Files installed to: {target_dir}")
+        
+        if platform.system() == "Linux" and platform.machine() == "x86_64":
+            shutil.copy("lib/Linux/x64/libten_vad.so", target_dir)
+            print(f"Linux x64 library installed to: {target_dir}")
+        elif platform.system() == "Darwin":
+            shutil.copy("lib/macOS/ten_vad.framework/Versions/A/ten_vad", 
+                       os.path.join(target_dir, "libten_vad"))
+            print(f"macOS library installed to: {target_dir}")
+        elif platform.system().upper() == 'WINDOWS':
+            if platform.machine().upper() in ['X64', 'X86_64', 'AMD64']:
+                shutil.copy("lib/Windows/x64/ten_vad.dll", 
+                       os.path.join(target_dir, "ten_vad.dll"))
+                print(f"Windows x64 library installed to: {target_dir}")
+            else:
+                shutil.copy("lib/Windows/x86/ten_vad.dll", 
+                       os.path.join(target_dir, "ten_vad.dll"))
+                print(f"Windows x86 library installed to: {target_dir}")
+        else:
+            raise NotImplementedError(f"Unsupported platform: {platform.system()} {platform.machine()}")
 
 root_dir = os.path.dirname(os.path.abspath(__file__))
 shutil.copy(f"{root_dir}/include/ten_vad.py", f"{root_dir}/ten_vad.py")