import os
from huggingface_hub import HfApi, HfFolder, upload_file

DATASET_REPO = "pcdoido2/video-gerados"
CATEGORIES = ["AVATAR WORLD", "BLOX FRUITS", "TOCA LIFE"]

def salvar_no_dataset(video_path, categoria):
    """
    Envia o vídeo finalizado para o Hugging Face Dataset na categoria escolhida.
    """
    if categoria not in CATEGORIES:
        raise ValueError("Categoria inválida.")

    # Caminho dentro do Dataset (exemplo: "AVATAR WORLD/video123.mp4")
    path_in_dataset = f"{categoria}/{os.path.basename(video_path)}"

    api = HfApi()

    # Usa o token do usuário (precisa estar configurado no Space ou no ambiente)
    token = os.getenv("HUGGINGFACEHUB_API_TOKEN") or HfFolder.get_token()

    if not token:
        raise ValueError(
            "Token do Hugging Face não encontrado. Configure a variável HUGGINGFACEHUB_API_TOKEN nos Settings > Variables do Space."
        )

    print(f"Fazendo upload de {video_path} para o dataset {DATASET_REPO} em {path_in_dataset}")
    upload_file(
        path_or_fileobj=video_path,
        path_in_repo=path_in_dataset,
        repo_id=DATASET_REPO,
        repo_type="dataset",
        token=token
    )
    print("Upload concluído.")