#!/bin/bash
# AI PPT Maker 模型分层回滚脚本
# 用法: sudo ./rollback-model-split.sh [时间戳, 默认 20260720163246]

set -e
TS="${1:-20260720163246}"
BACKUP="/root/ai-ppt-maker/backups/model-split-${TS}"

if [ ! -d "$BACKUP" ]; then
    echo "❌ 备份目录不存在: $BACKUP"
    echo "可用备份:"
    sudo ls /root/ai-ppt-maker/backups/
    exit 1
fi

echo "📦 从 $BACKUP 回滚..."
sudo cp ${BACKUP}/config.json /root/ai-ppt-maker/config.json && echo "✓ config.json"
sudo cp ${BACKUP}/openai_chat_provider.py /root/ai-ppt-maker/ppt_system/integrations/openai_chat_provider.py && echo "✓ openai_chat_provider.py"
sudo cp ${BACKUP}/direct_page_script.py /root/ai-ppt-maker/ppt_system/export/direct_page_script.py && echo "✓ direct_page_script.py"
sudo cp ${BACKUP}/job_agent_draft_model_planner.py /root/ai-ppt-maker/ppt_system/web/services/job_agent_draft_model_planner.py && echo "✓ job_agent_draft_model_planner.py"

# 重启 gunicorn worker
MASTER_PID=$(sudo ps aux | grep "gunicorn --workers 1" | grep -v grep | awk '{print $2}' | head -1)
if [ -n "$MASTER_PID" ]; then
    echo "🔄 重启 gunicorn master (PID $MASTER_PID)..."
    sudo kill -HUP $MASTER_PID
    sleep 3
    echo "✓ worker 已重启"
else
    echo "⚠️ 未找到 gunicorn master, 请手动重启"
fi

echo ""
echo "✅ 回滚完成"