import os
import subprocess
from huggingface_hub import HfApi, HfFolder

# ======== CONFIGURACIÓN ========
HF_TOKEN = os.environ.get("HF_TOKEN", "").strip()
REPO_ID = os.environ.get("HF_REPO", "sob111/xttsv2-voxpopuli-es-500")
CONFIG_PATH = "./config.json"

# ======== GUARDAR TOKEN ========
if HF_TOKEN:
    print("=== Guardando token de Hugging Face ===")
    HfFolder.save_token(HF_TOKEN)
else:
    raise RuntimeError("No se encontró HF_TOKEN. Configúralo como variable de entorno.")

# ======== ENTRENAMIENTO XTTSv2 ========
print("=== Iniciando entrenamiento XTTSv2 ===")
try:
    subprocess.run(
        ["tts", "--config_path", CONFIG_PATH],
        check=True
    )
except subprocess.CalledProcessError:
    raise RuntimeError("El entrenamiento XTTSv2 falló. Revisa los logs anteriores.")
print("=== Entrenamiento finalizado ===")

# ======== SUBIR MODELO A HUGGING FACE ========
print("=== Subiendo modelo fine-tune a Hugging Face ===")
api = HfApi()
output_dir = "./output_model"

api.upload_folder(
    folder_path=output_dir,
    repo_id=REPO_ID,
    repo_type="model",
    token=HF_TOKEN
)

print(f"=== Modelo subido correctamente a https://huggingface.co/{REPO_ID} ===")