#!/bin/bash

# HuggingFace模型上传脚本
# ================================

echo "🚀 HuggingFace模型上传脚本"
echo "=============================="

# 检查是否已登录
echo "📋 检查HuggingFace登录状态..."
if ! ~/.local/bin/hf whoami &>/dev/null; then
    echo "❌ 未登录HuggingFace，请先登录:"
    echo "   ~/.local/bin/hf login"
    echo "   然后输入你的Access Token"
    exit 1
fi

echo "✅ 已登录HuggingFace"

# 获取用户输入
read -p "请输入模型名称 (例如: 你的用户名/kaon-w): " REPO_NAME

if [ -z "$REPO_NAME" ]; then
    echo "❌ 模型名称不能为空"
    exit 1
fi

echo "📂 准备上传模型: $REPO_NAME"
echo "📁 文件列表:"
ls -la /tmp/kaon-w-upload/

echo ""
read -p "确认上传? (y/N): " CONFIRM

if [[ $CONFIRM != [yY] ]]; then
    echo "❌ 取消上传"
    exit 0
fi

echo "⬆️ 开始上传..."
cd /tmp/kaon-w-upload/

# 上传模型
~/.local/bin/hf upload $REPO_NAME . --repo-type model

if [ $? -eq 0 ]; then
    echo ""
    echo "🎉 上传成功！"
    echo "🔗 模型地址: https://huggingface.co/$REPO_NAME"
    echo ""
    echo "📋 使用方法:"
    echo "from peft import PeftModel"
    echo "model = PeftModel.from_pretrained(base_model, '$REPO_NAME')"
else
    echo "❌ 上传失败，请检查网络连接和权限"
fi