#!/bin/bash
set -x
B=/marimo/ai_studio
uv venv $B/envs/music 2>&1 | tail -2
CP=$B/envs/music/bin/python
uv pip install -p $CP -r $B/vendor/YuE/requirements.txt 2>&1 | tail -12
uv pip install -p $CP -e $B/vendor/YuE --no-deps 2>&1 | tail -3
PYTHONPATH= $CP -c "import torch;print('MUSIC torch',torch.__version__,'cuda',torch.cuda.is_available())" 2>&1 | tail -3
echo MUSIC_ENV_DONE