Update app.py

app.py

@@ -6,7 +6,9 @@ import json
 import os
 import requests
 from blockchain import Blockchain
-from mychain import MyChain
+from mychain import MyChainSend 
+from mychain import MyChainRec 
+
 from huggingface_hub import (create_repo,get_full_repo_name,upload_file,CommitOperationAdd,HfApi)
 
 main_chain='https://huggingface.co/datasets/Omnibus/blockchain-sim/raw/main/chains/'
@@ -408,7 +410,7 @@ def get_my_chain_send(sender_name=None):
     try:
         r = requests.get(f'{main_balance}{sender_name}.json') 
         print (f'r={r.text}')
-        mychain_send = MyChain(chain_load=main_balance,load=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}')
@@ -423,14 +425,14 @@ def get_my_chain_rec(recipient_name=None):
 
     try:
         r = requests.get(f'{main_balance}{recipient_name}.json') 
-        mychain_rec = MyChain(chain_load=main_balance,load=r.text)
+        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 = MyChain(chain_load=main_balance,create=recipient_name)
+            mychain_rec = MyChainRec(chain_load=main_balance,create=recipient_name)
             response = mychain_rec.chain
             message = f"Blockchain loaded from: {main_balance}{recipient_name}.json"
             return response,message