#!/bin/bash
set -e

# ============================================================
# Passport 生成环境部署脚本 - 基于 m03 模板
# 包含 8 个可编辑字段 + 4 个固定字段 + MRZ 坐标常量
# ============================================================

echo "🚀 Deploying passport generation environment (m03 version)..."

# 1. 安装系统依赖
echo "📦 Installing system packages..."
apt-get update && apt-get install -y --no-install-recommends \
    python3 python3-pip fonts-dejavu-core wget \
    && rm -rf /var/lib/apt/lists/*

# 2. 安装 Python 依赖
echo "🐍 Installing Python packages..."
pip3 install --break-system-packages --no-cache-dir \
    pillow==10.4.0 \
    opencv-python-headless==4.10.0.84 \
    numpy==1.26.4 \
    flask

# 3. 创建目录结构
echo "📁 Creating directories..."
mkdir -p /data/passport/{fonts,models,input,output,scripts,data_json}
cp /usr/share/fonts/truetype/dejavu/DejaVuSans*.ttf /data/passport/fonts/

# 4. 下载/复制 LaMa 模型
echo "📥 Downloading LaMa model..."
mkdir -p /data/passport/models
cd /data/passport/models
if [ ! -f big-lama.pt ]; then
    wget -q "https://github.com/Sanster/models/releases/download/add_big_lama/big-lama.pt" -O big-lama.pt
fi

echo "✅ Setup complete!"
echo ""
echo "接下来脚本请复制到目标机器执行："
echo "  curl -fsSL http://ipan.mbvpn.cn/data/passport_deploy.sh | bash"
echo ""
echo "或查看完整 Python 脚本："
echo "  cat /data/passport/scripts/write_passport.py"