Update app.py

app.py

@@ -8,6 +8,8 @@ import requests
 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/'
+main_balance='https://huggingface.co/datasets/Omnibus/blockchain-sim/raw/main/balance/'
+
 main_nodes='https://huggingface.co/datasets/Omnibus/blockchain-sim/raw/main/node_file1.json'
 main_pending='https://huggingface.co/datasets/Omnibus/blockchain-sim/raw/main/pending1.json'
 token_self = os.environ['HF_TOKEN']
@@ -18,6 +20,53 @@ node_file='node_file.json'
 space='blockchain-simulator-dev1'
 api = HfApi(token=token_self)
 
+class Balance_sheet:
+    def create_block(self, proof, previous_hash,chain_r=None,chain_n=None,balance=None):
+        if chain_r=="" or chain_r==None:
+            chain_r=f"{main_chain.split('datasets/',1)[1].split('/raw',1)[0]}"
+        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':self.balance
+                'proof': proof,
+                'previous_hash': previous_hash}
+        if self.block_valid(block) == True:
+            
+            self.pending_transactions = []
+            self.chain.append(block)
+            json_object = json.dumps(self.chain, indent=4)
+            with open("tmp.json", "w") as outfile:
+                outfile.write(json_object)
+            try: 
+                api.upload_file(
+                path_or_fileobj="tmp.json",
+                path_in_repo=chain_n,
+                repo_id=chain_r,
+                token=token_self,
+                repo_type="dataset",
+                )
+                os.remove("tmp.json")
+        
+            except Exception as e:
+                print(e)
+                pass         
+            return block
+        else:
+            block = {"Not Valid"}
+            print("not Valid")
+            return block
+    def print_previous_block(self):
+        return self.chain[-1]
+    def new_transaction(self, sender, recipient, amount):
+        transaction = {
+            'sender': sender,
+            'recipient': recipient,
+            'amount': amount
+        }
+        self.pending_transactions.append(transaction)
+
 class Blockchain:
 
     def __init__(self,load=None,create=None):
@@ -150,10 +199,15 @@ class Blockchain:
             block_index += 1
         
         return out, ind, mes
+#def check_balance(sender,amount):
+    
+
 
 def bc_transactions(sender,recipient,amount):
-    blockchain.new_transaction(f"{sender}",f"{recipient}",f"{amount}")
-    message = "Transaction Added to Pool"
+    out = check_balance(sender,amount)
+    if out == true:
+        blockchain.new_transaction(f"{sender}",f"{recipient}",f"{amount}")
+        message = "Transaction Added to Pool"
     return pd.DataFrame(blockchain.pending_transactions),message,None,None,None
 
 def create_chain(create=None):
@@ -319,6 +373,79 @@ def add_node(this_space,repo,space,chain_file):
     except Exception as e:
         pass         
 
+        
+start_tokens = 10000000
+source="Main"
+def issue_tokens(send,rec,amount):
+    #print(f"{api.whoami(['name'])}")
+    #repo = f'omnibus/{space}'
+    #is_valid='True'
+    mes = None
+    try:
+        r = requests.get(f'{main_balance}/{send}') 
+        lod = json.loads(r.text)
+    except:
+        lod=[]
+        pass
+    block = {'index': len(lod) + 1,
+            'timestamp': str(datetime.datetime.now()),
+            'sender': f'{send}',
+            'recipient':f'{rec}',
+             
+            'amount': f'{amount}'}
+    lod.append(block)    
+    
+    json_object = json.dumps(lod, indent=4)
+    with open("tmp1.json", "w") as outfile:
+        outfile.write(json_object)
+    try: 
+        api.upload_file(
+        path_or_fileobj="tmp1.json",
+        path_in_repo=f'{main_balance.split('main/',1)[1]}/{send}.json',
+        repo_id=main_balance.split('datasets/',1)[1].split('/raw',1)[0],
+        token=token_self,
+        repo_type="dataset",
+        )
+        os.remove("tmp1.json")
+    lod.clear()
+    except Exception as e:
+        mes=e
+        pass 
+        
+    try:
+        r = requests.get(f'{main_balance}/{rec}') 
+        lod = json.loads(r.text)
+    except:
+        lod=[]
+        pass
+    block = {'index': len(lod) + 1,
+            'timestamp': str(datetime.datetime.now()),
+            'sender': f'{send}',
+            'recipient':f'{rec}',
+             
+            'amount': f'{amount}'}
+    lod.append(block)    
+    
+    json_object = json.dumps(lod, indent=4)
+    with open("tmp1.json", "w") as outfile:
+        outfile.write(json_object)
+    try: 
+        api.upload_file(
+        path_or_fileobj="tmp1.json",
+        path_in_repo=f'{main_balance.split('main/',1)[1]}/{rec}',
+        repo_id=main_balance.split('datasets/',1)[1].split('/raw',1)[0],
+        token=token_self,
+        repo_type="dataset",
+        )
+        os.remove("tmp1.json")
+
+    except Exception as e:
+        mes = e
+        pass   
+    if mes = None:
+        mes = f'{amount} sent from {send} to {rec}'
+    return mes
+    
 with gr.Blocks() as bc:
     with gr.Row(visible=True) as invalid:
         pass_box = gr.Textbox()
@@ -356,7 +483,11 @@ with gr.Blocks() as bc:
                 node_space=gr.Textbox(label="Node Space")
                 node_file=gr.Textbox(label="Node File")
             node_add=gr.Button("Add Node")
+        with gr.Accordion("Tokens", open=False):
+            issue_btn=gr.Button()
 
+    issue_btn.click(issue_tokens,[send,rec,am],block_text)
+    
     node_add.click(add_node,[this_space,node_repo,node_space,node_file],block_text)
     pass_btn.click(checkp,pass_box,[invalid,valida])
     in_chain_btn.click(get_chain,[chain_repo,chain_n],[chain_json,block_text])