import requests

class SafeguardWalletAPI:
    def __init__(self, api_key):
        self.api_key = api_key
        self.base_url = "https://api.safeguardwallet.com"

    def get_balance(self, wallet_address):
        endpoint = f"{self.base_url}/balance"
        headers = {"Authorization": f"Bearer {self.api_key}"}
        params = {"wallet_address": wallet_address}

        response = requests.get(endpoint, headers=headers, params=params)

        if response.status_code == 200:
            return response.json()
        else:
            print(f"Error: {response.status_code}, {response.text}")