Update app.py

app.py

@@ -18,7 +18,48 @@ 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 = ""
+    try:
+        r = requests.get(f'{main_balance}{sender_name}.json') 
+        
+        #print (f'r={r.text}')
+        mychain_send = MyChainSend(chain_load=main_balance,load=r.text)
+
+        response = {'chain': mychain_send.chain,
+                'length': len(mychain_send.chain)}
+        #print (f'response={response}')
+        
+            
+        message = f"Blockchain loaded from: {main_balance}{sender_name}.json"
+        return response,message
+    except Exception:
+        message = f"Error loading from: {sender_name}"
+        print (message)
+        return ["Error Loading Chain"],message
+        
+def get_my_chain_rec(recipient_name=None):
+    global mychain_rec
 
+    try:
+        r = requests.get(f'{main_balance}{recipient_name}.json') 
+        mychain_rec = MyChainRec(chain_load=main_balance,load=r.text)
+        response = {'chain': mychain_rec.chain,
+                'length': len(mychain_rec.chain)}
+        message = f"Blockchain loaded from: {main_balance}{recipient_name}.json"
+        return response,message
+    except Exception:
+        try:
+            mychain_rec = MyChainRec(chain_load=main_balance,create=recipient_name)
+            response = {'chain': mychain_rec.chain,
+                    'length': len(mychain_rec.chain)}
+            
+            message = f"Blockchain loaded from: {main_balance}{recipient_name}.json"
+            return response,message
+        except Exception:
+            message = f"Error loading from: {recipient_name}"
+            return ["Error Loading Chain"],message
 ############################################
 def send_message(send,rec,message):
     response_send={}
@@ -37,9 +78,9 @@ def send_message(send,rec,message):
     balance_rec = 0
     mes = "blank message"
     try:
-        response,message =  chatchain.get_my_chain_send(send)
+        response,message =  get_my_chain_send(send)
         send_rec = response
-        response,message =  chatchain.get_my_chain_rec(rec)
+        response,message =  get_my_chain_rec(rec)
         lod_rec = response
         '''
         mes1,val,mes2,ind = chatchain.valid_rec()
@@ -62,12 +103,12 @@ def send_message(send,rec,message):
     if p==False:
         return (mes, p,None,None,None,None,None,None,None,None,None,None,None)
     if p==True:
-        chatchain.MyChainSend.new_transaction(sender=send,recipient=rec,message=message)
+        mychain_send.new_transaction(sender=send,recipient=rec,message=message)
         message_send = "Transaction Added to Pool"
-        data_send = pd.DataFrame(chatchain.MyChainSend.pending_transactions)
-        chatchain.MyChainRec.new_transaction(f"{send}",f"{rec}",f"{message}")
+        data_send = pd.DataFrame(mychain_send.pending_transactions)
+        mychain_rec.new_transaction(f"{send}",f"{rec}",f"{message}")
         message_rec = "Transaction Added to Pool"
-        data_rec = pd.DataFrame(chatchain.MyChainRec.pending_transactions)
+        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)
         
         #mychain_mine_block_trans(balance_send, chain_r=None,chain_n=send)