from huggingface_hub import HfApi, login
import os

HF_USERNAME = "SharleyK"
SPACE_NAME = "PredictiveMaintenance"
HF_TOKEN = os.getenv("HF_TOKEN")

if not HF_TOKEN:
    raise RuntimeError("HF_TOKEN environment variable not set")

login(token=HF_TOKEN)
api = HfApi()

repo_id = f"{HF_USERNAME}/{SPACE_NAME}"

api.create_repo(
    repo_id=repo_id,
    repo_type="space",
    space_sdk="docker",
    exist_ok=True
)

api.upload_folder(
    folder_path=".",
    repo_id=repo_id,
    repo_type="space"
)

print("Deployment successful!")
print(f"https://huggingface.co/spaces/{HF_USERNAME}/{SPACE_NAME}")