Add setup script

setup.py

@@ -0,0 +1,89 @@
+import os
+import subprocess
+import sys
+import shutil
+from pathlib import Path
+
+# setup.py will clone and install VibeVoice, copy voice files if they exist
+
+def setup_vibevoice():
+    repo_dir = "VibeVoice"
+    original_dir = os.getcwd()
+    
+    # clone repo if needed
+    if not os.path.exists(repo_dir):
+        print("Cloning the VibeVoice repository...")
+        try:
+            subprocess.run(
+                ["git", "clone", "https://github.com/microsoft/VibeVoice.git"],
+                check=True,
+                capture_output=True,
+                text=True
+            )
+            print("Repository cloned successfully.")
+        except subprocess.CalledProcessError as e:
+            print(f"Error cloning repository: {e.stderr}")
+            sys.exit(1)
+    else:
+        print("Repository already exists. Skipping clone.")
+    
+    # install the package
+    os.chdir(repo_dir)
+    print(f"Changed directory to: {os.getcwd()}")
+    
+    print("Installing the VibeVoice package...")
+    try:
+        subprocess.run(
+            [sys.executable, "-m", "pip", "install", "-e", "."],
+            check=True,
+            capture_output=True,
+            text=True
+        )
+        print("Package installed successfully.")
+    except subprocess.CalledProcessError as e:
+        print(f"Error installing package: {e.stderr}")
+        sys.exit(1)
+    
+    # add to python path
+    sys.path.insert(0, os.getcwd())
+    
+    # go back to original directory
+    os.chdir(original_dir)
+    print(f"Changed back to original directory: {os.getcwd()}")
+    
+    # copy voice files if they exist
+    if os.path.exists("public/voices"):
+        target_voices_dir = os.path.join(repo_dir, "demo", "voices")
+        
+        # clear existing voices and use only ours
+        if os.path.exists(target_voices_dir):
+            shutil.rmtree(target_voices_dir)
+        os.makedirs(target_voices_dir)
+        
+        for file in os.listdir("public/voices"):
+            if file.endswith(('.mp3', '.wav', '.flac', '.ogg', '.m4a', '.aac')):
+                src = os.path.join("public/voices", file)
+                dst = os.path.join(target_voices_dir, file)
+                shutil.copy2(src, dst)
+                print(f"Copied voice file: {file}")
+    
+    return repo_dir
+
+def setup_voice_presets():
+    # get voice files from vibevoice demo directory
+    voices_dir = Path("VibeVoice/demo/voices")
+    
+    voice_presets = {}
+    audio_extensions = ('.wav', '.mp3', '.flac', '.ogg', '.m4a', '.aac')
+    
+    if voices_dir.exists():
+        for audio_file in voices_dir.glob("*"):
+            if audio_file.suffix.lower() in audio_extensions:
+                name = audio_file.stem
+                voice_presets[name] = str(audio_file)
+    
+    # if no voices found, create directory
+    if not voice_presets and not voices_dir.exists():
+        voices_dir.mkdir(parents=True, exist_ok=True)
+    
+    return dict(sorted(voice_presets.items()))