#!/usr/bin/env bash
# 下载接口1（四庭七眼测量）所需的模型权重与中文字体。
# 内网/离线环境无需执行：文件已随仓库带入（见 OFFLINE_ASSETS.md）。
# 仅供联网环境（生产机重建）使用。
set -euo pipefail

ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
WEIGHTS="$ROOT/face_analysis/weights"
FONTS="$ROOT/face_analysis/fonts"
mkdir -p "$WEIGHTS" "$FONTS"

# BiSeNet 主权重（~53MB）
if [ ! -f "$WEIGHTS/79999_iter.pth" ]; then
  echo ">> 下载 BiSeNet 主权重 79999_iter.pth"
  curl -L -o "$WEIGHTS/79999_iter.pth" \
    "https://huggingface.co/ManyOtherFunctions/face-parse-bisent/resolve/main/79999_iter.pth"
fi

# BiSeNet 骨干 resnet18（~45MB）
if [ ! -f "$WEIGHTS/resnet18-5c106cde.pth" ]; then
  echo ">> 下载 resnet18 骨干 resnet18-5c106cde.pth"
  curl -L -o "$WEIGHTS/resnet18-5c106cde.pth" \
    "https://download.pytorch.org/models/resnet18-5c106cde.pth"
fi

# 中文字体（思源黑体，~16MB）
if [ ! -f "$FONTS/NotoSansCJKsc-Regular.otf" ]; then
  echo ">> 下载中文字体 NotoSansCJKsc-Regular.otf"
  curl -L -o "$FONTS/NotoSansCJKsc-Regular.otf" \
    "https://github.com/notofonts/noto-cjk/raw/main/Sans/OTF/SimplifiedChinese/NotoSansCJKsc-Regular.otf"
fi

# resnet18 骨干放进 torch 缓存，避免 BiSeNet 初始化联网下载
CACHE="$HOME/.cache/torch/hub/checkpoints"
mkdir -p "$CACHE"
cp -n "$WEIGHTS/resnet18-5c106cde.pth" "$CACHE/" || true

echo ">> 完成。校验 sha256（见 OFFLINE_ASSETS.md）："
sha256sum "$WEIGHTS/79999_iter.pth" "$WEIGHTS/resnet18-5c106cde.pth" "$FONTS/NotoSansCJKsc-Regular.otf"