Initial implementation of AudioCraft Studio

Complete web interface for Meta's AudioCraft AI audio generation:

- Gradio UI with tabs for all 5 model families (MusicGen, AudioGen,
  MAGNeT, MusicGen Style, JASCO)
- REST API with FastAPI, OpenAPI docs, and API key auth
- VRAM management with ComfyUI coexistence support
- SQLite database for project/generation history
- Batch processing queue for async generation
- Docker deployment optimized for RunPod with RTX 4090

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

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
2025-11-25 19:34:27 +01:00
commit ffbf02b12c
67 changed files with 12032 additions and 0 deletions

55
scripts/start.sh Executable file
View File

@@ -0,0 +1,55 @@
#!/bin/bash
# Startup script for AudioCraft Studio
# Used in Docker container and RunPod
set -e
echo "=========================================="
echo " AudioCraft Studio"
echo "=========================================="
# Create directories if they don't exist
mkdir -p "${AUDIOCRAFT_OUTPUT_DIR:-/workspace/outputs}"
mkdir -p "${AUDIOCRAFT_DATA_DIR:-/workspace/data}"
mkdir -p "${AUDIOCRAFT_MODEL_CACHE:-/workspace/models}"
# Check GPU availability
echo "Checking GPU..."
if command -v nvidia-smi &> /dev/null; then
nvidia-smi --query-gpu=name,memory.total,memory.free --format=csv
else
echo "Warning: nvidia-smi not found"
fi
# Check Python and dependencies
echo "Python version:"
python --version
echo "PyTorch version:"
python -c "import torch; print(f'PyTorch: {torch.__version__}, CUDA: {torch.cuda.is_available()}')"
# Check AudioCraft installation
echo "AudioCraft version:"
python -c "import audiocraft; print(audiocraft.__version__)" 2>/dev/null || echo "AudioCraft installed from source"
# Generate API key if not exists
if [ ! -f "${AUDIOCRAFT_DATA_DIR:-/workspace/data}/.api_key" ]; then
echo "Generating API key..."
python -c "
from src.api.auth import get_key_manager
km = get_key_manager()
if not km.has_key():
key = km.generate_new_key()
print(f'Generated API key: {key}')
print('Store this key securely - it will not be shown again!')
"
fi
# Start the application
echo "Starting AudioCraft Studio..."
echo "Gradio UI: http://0.0.0.0:${AUDIOCRAFT_GRADIO_PORT:-7860}"
echo "REST API: http://0.0.0.0:${AUDIOCRAFT_API_PORT:-8000}"
echo "API Docs: http://0.0.0.0:${AUDIOCRAFT_API_PORT:-8000}/api/docs"
echo "=========================================="
exec python main.py "$@"