#!/usr/bin/env bash

VGEN_ROOT="${VGEN_ROOT:-/apdcephfs_sgxy/share_304169095/hunyuan/juliayxzhu/dev/VGen}"
DEV_ROOT="${DEV_ROOT:-/apdcephfs_sgxy/share_304169095/hunyuan/juliayxzhu/dev}"
CONDA_ROOT="${CONDA_ROOT:-$DEV_ROOT/miniconda3}"
CONDA_SH="${CONDA_SH:-/apdcephfs_sgxy/share_304169095/hunyuan/juliayxzhu/dev/miniconda3/etc/profile.d/conda.sh}"
DIFFSYNTH_ROOT="${DIFFSYNTH_ROOT:-$VGEN_ROOT/third_party/DiffSynth-Studio}"
MODELS_ROOT="${MODELS_ROOT:-$DEV_ROOT/Models}"

DIFF_ENV="${DIFF_ENV:-diff}"

QWEN_IMAGE_MODEL="${QWEN_IMAGE_MODEL:-$MODELS_ROOT/Qwen-Image}"
COSYVOICE_MODEL="${COSYVOICE_MODEL:-$MODELS_ROOT/CosyVoice3-0.5B-2512}"
WAN_I2V_MODEL="${WAN_I2V_MODEL:-$MODELS_ROOT/Wan2.2-I2V-A14B}"
BINDWEAVE_ROOT="${BINDWEAVE_ROOT:-$VGEN_ROOT/third_party/BindWeave}"
BINDWEAVE_SOURCE_MODEL="${BINDWEAVE_SOURCE_MODEL:-$MODELS_ROOT/BindWeave}"
BINDWEAVE_CHECKPOINT="${BINDWEAVE_CHECKPOINT:-$MODELS_ROOT/BindWeave-MM}"
BINDWEAVE_WAN21_MODEL="${BINDWEAVE_WAN21_MODEL:-$MODELS_ROOT/Wan2.1-I2V-14B-720P-Diffusers}"
BINDWEAVE_QWEN_MODEL="${BINDWEAVE_QWEN_MODEL:-$MODELS_ROOT/Qwen2.5-VL-7B-Instruct}"
BINDWEAVE_PYTHON="${BINDWEAVE_PYTHON:-}"
BINDWEAVE_DEVICE="${BINDWEAVE_DEVICE:-npu}"
BINDWEAVE_NPROC_PER_NODE="${BINDWEAVE_NPROC_PER_NODE:-8}"
BINDWEAVE_CONTEXT_PARALLEL_SIZE="${BINDWEAVE_CONTEXT_PARALLEL_SIZE:-0}"
BINDWEAVE_MASTER_PORT="${BINDWEAVE_MASTER_PORT:-6000}"
QWEN35_27B_MODEL="${QWEN35_27B_MODEL:-$MODELS_ROOT/qwen3.5-27b}"

LOG_DIR="${LOG_DIR:-$VGEN_ROOT/logs}"
RUNTIME_DIR="${RUNTIME_DIR:-$VGEN_ROOT/runtime}"

export VGEN_ROOT DEV_ROOT CONDA_ROOT CONDA_SH DIFFSYNTH_ROOT MODELS_ROOT
export DIFF_ENV
export QWEN_IMAGE_MODEL COSYVOICE_MODEL WAN_I2V_MODEL QWEN35_27B_MODEL
export BINDWEAVE_ROOT BINDWEAVE_SOURCE_MODEL BINDWEAVE_CHECKPOINT BINDWEAVE_WAN21_MODEL BINDWEAVE_QWEN_MODEL
export BINDWEAVE_PYTHON BINDWEAVE_DEVICE BINDWEAVE_NPROC_PER_NODE BINDWEAVE_CONTEXT_PARALLEL_SIZE BINDWEAVE_MASTER_PORT
export LOG_DIR RUNTIME_DIR