import os
import sys
from huggingface_hub import HfApi, create_repo, upload_folder
from config import Config

def main():
    print("🚀 开始上传所有 Checkpoint 到 robot4/emotion ...")
    
    api = HfApi()
    try:
        user_info = api.whoami()
        username = user_info['name']
        print(f"✅ User: {username}")
    except:
        print("❌ Please login first.")
        return

    # 1. 目标仓库
    repo_id = f"{username}/emotion"
    print(f"📦 目标仓库: {repo_id}")
    create_repo(repo_id=repo_id, repo_type="model", exist_ok=True)
    
    # 2. 上传整个 results 目录
    # 我们会上传 results/checkpoint-500, results/checkpoint-1000, etc.
    # 也就是在仓库根目录下会有这些文件夹
    results_dir = Config.RESULTS_DIR
    
    print(f"⬆️ 正在上传 {results_dir} 下的所有模型文件...")
    print("   (已自动忽略 optimizer.pt 等大文件以节省时间和流量)")
    
    upload_folder(
        folder_path=results_dir,
        repo_id=repo_id,
        repo_type="model",
        # 排除非必要大文件
        ignore_patterns=["optimizer.pt", "scheduler.pt", "rng_state.pth", "*.zip"] 
    )
    
    print(f"🎉 所有模型上传完成！查看地址: https://huggingface.co/{repo_id}")

if __name__ == "__main__":
    current_dir = os.path.dirname(os.path.abspath(__file__))
    parent_dir = os.path.dirname(current_dir)
    sys.path.append(parent_dir)
    main()