app.py

import gradio as gr
from ocean_lib.config import Config
from ocean_lib.models.btoken import BToken #BToken is ERC20
from ocean_lib.ocean.ocean import Ocean
from ocean_lib.web3_internal.wallet import Wallet
from ocean_lib.web3_internal.currency import from_wei # wei is the smallest denomination of ether e.g. like cents
from ocean_lib.web3_internal.currency import pretty_ether_and_wei

config = Config('config.ini')
ocean = Ocean(config)

def wallet(private_key, did):
    wallet = Wallet(ocean.web3, private_key, transaction_timeout=20, block_confirmations=config.block_confirmations)
    address = wallet.address
    OCEAN_token = BToken(ocean.web3, ocean.OCEAN_address)

    eth_balance = from_wei(ocean.web3.eth.get_balance(wallet.address))
    ocean_balance = from_wei(OCEAN_token.balanceOf(wallet.address))

    asset = ocean.assets.resolve(did)

    ALG_ddo = ocean.assets.resolve(did)
    alg_token = ocean.get_data_token(ALG_ddo.data_token_address)

    alg_token_balance = pretty_ether_and_wei(alg_token.balanceOf(wallet.address))

    return address, eth_balance, ocean_balance, alg_token_balance

description = (
    "The Ocean marketplace allows you to publish and sell algorithms and datasets in exchange for crypto tokens. "
    "This demo counts the number of ETH and OCEAN tokens in your wallet for a given private key. It uses a test network (Rinkeby) where the tokens have no real value. "
    "You can export your private key from your MetaMask wallet. We highly recommend doing this with a wallet that has no real tokens in it (only test tokens). "
    "For a guide on how to install MetaMask wallet (an extension in your browser), check the link at the bottom of the page. "
    "Initially, your wallet should have no tokens. You can request ETH and OCEAN test tokens from faucets at the links at the bottom of the page. "
    "You can also count the number of algorithm tokens in your wallet for an algorithm that you specify. "
    "To buy an algorithm token, you can search for algorithms on the Ocean marketplace (see link at bottom). Make sure to use algorithms that are on the Rinkeby test network (you need to select Rinkeby from the dropdown menu). "
    "We have provided a link to our DCGAN model on the test network at the bottom. If you can't see it you are not on the test network. "
    "After you buy an algorithm token, you need to locate the DID in the metadata on the marketplace. Then enter it into the input textbox. "
    "Later we will add HF spaces apps to search algorithms and buy algorithm tokens, which you can use to run demos of the algorithms."
)

article = (
    "<p style='text-align: center'>"
    "<a href='https://market.oceanprotocol.com/' target='_blank'>Ocean marketplace</a> | "
    "<a href='https://market.oceanprotocol.com/asset/did:op:E2e123115d5758Dd4C6F434E1c142e72ed8B2820' target='_blank'>DCGAN algorithm</a> | "
    "<a href='https://github.com/oceanprotocol/ocean.py' target='_blank'>Ocean Python Library</a> | "
    "<a href='https://docs.oceanprotocol.com/tutorials/metamask-setup/' target='_blank'>Guide for installing MetaMask</a> | "
    "<a href='https://faucet.rinkeby.io/' target='_blank'>ETH faucet</a> | "
    "<a href='https://faucet.rinkeby.oceanprotocol.com/' target='_blank'>OCEAN faucet</a> | "
    "</p>"
)


interface = gr.Interface(
    wallet,
    [
        gr.inputs.Textbox(label="Private Key"),
        gr.inputs.Textbox(placeholder="did:op:E2e123115d5758Dd4C6F434E1c142e72ed8B2820", label="Algorithm DID"),
    ],
    [
        gr.outputs.Textbox(label="Public Key"),
        gr.outputs.Textbox(label="ETH balance"),
        gr.outputs.Textbox(label="OCEAN balance"),
        gr.outputs.Textbox(label="Algorithm token balance"),
    ],
    title="Ocean Token Calculator",
    description=description,
    article=article,
    theme="huggingface",
)

interface.launch()