Upload app.py with huggingface_hub

app.py

@@ -7,6 +7,7 @@ import os
 import tempfile
 import time
 import threading
+import subprocess
 import numpy as np
 import gradio as gr
 import librosa
@@ -15,41 +16,41 @@ import torch
 from pathlib import Path
 from typing import Iterator, Dict, Any
 
-# Install VibeVoice from GitHub:
-# git clone https://github.com/microsoft/VibeVoice.git
-# cd VibeVoice/
-# pip install -e .
+# Clone and setup VibeVoice if not already present
+import subprocess
+
+vibevoice_dir = Path('./VibeVoice')
+if not vibevoice_dir.exists():
+    print("Cloning VibeVoice repository...")
+    subprocess.run(['git', 'clone', 'https://github.com/microsoft/VibeVoice.git'], check=True)
+    print("Installing VibeVoice...")
+    subprocess.run(['pip', 'install', '-e', './VibeVoice'], check=True)
+
+# Add the VibeVoice directory to path
+import sys
+sys.path.insert(0, str(vibevoice_dir))
+
+# Change to VibeVoice directory for imports
+original_cwd = os.getcwd()
+os.chdir(vibevoice_dir)
 
 try:
     from vibevoice.modular.configuration_vibevoice import VibeVoiceConfig
     from vibevoice.modular.modeling_vibevoice_inference import VibeVoiceForConditionalGenerationInference
     from vibevoice.processor.vibevoice_processor import VibeVoiceProcessor
     from vibevoice.modular.streamer import AudioStreamer
-except ImportError:
-    try:
-        # Alternative import paths that might be used
-        from VibeVoice.vibevoice.modular.configuration_vibevoice import VibeVoiceConfig
-        from VibeVoice.vibevoice.modular.modeling_vibevoice_inference import VibeVoiceForConditionalGenerationInference
-        from VibeVoice.vibevoice.processor.vibevoice_processor import VibeVoiceProcessor
-        from VibeVoice.vibevoice.modular.streamer import AudioStreamer
-    except ImportError:
-        # Try adding the parent directory to path
-        import sys
-        parent_dir = Path(__file__).parent.parent
-        vibevoice_path = parent_dir / "VibeVoice"
-        if vibevoice_path.exists():
-            sys.path.insert(0, str(vibevoice_path))
-            from vibevoice.modular.configuration_vibevoice import VibeVoiceConfig
-            from vibevoice.modular.modeling_vibevoice_inference import VibeVoiceForConditionalGenerationInference
-            from vibevoice.processor.vibevoice_processor import VibeVoiceProcessor
-            from vibevoice.modular.streamer import AudioStreamer
-        else:
-            raise ImportError(
-                "VibeVoice module not found. Please install it:\n"
-                "git clone https://github.com/microsoft/VibeVoice.git\n"
-                "cd VibeVoice/\n"
-                "pip install -e ."
-            )
+except ImportError as e:
+    os.chdir(original_cwd)
+    raise ImportError(
+        f"VibeVoice module not found. Error: {e}\n"
+        "Please ensure VibeVoice is properly installed:\n"
+        "git clone https://github.com/microsoft/VibeVoice.git\n"
+        "cd VibeVoice/\n"
+        "pip install -e .\n"
+    )
+finally:
+    # Restore original directory after imports
+    os.chdir(original_cwd)
 from transformers.utils import logging
 from transformers import set_seed