from huggingface_hub import HfApi, upload_folder, HfFolder
import os

# Get token from environment variable (safe way)
HF_TOKEN = os.getenv("HF_TOKEN")
repo_name = "FakeNewsDeploy"
username = "Loki1one"

# Login and create repo if needed
HfFolder.save_token(HF_TOKEN)
api = HfApi()
full_repo_id = f"{username}/{repo_name}"
api.create_repo(repo_name, token=HF_TOKEN, exist_ok=True)

# Upload model files
upload_folder(
    repo_id=full_repo_id,
    folder_path=".",  # current directory
    commit_message="Re-upload Fake News BERT model",
    token=HF_TOKEN
)

print("✅ Upload successful!")
print(f"🔗 Check your model at: https://huggingface.co/{full_repo_id}")