%%writefile /content/setup.sh
set -e

cd /content

git clone https://github.com/RVC-Boss/GPT-SoVITS.git

cd GPT-SoVITS

mkdir -p GPT_weights

mkdir -p SoVITS_weights

if conda env list | awk '{print $1}' | grep -Fxq "GPTSoVITS"; then
    :
else
    conda create -n GPTSoVITS python=3.10 -y
fi

source activate GPTSoVITS

pip install ipykernel

bash install.sh --device CU126 --source HF

%pip install -q condacolab
import condacolab
condacolab.install_from_url("https://repo.anaconda.com/archive/Anaconda3-2024.10-1-Linux-x86_64.sh")
!cd /content && bash setup.sh

# Modify These
USER_ID = "AkitoP"
REPO_NAME = "GPT-SoVITS-v2-aegi"
BRANCH = "main"
GPT_PATH = "new_aegigoe-e100.ckpt"
SOVITS_PATH = "new_aegigoe_e60_s32220.pth"

# Do Not Modify
HF_BASE = "https://huggingface.co"
REPO_ID = f"{USER_ID}/{REPO_NAME}"
GPT_URL = f"{HF_BASE}/{REPO_ID}/blob/{BRANCH}/{GPT_PATH}"
SOVITS_URL = f"{HF_BASE}/{REPO_ID}/blob/{BRANCH}/{SOVITS_PATH}"

!cd "/content/GPT-SoVITS/GPT_weights" && wget "{GPT_URL}"
!cd "/content/GPT-SoVITS/SoVITS_weights" && wget "{SOVITS_URL}"

# Modify These
USER_ID = "aihobbyist"
REPO_NAME = "GPT-SoVits-V2-models"
BRANCH = "master"
GPT_PATH = "Genshin_Impact/EN/GPT_GenshinImpact_EN_5.1.ckpt"
SOVITS_PATH = "Wuthering_Waves/CN/SV_WutheringWaves_CN_1.3.pth"

# Do Not Modify
HF_BASE = "https://www.modelscope.cn/models"
REPO_ID = f"{USER_ID}/{REPO_NAME}"
GPT_URL = f"{HF_BASE}/{REPO_ID}/resolve/{BRANCH}/{GPT_PATH}"
SOVITS_URL = f"{HF_BASE}/{REPO_ID}/resolve/{BRANCH}/{SOVITS_PATH}"

!cd "/content/GPT-SoVITS/GPT_weights" && wget "{GPT_URL}"
!cd "/content/GPT-SoVITS/SoVITS_weights" && wget "{SOVITS_URL}"

!cd /content/GPT-SoVITS && source activate GPTSoVITS && export is_share=True && python webui.py