show ETH and OCEAN balance

app.py

@@ -1,18 +1,29 @@
 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.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
 from wallet import get_wallet
 
-# config = Config('config.ini')
-# ocean = Ocean(config)
+config = Config('config.ini')
+ocean = Ocean(config)
 
-def w(hello):
+def wallet(hello):
     account, mnemonic = get_wallet()
-    return account.address, mnemonic
+
+    private_key = account.key
+    address = account.address
+
+    wallet = Wallet(ocean.web3, private_key, transaction_timeout=20, block_confirmations=config.block_confirmations)
+
+    OCEAN_token = BToken(ocean.web3, ocean.OCEAN_address)
+
+    eth_balance = from_wei(ocean.web3.eth.get_balance(address))
+    ocean_balance = from_wei(OCEAN_token.balanceOf(address))
+
+    return address, mnemonic, eth_balance, ocean_balance
 
 # def wallet(private_key, did):
 #     wallet = Wallet(ocean.web3, private_key, transaction_timeout=20, block_confirmations=config.block_confirmations)
@@ -57,17 +68,17 @@ article = (
 
 
 interface = gr.Interface(
-    w,
+    wallet,
     [
         gr.inputs.Textbox(label="Greetings"),
     ],
     [
         #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"),
         gr.outputs.Textbox(label="Account Address"),
         gr.outputs.Textbox(label="Recovery Passphrase"),
+        gr.outputs.Textbox(label="ETH balance"),
+        gr.outputs.Textbox(label="OCEAN balance"),
     ],
     title="Algorithm Web3 Wallet",
     description=description,