import sys
from huggingface_hub import snapshot_download, upload_folder

REPO = "EagleDev-Innovation/Modelos-EagleDev-Innovation"
PATRONES = [
    "entrenamiento/*.py", "entrenamiento/*.md", "entrenamiento/*.ipynb",
    "datos/peso/*.csv",
    "datos/externos/emma_mat.csv",
    "datos/externos/emma_clip.*",
]

def bajar():
    snapshot_download(repo_id=REPO, repo_type="model",
                      allow_patterns=PATRONES, local_dir=".")
    print("Código y datos descargados")

def subir(mensaje):
    upload_folder(repo_id=REPO, repo_type="model", folder_path=".",
                  allow_patterns=PATRONES,
                  ignore_patterns=[".venv/**", "**/__pycache__/**"],
                  commit_message=mensaje)
    print("Código y datos subidos")

if __name__ == "__main__":
    if sys.argv[1] == "bajar":
        bajar()
    elif sys.argv[1] == "subir":
        subir(sys.argv[2] if len(sys.argv) > 2 else "Actualizar código y datos")