Add ZeroGPU support and Windows local startup

- Add @spaces.GPU decorators to all GPU-intensive functions
- Update requirements.txt with proper versions and dependencies
- Add ZeroGPU support to vynl_generator.py, ai_studio.py, vynl_rvc.py
- Create Windows startup batch file (start_vynl.bat)
- Add local_config.py for environment detection
- Update gradio to 4.44.0 for ZeroGPU compatibility
- Fix AudioCraft dependency version

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

README.md

@@ -4,11 +4,10 @@ emoji: '🎸'
 colorFrom: yellow
 colorTo: red
 sdk: gradio
-sdk_version: 4.19.2
+sdk_version: 4.44.0
 app_file: app.py
 pinned: true
 license: other
-hardware: zero-a10g
 ---
 
 # VYNL - Complete Music Production Suite

ai_studio.py

@@ -13,6 +13,19 @@ from datetime import datetime
 from typing import Optional, List, Tuple
 import subprocess
 
+# ZeroGPU support for HuggingFace Spaces
+try:
+    import spaces
+    HAS_ZEROGPU = True
+except ImportError:
+    HAS_ZEROGPU = False
+    class spaces:
+        @staticmethod
+        def GPU(duration=60):
+            def decorator(func):
+                return func
+            return decorator
+
 # Persistent storage for models
 MODELS_DIR = Path(os.environ.get('VYNL_MODELS_DIR', Path.home() / '.vynl_models'))
 MODELS_DIR.mkdir(parents=True, exist_ok=True)
@@ -271,11 +284,12 @@ def create_training_manifest(audio_files: List[str], descriptions: List[str],
     return str(manifest_path)
 
 
+@spaces.GPU(duration=600)
 def train_model(training_dir: Path, model_name: str, style_description: str,
                 base_model: str = "small", epochs: int = 10,
                 progress_callback=None) -> Tuple[Optional[str], str]:
     """
-    Fine-tune MusicGen on custom audio
+    Fine-tune MusicGen on custom audio (GPU accelerated)
 
     Note: Full fine-tuning requires significant GPU memory.
     This uses a simplified approach with style conditioning.
@@ -404,10 +418,11 @@ def analyze_training_style(audio_files: List[Path]) -> dict:
 # GENERATION
 # ============================================================================
 
+@spaces.GPU(duration=300)
 def generate_music(prompt: str, model_choice: str, duration: int = 30,
                    temperature: float = 1.0, progress_callback=None) -> Tuple[Optional[str], str]:
     """
-    Generate music using MusicGen with optional custom style
+    Generate music using MusicGen with optional custom style (GPU accelerated)
     """
     if not HAS_AUDIOCRAFT:
         return None, "AudioCraft not installed. Install with: pip install audiocraft"
@@ -498,9 +513,10 @@ def generate_music(prompt: str, model_choice: str, duration: int = 30,
         return None, f"Generation error: {str(e)}"
 
 
+@spaces.GPU(duration=300)
 def generate_with_melody(prompt: str, melody_audio: str, model_choice: str,
                         duration: int = 30, progress_callback=None) -> Tuple[Optional[str], str]:
-    """Generate music conditioned on a melody/reference audio"""
+    """Generate music conditioned on a melody/reference audio (GPU accelerated)"""
     if not HAS_AUDIOCRAFT or not HAS_LIBROSA:
         return None, "AudioCraft and librosa required"
 

app.py

@@ -21,9 +21,11 @@ warnings.filterwarnings('ignore')
 try:
     import spaces
     HAS_ZEROGPU = True
+    print("ZeroGPU available - running on HuggingFace Spaces")
 except ImportError:
     HAS_ZEROGPU = False
-    # Dummy decorator
+    print("ZeroGPU not available - running locally")
+    # Dummy decorator for local use
     class spaces:
         @staticmethod
         def GPU(duration=60):
@@ -31,6 +33,9 @@ except ImportError:
                 return func
             return decorator
 
+# Check if running on HuggingFace Spaces
+IS_HF_SPACE = os.environ.get('SPACE_ID') is not None
+
 # ============================================================================
 # MODULE IMPORTS
 # ============================================================================
@@ -274,8 +279,9 @@ def download_youtube(url: str) -> tuple:
     except Exception as e:
         return None, str(e)
 
+@spaces.GPU(duration=120)
 def separate_stems(audio_path: str, progress=None) -> dict:
-    """Separate audio into stems using Demucs"""
+    """Separate audio into stems using Demucs (GPU accelerated)"""
     try:
         import subprocess
         stems_dir = OUTPUT_DIR / f"stems_{datetime.now().strftime('%H%M%S')}"
@@ -284,6 +290,7 @@ def separate_stems(audio_path: str, progress=None) -> dict:
         if progress:
             progress(0.3, "Running Demucs stem separation...")
 
+        # Use GPU if available
         cmd = ['python', '-m', 'demucs', '-o', str(stems_dir), audio_path]
         subprocess.run(cmd, capture_output=True, check=True)
 
@@ -411,8 +418,9 @@ def process_song(audio, yt_url, name, lyrics, do_stems, do_chords, do_daw, user_
 # AI STUDIO FUNCTIONS
 # ============================================================================
 
+@spaces.GPU(duration=300)
 def generate_ai_music(prompt, genre, mood, key, bpm, time_sig, duration, instruments, temperature, user_email, progress=gr.Progress()):
-    """Generate AI music with full parameters"""
+    """Generate AI music with full parameters (GPU accelerated)"""
     if not HAS_AUDIOCRAFT:
         return None, "AudioCraft not installed.\n\nThis feature requires a GPU-enabled environment.\n\nFor local installation with GPU:\npip install audiocraft xformers\npip install torch torchaudio --index-url https://download.pytorch.org/whl/cu118"
 
@@ -454,8 +462,9 @@ def generate_ai_music(prompt, genre, mood, key, bpm, time_sig, duration, instrum
 
     return audio_path, status
 
+@spaces.GPU(duration=120)
 def apply_voice_clone(audio, voice_model, pitch_shift, user_email, progress=gr.Progress()):
-    """Apply voice cloning/conversion"""
+    """Apply voice cloning/conversion (GPU accelerated)"""
     if not audio:
         return None, "Please provide audio"
 

local_config.py

@@ -0,0 +1,95 @@
+#!/usr/bin/env python3
+"""
+VYNL Local Configuration
+Environment detection and configuration for local vs HuggingFace deployment
+"""
+
+import os
+import sys
+from pathlib import Path
+
+# ============================================================================
+# ENVIRONMENT DETECTION
+# ============================================================================
+
+def is_huggingface_space():
+    """Check if running on HuggingFace Spaces"""
+    return os.environ.get('SPACE_ID') is not None
+
+def is_gpu_available():
+    """Check if GPU is available"""
+    try:
+        import torch
+        return torch.cuda.is_available()
+    except ImportError:
+        return False
+
+def get_device():
+    """Get the compute device"""
+    if is_gpu_available():
+        import torch
+        return torch.device('cuda')
+    return 'cpu'
+
+# ============================================================================
+# PATHS
+# ============================================================================
+
+def get_output_dir():
+    """Get output directory"""
+    if is_huggingface_space():
+        import tempfile
+        return Path(tempfile.gettempdir()) / 'vynl_output'
+    else:
+        return Path(os.environ.get('VYNL_OUTPUT_DIR', Path.cwd() / 'output'))
+
+def get_catalog_dir():
+    """Get catalog directory"""
+    return Path(os.environ.get('VYNL_CATALOG_DIR', Path.home() / '.vynl_catalog'))
+
+def get_models_dir():
+    """Get models directory"""
+    return Path(os.environ.get('VYNL_MODELS_DIR', Path.home() / '.vynl_models'))
+
+# ============================================================================
+# CONFIGURATION
+# ============================================================================
+
+class Config:
+    """VYNL Configuration"""
+
+    IS_HF_SPACE = is_huggingface_space()
+    HAS_GPU = is_gpu_available()
+    DEVICE = get_device()
+
+    OUTPUT_DIR = get_output_dir()
+    CATALOG_DIR = get_catalog_dir()
+    MODELS_DIR = get_models_dir()
+
+    # Create directories
+    OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
+    CATALOG_DIR.mkdir(parents=True, exist_ok=True)
+    MODELS_DIR.mkdir(parents=True, exist_ok=True)
+
+    # Feature flags
+    ENABLE_AUDIOCRAFT = True
+    ENABLE_RVC = True
+    ENABLE_DEMUCS = True
+
+    # Generation limits
+    MAX_DURATION_SECONDS = 600  # 10 minutes
+    MAX_FILE_SIZE_MB = 100
+
+    @classmethod
+    def print_status(cls):
+        """Print configuration status"""
+        print(f"VYNL Configuration:")
+        print(f"  Environment: {'HuggingFace Space' if cls.IS_HF_SPACE else 'Local'}")
+        print(f"  GPU Available: {cls.HAS_GPU}")
+        print(f"  Device: {cls.DEVICE}")
+        print(f"  Output Dir: {cls.OUTPUT_DIR}")
+        print(f"  Catalog Dir: {cls.CATALOG_DIR}")
+        print(f"  Models Dir: {cls.MODELS_DIR}")
+
+if __name__ == "__main__":
+    Config.print_status()

requirements.txt

@@ -1,13 +1,15 @@
-gradio==4.19.2
-huggingface_hub==0.20.3
-numpy>=1.24.0
+gradio>=4.44.0
+huggingface_hub>=0.21.0
+spaces>=0.30.0
+numpy>=1.24.0,<2.0.0
 librosa>=0.10.0
 soundfile>=0.12.0
 scipy>=1.10.0
-yt-dlp>=2023.10.0
-torch>=2.0.0
-torchaudio>=2.0.0
+yt-dlp>=2024.1.0
+torch>=2.1.0
+torchaudio>=2.1.0
 demucs>=4.0.0
 pyloudnorm>=0.1.0
-audiocraft
-xformers
+audiocraft>=1.3.0
+xformers>=0.0.23
+accelerate>=0.25.0

start_vynl.bat

@@ -0,0 +1,112 @@
+@echo off
+REM ============================================================================
+REM VYNL - Complete Music Production Suite
+REM Windows Startup Script
+REM Copyright (c) 2024-2026 Robert T. Lackey. All rights reserved.
+REM ============================================================================
+
+title VYNL - Music Production Suite
+
+echo.
+echo ============================================================
+echo   VYNL - Complete Music Production Suite
+echo   Copyright (c) 2024-2026 Robert T. Lackey
+echo   Stone and Lantern Music Group
+echo ============================================================
+echo.
+
+REM Check if Python is installed
+python --version >nul 2>&1
+if errorlevel 1 (
+    echo ERROR: Python is not installed or not in PATH
+    echo Please install Python 3.10 or higher from https://python.org
+    pause
+    exit /b 1
+)
+
+REM Check Python version
+for /f "tokens=2" %%i in ('python --version 2^>^&1') do set PYVER=%%i
+echo Found Python %PYVER%
+
+REM Set up virtual environment
+if not exist "venv" (
+    echo Creating virtual environment...
+    python -m venv venv
+    if errorlevel 1 (
+        echo ERROR: Failed to create virtual environment
+        pause
+        exit /b 1
+    )
+)
+
+REM Activate virtual environment
+echo Activating virtual environment...
+call venv\Scripts\activate.bat
+
+REM Check if dependencies are installed
+if not exist "venv\.installed" (
+    echo Installing dependencies (this may take several minutes)...
+    echo.
+
+    REM Upgrade pip first
+    python -m pip install --upgrade pip
+
+    REM Install PyTorch with CUDA support
+    echo Installing PyTorch with CUDA support...
+    pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu118
+
+    REM Install other requirements
+    echo Installing remaining dependencies...
+    pip install gradio>=4.44.0
+    pip install numpy>=1.24.0
+    pip install librosa>=0.10.0
+    pip install soundfile>=0.12.0
+    pip install scipy>=1.10.0
+    pip install yt-dlp>=2024.1.0
+    pip install demucs>=4.0.0
+    pip install pyloudnorm>=0.1.0
+    pip install audiocraft>=1.3.0
+    pip install xformers>=0.0.23
+    pip install accelerate>=0.25.0
+
+    REM Create marker file
+    echo. > venv\.installed
+    echo Dependencies installed successfully!
+)
+
+REM Check for NVIDIA GPU
+nvidia-smi >nul 2>&1
+if errorlevel 1 (
+    echo.
+    echo WARNING: NVIDIA GPU not detected or drivers not installed
+    echo VYNL will run on CPU (slower performance)
+    echo For best results, install NVIDIA drivers from https://nvidia.com/drivers
+    echo.
+) else (
+    echo NVIDIA GPU detected - GPU acceleration enabled
+)
+
+REM Set environment variables
+set VYNL_LOCAL=1
+set VYNL_OUTPUT_DIR=%~dp0output
+set VYNL_CATALOG_DIR=%USERPROFILE%\.vynl_catalog
+set VYNL_MODELS_DIR=%USERPROFILE%\.vynl_models
+
+REM Create output directory
+if not exist "output" mkdir output
+
+echo.
+echo ============================================================
+echo   Starting VYNL...
+echo   Open http://localhost:7860 in your browser
+echo   Press Ctrl+C to stop the server
+echo ============================================================
+echo.
+
+REM Start the application
+python app.py
+
+REM Deactivate on exit
+call venv\Scripts\deactivate.bat
+
+pause

vynl_generator.py

@@ -15,6 +15,19 @@ from typing import Optional, Tuple, List
 import warnings
 warnings.filterwarnings('ignore')
 
+# ZeroGPU support for HuggingFace Spaces
+try:
+    import spaces
+    HAS_ZEROGPU = True
+except ImportError:
+    HAS_ZEROGPU = False
+    class spaces:
+        @staticmethod
+        def GPU(duration=60):
+            def decorator(func):
+                return func
+            return decorator
+
 # ============================================================================
 # AUDIOCRAFT IMPORTS
 # ============================================================================
@@ -361,6 +374,7 @@ def get_generator() -> MusicGenerator:
 # HIGH-LEVEL API
 # ============================================================================
 
+@spaces.GPU(duration=300)
 def generate_song(
     prompt: str = "",
     genre: str = "",
@@ -376,7 +390,7 @@ def generate_song(
     progress_callback=None
 ) -> Tuple[Optional[str], str]:
     """
-    High-level song generation API.
+    High-level song generation API (GPU accelerated).
 
     Args:
         prompt: Custom text prompt

vynl_rvc.py

@@ -17,6 +17,19 @@ from typing import Optional, Tuple, List, Dict
 from datetime import datetime
 import json
 
+# ZeroGPU support for HuggingFace Spaces
+try:
+    import spaces
+    HAS_ZEROGPU = True
+except ImportError:
+    HAS_ZEROGPU = False
+    class spaces:
+        @staticmethod
+        def GPU(duration=60):
+            def decorator(func):
+                return func
+            return decorator
+
 # ============================================================================
 # CONFIGURATION
 # ============================================================================
@@ -282,6 +295,7 @@ PRESET_VOICES = {
 # RVC INFERENCE
 # ============================================================================
 
+@spaces.GPU(duration=120)
 def convert_voice(
     input_audio: str,
     model_id: str = None,
@@ -296,7 +310,7 @@ def convert_voice(
     progress_callback=None
 ) -> Tuple[Optional[str], str]:
     """
-    Convert voice using RVC model.
+    Convert voice using RVC model (GPU accelerated).
 
     Args:
         input_audio: Path to input audio file
@@ -401,6 +415,7 @@ def convert_voice(
 # RVC TRAINING
 # ============================================================================
 
+@spaces.GPU(duration=600)
 def train_voice_model(
     name: str,
     training_files: List[str],
@@ -412,7 +427,7 @@ def train_voice_model(
     progress_callback=None
 ) -> Tuple[Optional[str], str]:
     """
-    Train a custom RVC voice model.
+    Train a custom RVC voice model (GPU accelerated).
 
     Args:
         name: Name for the voice model