add eth and ocean balance as outputs

app.py

@@ -1,15 +1,22 @@
 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
-import os
+from ocean_lib.web3_internal.currency import from_wei # wei is the smallest denomination of ether e.g. like cents
 
 config = Config('config.ini')
 ocean = Ocean(config)
 
-def greet(private_key):
+def wallet(private_key):
     wallet = Wallet(ocean.web3, private_key, transaction_timeout=20, block_confirmations=config.block_confirmations)
-    return wallet.address
+    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))
+
+    return address, eth_balance, ocean_balance
 
 description = (
     "This demo calculate the number of tokens in your wallet on the Rinkeby test network using the Ocean Python library. "
@@ -25,11 +32,14 @@ article = (
     "</p>"
 )
 
+
 interface = gr.Interface(
-    greet,
+    wallet,
     "text",
     [
-        gr.outputs.Textbox(label="Public Key")
+        gr.outputs.Textbox(label="Public Key"),
+        gr.outputs.Textbox(label="ETH balance"),
+        gr.outputs.Textbox(label="ETOCEANH balance"),
     ],
     title="Ocean Token Calculator",
     description=description,