Create chain.py

chain.py

@@ -0,0 +1,371 @@
+import gradio as gr
+import pandas as pd 
+import datetime
+import hashlib
+import json
+import os
+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_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']
+pa=os.environ['PASS']
+repo_d='Omnibus/static-bin'
+chain_d='chain1.json'
+node_file='node_file.json'
+space='blockchain-simulator-dev1'
+api = HfApi(token=token_self)
+
+class Blockchain:
+
+    def __init__(self,load=None,create=None):
+        
+        self.pending_transactions = []
+        if load == None or load=="":
+            self.chain = []
+            self.create_block(proof=1, previous_hash='0',chain_n=create)
+        elif load != None and load !="":
+            #r = requests.get(load) 
+            lod = json.loads(load)
+            self.chain = lod
+    def reset(self,create=None):
+        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):
+        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,
+                '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)
+    def proof_of_work(self, previous_proof):
+        new_proof = 1
+        check_proof = False
+        while check_proof is False:
+            hash_operation = hashlib.sha256(
+                str(new_proof**2 - previous_proof**2).encode()).hexdigest()
+            if hash_operation[:5] == '00000':
+                check_proof = True
+            else:
+                new_proof += 1
+        return new_proof
+
+    def hash(self, block):
+        encoded_block = json.dumps(block, sort_keys=True).encode()
+        return hashlib.sha256(encoded_block).hexdigest() 
+    def block_valid(self, block):
+        print (block)
+        #prev_block=len(self.chain)
+        if len(self.chain) > 0:
+            prev_block = len(self.chain)-1
+            previous_block = self.chain[prev_block]
+            print (previous_block)
+            out=True
+            ind=None
+            mes=None
+            if block['previous_hash'] != self.hash(previous_block):
+                out=False
+                #ind=block_index
+                mes='Hash'
+    
+            previous_proof = previous_block['proof']
+            proof = block['proof']
+            hash_operation = hashlib.sha256(
+                str(proof**2 - previous_proof**2).encode()).hexdigest()
+    
+            if hash_operation[:5] != '00000':
+                out=False
+                #ind=block_index+1
+                mes='Proof'
+            previous_block = block
+        else: 
+            out = True
+        return out 
+
+    def chain_valid(self, chain):
+        previous_block = chain[0]
+        block_index = 1 
+        out=True
+        ind=None
+        mes=None
+        while block_index < len(chain):
+            block = chain[block_index]
+            if block['previous_hash'] != self.hash(previous_block):
+                out=False
+                ind=block_index
+                mes='Hash'
+                break
+
+            previous_proof = previous_block['proof']
+            proof = block['proof']
+            hash_operation = hashlib.sha256(
+                str(proof**2 - previous_proof**2).encode()).hexdigest()
+
+            if hash_operation[:5] != '00000':
+                out=False
+                ind=block_index+1
+                mes='Proof'
+                break
+            previous_block = block
+            block_index += 1
+        
+        return out, ind, mes
+
+def bc_transactions(sender,recipient,amount):
+    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):
+    global blockchain
+    blockchain = Blockchain(create=create)
+    #blockchain.reset(create=create)
+    return "New Chain Created",None,display_chain()
+
+def display_chain():
+    response = {'chain': blockchain.chain,
+                'length': len(blockchain.chain)}
+    return response
+    
+def mine_block(chain_r=None,chain_n=None):
+    previous_block = blockchain.print_previous_block()
+    previous_proof = previous_block['proof']
+    proof = blockchain.proof_of_work(previous_proof)
+    previous_hash = blockchain.hash(previous_block)
+    block = blockchain.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()
+    if len(blockchain.chain) > 20:
+        blockchain.reset()
+        response = None
+        show_chain=display_chain()
+        message = "New Chain Created at Max 20 Blocks"
+    return response, show_chain,pd.DataFrame(blockchain.pending_transactions), message
+
+def sort_valid():
+    #nodes=main_nodes.replace("'","")
+    f = requests.get(f'{main_nodes}')
+    t = open('tmp.json','w')
+    t.write(f.text)
+    t.close()
+    f = open('tmp.json','r')
+    f = f.read()
+    j = json.loads(f)
+    #j=f.json()
+    print (len(j))
+    cnt=0
+    valid_chains=[]
+    not_valid=[]
+    response=''
+    while cnt < len(j):
+        try:   
+            r = requests.get(f'{j[cnt]["url"]}')
+            g = open('tmp1.json','w')
+            g.write(r.text)
+            g.close()
+            gg = open('tmp1.json','r')
+            ggg=gg.read()
+            print (ggg)
+            leng=len(json.loads(ggg))
+            print(f'Blockchain {cnt}: Length {leng} ')
+            valid,ind,mes = blockchain.chain_valid(json.loads(ggg))
+        except Exception:
+            valid=False
+            leng="No Data"
+        if valid:
+            valid_chains.append({"Chain": cnt, "Length": leng})
+            response = f'{response} Valid: {cnt}'
+        else:
+            not_valid.append({"Chain": cnt, "Length": leng})
+            response = f'{response} Invalid:{cnt}'
+            
+        per = ((len(valid_chains)+1)/(len(j)+1))*100
+        cnt+=1
+    response=f'{int(per)}%-{response}'
+    print (f'Valid: {valid_chains}')
+    print (f'Not Valid: {not_valid}')
+    #p = json.loads(str(valid_chains))
+    p = valid_chains
+    cnt2=0
+    bot=0
+    out=[]
+    while cnt2 < len(p):
+        if p[cnt2]['Length'] > bot:
+            bot = p[cnt2]['Length']
+            out = [cnt2,bot]
+        else:
+            pass
+        cnt2+=1
+    print (out)
+    return response
+    
+def valid():
+    valid,ind,mes = blockchain.chain_valid(blockchain.chain)
+    if valid:
+        response = 'The Blockchain is valid.'
+    else:
+        response = f'Blockchain is not valid. {mes} at Index {ind}'
+    return response
+    
+def get_chain(repo_name=None,chain_name=None,token=None):
+    if repo_name == "":
+        repo_name = repo_d
+    if chain_name=="":
+        chain_name = chain_d
+    #src = f"https://huggingface.co/spaces/{repo_name}/raw/main/{chain_name}"
+    
+    #ff = open('chain1.json','r')
+    #src = ff.read()
+    #src = json.loads(ff)    
+    
+    
+    try:
+        r = requests.get(f'{main_chain}{chain_name}') 
+        #create_chain(load=r.text)
+        global blockchain
+        blockchain = Blockchain(load=r.text)
+        #return "New Chain Created",display_chain()
+
+
+        
+        #create_chain(src)
+        response = {'chain': blockchain.chain,
+                    'length': len(blockchain.chain)}
+        message = f"Blockchain loaded from: {main_chain}{chain_name}"
+        return response,message
+    except:
+        message = f"Error loading from: {src}"
+        return ["Error Loading Chain"],message
+def checkp(inp):
+    if inp == pa:
+        return gr.update(visible=False), gr.update(visible=True)
+    elif inp != pa:
+        return gr.update(visible=True), gr.update(visible=False)
+def add_node(this_space,repo,space,chain_file):
+    #print(f"{api.whoami(['name'])}")
+    #repo = f'omnibus/{space}'
+    is_valid='True'
+    r = requests.get(f'{main_nodes}') 
+    try:
+        lod = json.loads(r.text)
+    except:
+        lod=[]
+        pass
+    block = {'index': len(lod) + 1,
+            'timestamp': str(datetime.datetime.now()),
+            'url': f'https://huggingface.co/datasets/{repo}/{space}/raw/main/{chain_file}',
+            'valid': f'{is_valid}'}
+    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=main_nodes.split('main/',1)[1],
+        repo_id=main_nodes.split('datasets/',1)[1].split('/raw',1)[0],
+        token=token_self,
+        repo_type="dataset",
+        )
+        os.remove("tmp1.json")
+
+    except Exception as e:
+        pass         
+
+with gr.Blocks() as bc:
+    with gr.Row(visible=True) as invalid:
+        pass_box = gr.Textbox()
+        pass_btn = gr.Button()
+        
+    with gr.Box(visible=False) as valida:
+        gr.Markdown("""<h1><center>Blockchain Simulator<br><h3>(Transactions have no value)<br><h4>Chain will reset at 20 blocks""")
+        blockchain = gr.State()
+        with gr.Row():
+            with gr.Column():
+                with gr.Accordion(label="Load",open=False):
+                    with gr.Row():
+                        chain_repo=gr.Textbox(label="repo/name")
+                        chain_n=gr.Textbox(label="Chain file")
+                    with gr.Row():
+                        in_chain_btn=gr.Button("Load Chain")
+                        create_bc = gr.Button("Create New Blockchain")
+                send=gr.Textbox(label="Sender")
+                rec=gr.Textbox(label="Recipient")
+                am=gr.Textbox(label="Amount")
+                send_trans=gr.Button("Post Transaction")
+                mine_b = gr.Button("Mine Block")
+                check = gr.Button("Check Chain")
+                check_all = gr.Button("Check All")
+            with gr.Column():
+                block_text = gr.Textbox()
+                trans_data = gr.Dataframe()
+                json_out = gr.JSON()
+                chain_json = gr.JSON()
+        with gr.Accordion("Nodes", open=False):
+            with gr.Row():
+                this_space=gr.Textbox(label="This Repo/Space")
+            with gr.Row():
+                node_repo=gr.Textbox(label="Node Repo")
+                node_space=gr.Textbox(label="Node Space")
+                node_file=gr.Textbox(label="Node File")
+            node_add=gr.Button("Add Node")
+
+    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])
+    create_bc.click(create_chain,[chain_n],[block_text,json_out,chain_json])
+    check.click(valid,None,block_text)
+    check_all.click(sort_valid,None,block_text)
+
+    send_trans.click(bc_transactions,[send,rec,am],[trans_data,block_text,send,rec,am])
+    mine_b.click(mine_block,[chain_repo,chain_n],[json_out,chain_json,trans_data,block_text])
+
+
+bc.launch(enable_queue=False)