check input type

app.py

@@ -10,11 +10,15 @@ from wallet import get_wallet
 config = Config('config.ini')
 ocean = Ocean(config)
 
-def wallet(hello):
-    account, mnemonic = get_wallet()
+def wallet(private_key):
+    
+    # if private_key == "":
+    #     account, mnemonic = get_wallet()
+
+    #     private_key = account.key.hex()
+    #     address = account.address
+
 
-    private_key = account.key.hex()
-    address = account.address
 
     wallet = Wallet(ocean.web3, private_key, transaction_timeout=20, block_confirmations=config.block_confirmations)
 
@@ -23,7 +27,7 @@ def wallet(hello):
     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
+    return type(private_key), private_key, eth_balance, ocean_balance
 
 # def wallet(private_key, did):
 #     wallet = Wallet(ocean.web3, private_key, transaction_timeout=20, block_confirmations=config.block_confirmations)