import os
from huggingface_hub import HfApi, create_repo, upload_folder

def deploy():
    token = os.getenv("HUGGINGFACE_ACCSS_TOKEN")
    if not token:
        print("Erro: HUGGINGFACE_ACCSS_TOKEN não encontrada.")
        return

    api = HfApi(token=token)
    repo_id = "MatverseHub/mnb-memory-registry-space"
    
    print(f"Iniciando deploy para {repo_id}...")
    
    try:
        # Tentar criar o repositório se não existir
        create_repo(repo_id=repo_id, repo_type="space", space_sdk="docker", token=token, exist_ok=True)
        print(f"Repositório {repo_id} pronto.")
        
        # Upload da pasta do projeto
        upload_folder(
            folder_path="/home/ubuntu/matverse-mnb",
            repo_id=repo_id,
            repo_type="space",
            token=token,
            ignore_patterns=["__pycache__", "*.pyc"]
        )
        print("Deploy concluído com sucesso!")
        print(f"Acesse em: https://huggingface.co/spaces/{repo_id}")
        
    except Exception as e:
        print(f"Erro durante o deploy: {e}")

if __name__ == "__main__":
    deploy()