display eth and ocean balance

app.py

@@ -16,14 +16,14 @@ def wallet(hello):
     private_key = account.key.hex()
     address = account.address
 
-    # wallet = Wallet(ocean.web3, private_key, transaction_timeout=20, block_confirmations=config.block_confirmations)
+    wallet = Wallet(ocean.web3, private_key, transaction_timeout=20, block_confirmations=config.block_confirmations)
 
-    # OCEAN_token = BToken(ocean.web3, ocean.OCEAN_address)
+    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))
+    eth_balance = from_wei(ocean.web3.eth.get_balance(address))
+    ocean_balance = from_wei(OCEAN_token.balanceOf(address))
 
-    return address, mnemonic, private_key # eth_balance, ocean_balance
+    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)
@@ -76,10 +76,9 @@ interface = gr.Interface(
         #gr.outputs.Textbox(label="Public Key"),
         #gr.outputs.Textbox(label="Algorithm token balance"),
         gr.outputs.Textbox(label="Public Address"),
-        gr.outputs.Textbox(label="Private Key"),
         gr.outputs.Textbox(label="Recovery Passphrase"),
-        # gr.outputs.Textbox(label="ETH balance"),
-        # gr.outputs.Textbox(label="OCEAN balance"),
+        gr.outputs.Textbox(label="ETH balance"),
+        gr.outputs.Textbox(label="OCEAN balance"),
     ],
     title="Algorithm Web3 Wallet",
     description=description,