import requests

def get_nft_balance(api_key, account_address, contract_address):
    # PolygonScan API URL 구성
    url = f"https://api.polygonscan.com/api?module=account&action=tokenbalance&contractaddress={contract_address}&address={account_address}&tag=latest&apikey={api_key}"

    # API 요청
    response = requests.get(url)

    # 응답 확인 및 처리
    if response.status_code == 200:
        data = response.json()
        balance = data.get('result')
        if balance is not None:
            # ERC-721은 보통 1 NFT당 1을 가지므로, balance를 직접 반환
            real_balance = int(balance)
            print(f"계정 주소 {account_address}는(은) 토큰 컨트랙트 {contract_address}의 NFT를 {real_balance}개 가지고 있습니다.")
            return real_balance
        else:
            print("NFT 잔액 정보를 가져올 수 없습니다.")
    else:
        print("API 요청에 실패했습니다.")

# 사용 예
api_key = "4H655NPT5229MKFE2NNHIS2GNZ3U4UTGT2"
account_address = "0x9526E51ee3D9bA02Ef674eB1E41FB24Dc2165380"
contract_address = "0xA4B37bE40F7b231Ee9574c4b16b7DDb7EAcDC99B"

# 함수 호출
get_nft_balance(api_key, account_address, contract_address)