#!/bin/bash
set -e

echo "================================================"
echo "   DreamOmni2 一键部署脚本 (本地下载 -> 服务器)"
echo "================================================"

# 0. Prepare Environment
echo "[1/4] 准备本地环境..."
if [ -d "DreamOmni2" ]; then
    echo "清理旧的 DreamOmni2 代码..."
    rm -rf DreamOmni2
fi

# 1. Clone Code (Try mirror first, then official)
echo "[2/4] 下载代码仓库..."
# 尝试使用 GitHub 镜像加速
if ! git clone https://mirror.ghproxy.com/https://github.com/dvlab-research/DreamOmni2.git; then
    echo "镜像下载失败，尝试官方源..."
    git clone https://github.com/dvlab-research/DreamOmni2.git
fi

# 2. Download Model
echo "[3/4] 下载模型权重 (Hugging Face)..."
if [ -f "./local_deploy_env/bin/python3" ]; then
    ./local_deploy_env/bin/python3 download_model.py
else
    python3 download_model.py
fi

# 3. Upload and Deploy
echo "[4/4] 上传并部署到服务器..."
chmod +x upload_and_deploy.exp
./upload_and_deploy.exp

echo "================================================"
echo "   部署流程结束！"
echo "================================================"