#!/usr/bin/env python3
"""
Upload CardVault+ model to HuggingFace Hub
"""

import os
import sys
from huggingface_hub import HfApi, login

def upload_model():
    # Check for token
    token = os.getenv("HF_TOKEN")
    if not token:
        print("Please set HF_TOKEN environment variable")
        print("Get your token from: https://huggingface.co/settings/tokens")
        return False
    
    try:
        # Login
        login(token=token)
        print("✅ Successfully logged in to HuggingFace")
        
        # Initialize API
        api = HfApi(token=token)
        
        # Check if repo exists, create if not
        try:
            api.repo_info(repo_id="sugiv/cardvaultplus", repo_type="model")
            print("✅ Repository exists")
        except:
            print("🔄 Creating new repository...")
            api.create_repo(repo_id="sugiv/cardvaultplus", repo_type="model", private=False)
            print("✅ Repository created")
        
        # Upload folder
        print("🚀 Uploading model files (4.2GB)...")
        print("This may take 10-15 minutes...")
        
        api.upload_folder(
            folder_path=".",
            repo_id="sugiv/cardvaultplus",
            repo_type="model",
            commit_message="Upload production CardVault+ SmolVLM model - 4 epochs trained, 0.000133 validation loss"
        )
        
        print("🎉 Successfully uploaded CardVault+ model!")
        print("🌐 Available at: https://huggingface.co/sugiv/cardvaultplus")
        return True
        
    except Exception as e:
        print(f"❌ Upload failed: {e}")
        return False

if __name__ == "__main__":
    success = upload_model()
    sys.exit(0 if success else 1)