check for input private key

app.py

@@ -12,13 +12,13 @@ ocean = Ocean(config)
 
 def wallet(private_key):
     
-    # if private_key == "":
-    #     account, mnemonic = get_wallet()
-
-    #     private_key = account.key.hex()
-    #     address = account.address
-
+    if private_key:
+        pass
+    else:
+        account, mnemonic = get_wallet()
 
+        private_key = account.key.hex()
+        address = account.address
 
     wallet = Wallet(ocean.web3, private_key, transaction_timeout=20, block_confirmations=config.block_confirmations)
 
@@ -27,7 +27,7 @@ def wallet(private_key):
     eth_balance = from_wei(ocean.web3.eth.get_balance(address))
     ocean_balance = from_wei(OCEAN_token.balanceOf(address))
 
-    return type(private_key), 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)