Update mychain.py

mychain.py

@@ -25,16 +25,17 @@ class MyChain:
         self.chain = []
         self.pending_transactions = []
         self.create_block(proof=1, previous_hash='0',chain_n=create)
-    def create_block(self, proof, previous_hash,chain_r=None,chain_n=None):
+    def create_block(self, balance, proof, previous_hash,chain_r=None,chain_n=None):
         if chain_r=="" or chain_r==None:
-            chain_r=f"{main_chain.split('datasets/',1)[1].split('/raw',1)[0]}"
+            chain_r=f"{balance_chain.split('datasets/',1)[1].split('/raw',1)[0]}"
         if chain_n !="" and chain_n !=None:
-            chain_n = f"{main_chain.split('main/',1)[1]}{chain_n}"
+            chain_n = f"{balance_chain.split('main/',1)[1]}{chain_n}.json"
         if chain_n=="" or chain_n==None:
             chain_n=chain_d
         block = {'index': len(self.chain) + 1,
                 'timestamp': str(datetime.datetime.now()),
                 'transactions': self.pending_transactions,
+                'balance': balance, 
                 'proof': proof,
                 'previous_hash': previous_hash}
         if self.block_valid(block) == True: