#!/usr/bin/env python3
"""Automated Hugging Face Upload Script"""

from huggingface_hub import HfApi, upload_folder

def upload_model():
    model_dir = "./hf_model"
    repo_id = "your-username/custom-57m-language-model"

    print("🤗 Starting Hugging Face upload...")
    print(f"📁 Model directory: {model_dir}")
    print(f"🔗 Repository: {repo_id}")

    try:
        api = HfApi()
        api.create_repo(repo_id=repo_id, exist_ok=True)
        print("✅ Repository created/verified")

        upload_folder(
            folder_path=model_dir,
            repo_id=repo_id,
            repo_type="model",
            commit_message="Upload 57.6M parameter custom language model"
        )

        print("🎉 Upload completed!")
        print(f"🔗 Your model: https://huggingface.co/{repo_id}")

    except Exception as e:
        print(f"❌ Upload failed: {e}")
        print("💡 Make sure you're logged in: huggingface-cli login")

if __name__ == "__main__":
    upload_model()