Update arquivos.py

arquivos.py

@@ -1,19 +1,36 @@
 import os
 import shutil
+from huggingface_hub import HfApi, HfFolder, upload_file
 
-BASE_FOLDER = "uploaded_files"
+# Nome do Dataset Hugging Face
+DATASET_REPO = "pcdoido2/video-gerados"
 CATEGORIES = ["AVATAR WORLD", "BLOX FRUITS", "TOCA LIFE"]
 
-# Criar pastas se não existirem
-for cat in CATEGORIES:
-    os.makedirs(os.path.join(BASE_FOLDER, cat), exist_ok=True)
-
-def salvar_no_gerenciador(video_path, categoria):
+def salvar_no_dataset(video_path, categoria):
     """
-    Copia o vídeo finalizado para a categoria no gerenciador de arquivos.
+    Envia o vídeo finalizado para o Hugging Face Dataset na categoria escolhida.
     """
     if categoria not in CATEGORIES:
         raise ValueError("Categoria inválida.")
-    destino = os.path.join(BASE_FOLDER, categoria, os.path.basename(video_path))
-    shutil.copy(video_path, destino)
-    print(f"Arquivo {video_path} salvo em {destino}")
+
+    # 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 local com HfFolder)
+    token = HfFolder.get_token()
+
+    if token is None:
+        raise ValueError("Token do Hugging Face não encontrado. Configure com HfFolder ou no ambiente.")
+
+    # Faz upload do arquivo
+    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.")