#!/bin/bash
set -x
B=/marimo/ai_studio
cd $B/vendor/CosyVoice || exit 1
git submodule update --init --recursive --depth 1 third_party/Matcha-TTS 2>&1 | tail -3
python3 -m venv --system-site-packages $B/envs/cosy 2>/dev/null || uv venv $B/envs/cosy
CP=$B/envs/cosy/bin/python
$CP -m ensurepip -q 2>/dev/null
uv pip install -p $CP "setuptools<81" wheel
uv pip install -p $CP -r requirements.txt
uv pip install -p $CP -e . --no-deps
uv pip install -p $CP -e third_party/Matcha-TTS --no-deps
echo COSY_SETUP_DONE