from huggingface_hub.utils import RepositoryNotFoundError
from huggingface_hub import HfApi, create_repo
import os

repo_id = "kalrap/predictive-maintenance-engine-data"
repo_type = "dataset"

token = os.getenv("HF_TOKEN")

# Initialize API
api = HfApi(token=token)

# Check if dataset repo exists
try:
    api.repo_info(repo_id=repo_id, repo_type=repo_type)
    print(f"Dataset '{repo_id}' already exists.")
except RepositoryNotFoundError:
    print(f"Dataset '{repo_id}' not found. Creating...")
    create_repo(
        repo_id=repo_id,
        repo_type=repo_type,
        private=False,
        token=token
    )
    print("Dataset repository created.")

# Upload dataset
api.upload_folder(
    folder_path="/content/drive/MyDrive/M12_Capstone_Project/predictive_maintenance_project/data",
    repo_id=repo_id,
    repo_type=repo_type
)

print("Dataset uploaded successfully.")