Update app.py

app.py

@@ -18,6 +18,8 @@ def checkp(inp):
     elif inp != pa:
         return gr.update(visible=True), gr.update(visible=False)
 
+
+
 def get_my_chain_send(sender_name=None):
     global mychain_send
     mes_error = ""
@@ -60,6 +62,59 @@ def get_my_chain_rec(recipient_name=None):
         except Exception:
             message = f"Error loading from: {recipient_name}"
             return ["Error Loading Chain"],message
+
+
+###################################################
+
+
+def mychain_mine_block_send(chain_r=None,chain_n=None):
+    previous_block = mychain_send.print_previous_block()
+    previous_proof = previous_block['proof']
+    proof = mychain_send.proof_of_work(previous_proof)
+    previous_hash = mychain_send.hash(previous_block)
+    block = mychain_send.create_block(proof, previous_hash,chain_r,chain_n)
+
+    response = {'message': 'A block is MINED',
+                'index': block['index'],
+                'timestamp': block['timestamp'],
+                'proof': block['proof'],
+                'previous_hash': block['previous_hash']
+               }
+   
+    message = "A block is MINED"
+    show_chain = display_chain_send()
+    
+    if len(mychain_send.chain) > 1000:
+        mychain_send.reset()
+        response = None
+        show_chain=display_chain_send()
+        message = "New Chain Created at Max 20 Blocks"
+    
+    return response, show_chain,message
+
+def mychain_mine_block_rec(chain_r=None,chain_n=None):
+    previous_block = mychain_rec.print_previous_block()
+    previous_proof = previous_block['proof']
+    proof = mychain_rec.proof_of_work(previous_proof)
+    previous_hash = mychain_rec.hash(previous_block)
+    block = mychain_rec.create_block(proof, previous_hash,chain_r,chain_n)
+
+    response = {'message': 'A block is MINED',
+                'index': block['index'],
+                'timestamp': block['timestamp'],
+                'proof': block['proof'],
+                'previous_hash': block['previous_hash']
+               }
+    message = "A block is MINED"
+    show_chain = display_chain_rec()
+    if len(mychain_rec.chain) > 1000:
+        mychain_rec.reset()
+        response = None
+        show_chain=display_chain_rec()
+        message = "New Chain Created at Max 20 Blocks"
+    return response, show_chain, message
+##############################################        
+
 ############################################
 def send_message(send,rec,message):
     response_send={}
@@ -109,11 +164,11 @@ def send_message(send,rec,message):
         mychain_rec.new_transaction(f"{send}",f"{rec}",f"{message}")
         message_rec = "Transaction Added to Pool"
         data_rec = pd.DataFrame(mychain_rec.pending_transactions)
-        response_send, show_chain_send, message_send = chatchain.mychain_mine_block_send(chain_r=None,chain_n=send)
+        response_send, show_chain_send, message_send = mychain_mine_block_send(chain_r=None,chain_n=send)
         
         #mychain_mine_block_trans(balance_send, chain_r=None,chain_n=send)
         
-        response_rec, show_chain_rec, message_rec =  chatchain.mychain_mine_block_rec(chain_r=None,chain_n=rec)
+        response_rec, show_chain_rec, message_rec =  mychain_mine_block_rec(chain_r=None,chain_n=rec)
         mes = (f'Send: {message_send} :: Recieve: {message_rec}')
         _,rec_send=update_send_list()
         _,rec_drop=update_rec_list()