import os
from huggingface_hub import upload_folder, upload_file, login

token = os.environ.get("hfToken")
login(token)

repo_id = "Neon-coding/Ureola-vision"
repo_type = "dataset"

# 1. Upload parquet shards into /data
upload_folder(
    folder_path="/data/image-shards",   # your local folder
    path_in_repo="data",                # goes into /data in repo
    repo_id=repo_id,
    repo_type=repo_type
)

# 2. Upload JSON to root
upload_file(
    path_or_fileobj="/data/images_state.json",
    path_in_repo="images_state.json",
    repo_id=repo_id,
    repo_type=repo_type
)

print("Done. Structure is correct.")