How to get token or contract balance with wallet public key

gistfile1.txt

//intitate web3 here

export const getBalanceWithContract = async (address: string) => {
  let tokenContract = await new web3.eth.Contract(tokenAbi, address);
  let tokenBalance = await tokenContract.methods.balanceOf(WALLET_PUB).call();
   let decimals = await getTokenDecimals(address);
  tokenBalance = tokenBalance * (1 / Math.pow(10, decimals));
  tokenBalance = tokenBalance.toString();

  return tokenBalance;
};

const getTokenDecimals = async (address: string) => {
  const tokenContract = await new web3.eth.Contract(tokenAbi, address);
  let decimals = await tokenContract.methods.decimals().call();
  return decimals;
};