#!/bin/bash
# 主入口脚本 - 串联所有训练步骤
# 适用于 Qwen3.5-0.8B-HOS 训练流程

set -e

echo "============================================"
echo "  HOS Security Model Training"
echo "  Qwen3.5-0.8B-HOS"
echo "============================================"

# 解析参数
NO_PROXY=false
for arg in "$@"; do
    case $arg in
        --no-proxy)
            NO_PROXY=true
            shift
            ;;
    esac
done

# Step 1: 代理配置（可选）
if [ "$NO_PROXY" = false ]; then
    echo "[1/5] 配置网络代理..."
    source scripts/setup_proxy.sh
else
    echo "[1/5] 跳过代理配置"
fi

# Step 2: 安装依赖
echo "[2/5] 安装环境依赖..."
bash scripts/prepare_env.sh

# Step 3: 下载并清洗数据
echo "[3/5] 准备数据集..."
bash scripts/prepare_data.sh

# Step 4: 训练
echo "[4/5] 启动训练..."
bash scripts/train.sh

# Step 5: 上传
echo "[5/5] 上传模型..."
bash scripts/upload.sh

echo "============================================"
echo "  训练完成!"
echo "  模型: https://huggingface.co/lxcxjxhx/Qwen3.5-0.8B-HOS"
echo "============================================"