import sys
from huggingface_hub import snapshot_download, upload_folder

REPO = "EagleDev-Innovation/Modelos-EagleDev-Innovation"
CARPETA = "modelos_entrenados"

def bajar():
    snapshot_download(
        repo_id=REPO,
        repo_type="model",
        allow_patterns=f"{CARPETA}/**",
        local_dir=".",
    )
    print("Modelos descargados en", CARPETA)

def subir(mensaje):
    upload_folder(
        repo_id=REPO,
        repo_type="model",
        folder_path=CARPETA,
        path_in_repo=CARPETA,
        commit_message=mensaje,
        ignore_patterns=[".locks/**", ".cache/**", "__pycache__/**", "*.tmp"],
    )
    print("Cambios subidos")

if __name__ == "__main__":
    accion = sys.argv[1]
    if accion == "bajar":
        bajar()
    elif accion == "subir":
        subir(sys.argv[2] if len(sys.argv) > 2 else "Actualizar modelos")