import os
from dotenv import load_dotenv
from huggingface_hub import HfApi, login

load_dotenv()
hf_token = os.getenv("HF_TOKEN")

if not hf_token:
    print("❌ Error: No se encontró HF_TOKEN en el archivo .env")
    exit()

# TODO: Configura tu usuario y el nuevo nombre en inglés
usuario_hf = os.getenv("HF_USERNAME")  # Asegúrate de tener esto en tu .env 
nombre_modelo = "student-dropout-predictor-rf"
repo_id = f"{usuario_hf}/{nombre_modelo}"

# Rutas locales
ruta_modelo = "modelos/modelo_rf_optimizado.pkl"
ruta_readme = "modelos/README.md"

print(f"🔐 Autenticando en Hugging Face...")
login(token=hf_token)
api = HfApi()

print(f"🚀 Preparando repositorio '{repo_id}'...")

try:
    api.create_repo(repo_id=repo_id, repo_type="model", exist_ok=True)

    print("📤 Subiendo archivo modelo_rf_optimizado.pkl...")
    api.upload_file(
        path_or_fileobj=ruta_modelo,
        path_in_repo="modelo_rf_optimizado.pkl",
        repo_id=repo_id,
        repo_type="model",
    )
    
    print("📝 Subiendo el Model Card (README.md)...")
    api.upload_file(
        path_or_fileobj=ruta_readme,
        path_in_repo="README.md",
        repo_id=repo_id,
        repo_type="model",
    )
    
    print("\n✅ ¡Modelo y documentación subidos con éxito!")
    print(f"🔗 Puedes verlo público aquí: https://huggingface.co/{repo_id}")

except Exception as e:
    print(f"\n❌ Error al subir los archivos: {e}")