add alg token balance

app.py

@@ -8,7 +8,7 @@ from ocean_lib.web3_internal.currency import from_wei # wei is the smallest deno
 config = Config('config.ini')
 ocean = Ocean(config)
 
-def wallet(private_key):
+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)
@@ -16,30 +16,51 @@ def wallet(private_key):
     eth_balance = from_wei(ocean.web3.eth.get_balance(wallet.address))
     ocean_balance = from_wei(OCEAN_token.balanceOf(wallet.address))
 
-    return address, eth_balance, ocean_balance
+    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 = (
-    "This demo calculate the number of tokens in your wallet on the Rinkeby test network using the Ocean Python library. \n"
-    "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 Rinkeby tokens)."
-    "For more info on private keys, see [this](https://github.com/oceanprotocol/ocean.py/blob/main/READMEs/wallets.md) from the ocean.py documentation"
+    "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://oceanprotocol.com/' target='_blank'>Ocean Protocol</a> | "
+    "<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://www.algovera.ai' target='_blank'>Algovera</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,
-    "text",
+    [
+        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,