import gradio as gr import pandas as pd import datetime import hashlib import json import os import requests from mychain import Blockchain from mychain import MyChainSend from mychain import MyChainRec from mychain import MyChainTrans import bc_utils import crypt from huggingface_hub import (create_repo,get_full_repo_name,upload_file,CommitOperationAdd,HfApi) main_chain='https://huggingface.co/datasets/Omnibus/chat-at/raw/main/chains/' main_balance='https://huggingface.co/datasets/Omnibus/chat-at/raw/main/balance/' main_trans='https://huggingface.co/datasets/Omnibus/chat-at/raw/main/transact/' #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'] trans_name = 'trans1.json' #repo_d='Omnibus/static-bin' chain_d='chain1.json' #node_file='node_file.json' #space='blockchain-simulator-dev1' api = HfApi(token=token_self) repo = main_balance.split('datasets/',1)[1].split('/raw',1)[0].split('/',1)[0] name = main_balance.split('datasets/',1)[1].split('/raw',1)[0].split('/',1)[1] ############################## BLOCKCHAIN END ############################### 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) ############################## MYCHAIN START ############################### def get_my_chain_send(sender_name=None): global mychain_send global mychain_trans 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}') try: d = requests.get(f'{main_trans}{trans_name}') mychain_trans = MyChainTrans(chain_load=main_trans,load=d.text) except Exception as e: print (f'error1:: {e}') try: mychain_trans = MyChainTrans(chain_load=main_trans,create=trans_name) print ("this worked") except Exception as ee: print (f'error2:: {ee}') pass try: response_trans = {'chain': mychain_trans.chain, 'length': len(mychain_trans.chain)} print(f'response_trans:: {response_trans}') except Exception as e: print (e) pass 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 display_chain_send(): response = {'chain': mychain_send.chain, 'length': len(mychain_send.chain)} return response def display_chain_rec(): response = {'chain': mychain_rec.chain, 'length': len(mychain_rec.chain)} return response def valid_send(): valid,ind,mes = mychain_send.chain_valid(mychain_send.chain) if valid: response = 'The Blockchain is valid.' z=True else: response = f'Sender Blockchain is not valid. {mes} at Index {ind}' z=False return response,z,mes,ind def valid_rec(): valid,ind,mes = mychain_rec.chain_valid(mychain_rec.chain) if valid: response = 'The Blockchain is valid.' z=True else: response = f'Blockchain is not valid. {mes} at Index {ind}' z=False return response,z, mes, ind 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_trans(balance, chain_r=None,chain_n=None): previous_block = mychain_trans.print_previous_block() previous_proof = previous_block['proof'] proof = mychain_trans.proof_of_work(previous_proof) previous_hash = mychain_trans.hash(previous_block) block = mychain_trans.create_block(balance, proof, previous_hash,len(bc_utils.blockchain.chain)+1,chain_r,chain_n) response = {'message': 'A block is MINED', 'index': block['index'], 'timestamp': block['timestamp'], 'balance': block['balance'], 'proof': block['proof'], 'previous_hash': block['previous_hash'] } message = "A block is MINED" show_chain = display_chain_send() if len(mychain_trans.chain) > 1000: mychain_trans.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 merge_trans(): trans_bx = [] previous_post_send = mychain_send.print_previous_block() previous_post_rec = mychain_rec.print_previous_block() send_block = { 'role': 'Sender', 'name': previous_post_send['transactions'][0]['sender'], 'index': previous_post_send['index'], 'timestamp': previous_post_send['timestamp'], 'recipient': previous_post_send['transactions'][0]['recipient'], 'amount': previous_post_send['transactions'][0]['amount'], 'balance': previous_post_send['balance'], 'proof': previous_post_send['proof'], 'previous_hash': previous_post_send['previous_hash'], 'block': len(bc_utils.blockchain.chain)+1} rec_block = { 'role': 'Recipient', 'name': previous_post_rec['transactions'][0]['recipient'], 'index': previous_post_rec['index'], 'timestamp': previous_post_rec['timestamp'], 'sender': previous_post_rec['transactions'][0]['sender'], 'amount': previous_post_rec['transactions'][0]['amount'], 'balance': previous_post_rec['balance'], 'proof': previous_post_rec['proof'], 'previous_hash': previous_post_rec['previous_hash'], 'block': len(bc_utils.blockchain.chain)+1} trans_bx.append(send_block) trans_bx.append(rec_block) return trans_bx def issue_tokens(send,rec,amount): response_send={} response_rec={} show_chain_send={} show_chain_rec={} data_send=None data_rec=None rec_send=None rec_drop=None message_send=None message_rec=None trans_data=None trans_mes=None balance_send = 0 balance_rec = 0 mes = "blank message" try: b_mes,z=bc_utils.valid() if z == True: try: response,message = get_my_chain_send(send) #print (f'response:{response}::message{message}') #r = requests.get(f'{main_balance}{send}.json') lod=response #lod = json.loads(response) #print (lod) p=True except Exception as e: lod=[] p=False mes = f"Sender has no wallet {e}" pass mes1,val,mes2,ind = valid_send() print (f'val:: {val}') if val == False: p=False mes = f"Sender: {send} Blockchain is not valid. {mes2} at Index {ind}" #mes = "Blockchain is not valid." #if val == True: if p==True: try: balance = lod["chain"][-1]["balance"] except Exception: balance = 0 p=False #print (balance) balance_send =int(balance)-int(amount) if balance_send >=0: p=True response,message = get_my_chain_rec(rec) lod_rec = response mes1,val,mes2,ind = valid_rec() print (f'val:: {val}') if val == False: p=False mes = f"Recipient: {rec} Blockchain is not valid. {mes2} at Index {ind}" #mes = "Blockchain is not valid." #print (lod_rec) if val == True: try: balance = lod_rec["chain"][-1]["balance"] #print (balance) except Exception: p=False balance = 0 balance_rec =int(balance)+int(amount) if balance_send < 0: mes ="Not enough tokens" p = False print(mes) if z==False: mes = b_mes p=False except Exception as e: mes = f"Blockchain not loaded? {e}" p=False print (mes) if p==False: return (mes, p,None,None,None,None,None,None,None,None,None,None,None) #return (mes, p,{},{},{},{},{},{},None,None,None,None,{}) #json, json, json, json, dataframe, dataframe, textbox, textbox, dropdown, dropdown, dataframe if p==True: mychain_send.new_transaction(f"{send}",f"{rec}",f"{amount}",f"{balance_send}") mychain_trans.new_transaction(f"{send}",f"{rec}",f"{amount}",f"{balance_send}") message_send = "Transaction Added to Pool" data_send = pd.DataFrame(mychain_send.pending_transactions) mychain_rec.new_transaction(f"{send}",f"{rec}",f"{amount}",f"{balance_rec}") message_rec = "Transaction Added to Pool" data_rec = pd.DataFrame(mychain_rec.pending_transactions) response_send, show_chain_send, message_send = mychain_mine_block_send(balance_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 = mychain_mine_block_rec(balance_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() trans_bx = merge_trans() trans_data, mes = bc_utils.bc_transactions(trans_bx) return (mes, p, response_send, response_rec, show_chain_send, show_chain_rec, data_send, data_rec, message_send, message_rec, rec_send, rec_drop,trans_data) #global send_list #global rec_list def create_new_chain(address): address = str(address.strip("b").strip("'")) mychain_rec = MyChainRec(chain_load=main_balance,create=address) response = {'chain': mychain_rec.chain, 'length': len(mychain_rec.chain)} message = f"Blockchain loaded from: {main_balance}{address}.json" send_list,send_drop = update_send_list() rec_list, rec_drop = update_rec_list() return response,message,send_drop,rec_drop ############################## MYCHAIN END ############################### def res_source(): block = {'index': 1, 'timestamp': str(datetime.datetime.now()), 'transactions': [], 'balance': 10000000000000000000, 'proof': 1, 'previous_hash': 0} pending_transactions_x = [] pending_transactions_x.append(block) json_object = json.dumps(pending_transactions_x, indent=4) with open("tmp_send.json", "w") as outfile: outfile.write(json_object) try: api.upload_file( path_or_fileobj="tmp_send.json", path_in_repo="balance/__Source__.json", repo_id=main_balance.split('datasets/',1)[1].split('/raw',1)[0], token=token_self, repo_type="dataset", ) os.remove("tmp_send.json") return "__Source__ Reset" except Exception as e: return e def update_send_list(): f_ist = (api.list_repo_files(repo_id=f'{repo}/{name}', repo_type="dataset")) send_list =[] for i,ea in enumerate(f_ist): if "balance/" in ea: try: send_list.append(ea.split("/",1)[1].split(".",1)[0]) except Exception: pass return send_list, gr.Dropdown.update(label="Sender", choices=[f for f in send_list]) send_list,send_drop = update_send_list() def update_rec_list(): f_ist = (api.list_repo_files(repo_id=f'{repo}/{name}', repo_type="dataset")) rec_list =[] for i,ea in enumerate(f_ist): if "balance/" in ea: try: if not "__Source__" in ea: rec_list.append(ea.split("/",1)[1].split(".",1)[0]) except Exception: pass return rec_list, gr.Dropdown.update(label="Recipient", choices=[f for f in rec_list]) rec_list, rec_drop = update_rec_list() 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("""

Blockchain Simulator

(Transactions have no value)

Chain will reset at 20 blocks""") #blockchain = gr.State() with gr.Row(): with gr.Tab("OG"): 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") send = gr.Dropdown(label="Sender", choices=[f for f in send_list], value = "Bank") rec=gr.Dropdown(label="Recipient", choices=[f for f in rec_list], allow_custom_value=True) send_mes=gr.Textbox(label="Message", lines=6) #am=gr.Textbox(label="Amount") send_trans=gr.Button("Post Transaction") mine_b = gr.Button("Mine Block") check = gr.Button("Check Chain") check_trans = gr.Button("Check Transactions") 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") with gr.Accordion("Tokens", open=False): with gr.Row(): with gr.Column(): issue_btn=gr.Button() with gr.Column(): with gr.Row(): reset_blockchain=gr.Button("Reset Main Blockchain") reset_source=gr.Button("Reset __Source__") with gr.Row(): reset_sender=gr.Button("Reset Sender") reset_recieve=gr.Button("Reset Recipient") out_box_bool=gr.Textbox() with gr.Row(): with gr.Column(): block_text_send = gr.Textbox() trans_data_send = gr.Dataframe() json_out_send = gr.JSON() chain_json_send = gr.JSON() with gr.Column(): block_text_rec = gr.Textbox() trans_data_rec = gr.Dataframe() json_out_rec = gr.JSON() chain_json_rec = gr.JSON() with gr.Tab("BC"): with gr.Row(): with gr.Tab("Gen Wal"): gen_wal_btn=gr.Button() seed = gr.Textbox(label='Seed Phrase') img1=gr.Pil(label='Private Key') out1 = gr.Textbox(label='Private Key',max_lines=4) img2=gr.Pil(label='Public Key') out2 = gr.Textbox(label='Public Key',max_lines=4) img3=gr.Pil(label='Address') out3 = gr.Textbox(label='Address') with gr.Tab("Encrypt"): rsa_to_enc = gr.Textbox(label="txt to encrypt") pub_key_in = gr.Image(label="Public Key", type="filepath") priv_key_in1 = gr.Image(label="Private Key(sig)", type="filepath") rsa_enc_btn = gr.Button("RSA Encrypt") rsa_enc_mes = gr.Textbox(label="encoded", max_lines=4) qr_enc_mes = gr.Image(type="filepath") with gr.Tab("Decrypt"): mes_in = gr.Image(label="Message", type="filepath") priv_key_in = gr.Image(label="Private Key", type="filepath") rsa_dec_btn = gr.Button("RSA Decrypt") rsa_dec_mes = gr.Textbox(label="decoded") with gr.Tab("Messages"): with gr.Accordion("Key"): with gr.Row(): input_address = gr.Image(label="Wallet",type="filepath") input_key = gr.Image(label="Key",type="filepath") with gr.Row(): with gr.Column(): #send_mes=gr.Textbox(label="Message", lines=6) send_mes_btn=gr.Button() rec_mes = gr.Textbox(lines=6) def res_bc(): bc_utils.blockchain.reset(create="chain1.json") return ("Main Blockchain Reset") def res_send(send): mychain_send = MyChainSend(chain_load=main_balance,create=send) return ("Sender Blockchain Reset") def deep_trans(): chain1=mychain_send.chain chain2=bc_utils.blockchain.chain valid,ind,mes = mychain_send.deep_valid_send(chain1,chain2) if valid: response = 'The Blockchain is valid.' z=True else: response = f'Blockchain is not valid. {mes} at Index {ind}' z=False return response,z def mine_block(chain_r=None,chain_n=None): try: a,b,c,d = bc_utils.mine_block(chain_r=None,chain_n=None) mychain_trans.reset(create=trans_name) return a,b,c,d except Exception as e: print (e) return e,None,None,e def send_message(send,rec,amount): response_send={} response_rec={} show_chain_send={} show_chain_rec={} data_send=None data_rec=None rec_send=None rec_drop=None message_send=None message_rec=None trans_data=None trans_mes=None balance_send = 0 balance_rec = 0 mes = "blank message" try: #b_mes,z=bc_utils.valid() #if z == True: try: response,message = get_my_chain_send(send) #print (f'response:{response}::message{message}') #r = requests.get(f'{main_balance}{send}.json') lod=response #lod = json.loads(response) #print (lod) p=True except Exception as e: lod=[] p=False mes = f"Sender has no wallet {e}" pass mes1,val,mes2,ind = valid_send() print (f'val:: {val}') if val == False: p=False mes = f"Sender: {send} Blockchain is not valid. {mes2} at Index {ind}" #mes = "Blockchain is not valid." #if val == True: if p==True: response,message = get_my_chain_rec(rec) lod_rec = response mes1,val,mes2,ind = valid_rec() print (f'val:: {val}') if val == False: p=False mes = f"Recipient: {rec} Blockchain is not valid. {mes2} at Index {ind}" #mes = "Blockchain is not valid." #print (lod_rec) print(mes) #if z==False: # mes = b_mes # p=False except Exception as e: mes = f"Blockchain not loaded? {e}" p=False print (mes) if p==False: return (mes, p,None,None,None,None,None,None,None,None,None,None,None) #return (mes, p,{},{},{},{},{},{},None,None,None,None,{}) #json, json, json, json, dataframe, dataframe, textbox, textbox, dropdown, dropdown, dataframe if p==True: mychain_send.new_transaction(f"{send}",f"{rec}",f"{amount}") #mychain_trans.new_transaction(f"{send}",f"{rec}",f"{amount}") message_send = "Transaction Added to Pool" data_send = pd.DataFrame(mychain_send.pending_transactions) mychain_rec.new_transaction(f"{send}",f"{rec}",f"{amount}") message_rec = "Transaction Added to Pool" data_rec = pd.DataFrame(mychain_rec.pending_transactions) 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 = 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() #trans_bx = merge_trans() #trans_data, mes = bc_utils.bc_transactions(trans_bx) return (mes, p, response_send, response_rec, show_chain_send, show_chain_rec, data_send, data_rec, message_send, message_rec, rec_send, rec_drop) #os.environ['PRIV'] = "" send_trans.click(send_message,[send,rec,send_mes],[block_text,out_box_bool,json_out_send,json_out_rec,chain_json_send,chain_json_rec,trans_data_send,trans_data_rec,block_text_send,block_text_rec,send,rec,trans_data]) #send_mes_btn.click(send_message,[send_mes,input_address,input_key],[rec_mes]) gen_wal_btn.click(crypt.generate_keys,None,[out2,out1, img3,out3,img1,img2]).then(create_new_chain,out3,[json_out,block_text,send,rec]).then(crypt.test_fn,[img1,img2],[priv_key_in,pub_key_in,priv_key_in1]) rsa_enc_btn.click(crypt.encrypt_text,[rsa_to_enc,pub_key_in,priv_key_in1,out3],[rsa_enc_mes,qr_enc_mes]).then(crypt.test_fn2,qr_enc_mes,mes_in) rsa_dec_btn.click(crypt.decrypt_text,[mes_in,priv_key_in],rsa_dec_mes) check_trans.click(deep_trans,None,[block_text,out_box_bool]) reset_sender.click(res_send,send,block_text) reset_blockchain.click(res_bc,None,block_text) reset_source.click(res_source,None,block_text) #send_trans.click(issue_tokens,[send,rec,am],[block_text,out_box_bool,json_out_send,json_out_rec,chain_json_send,chain_json_rec,trans_data_send,trans_data_rec,block_text_send,block_text_rec,send,rec,trans_data]) #issue_btn.click(issue_tokens,[send,rec,am],[block_text,out_box_bool,json_out_send,json_out_rec,chain_json_send,chain_json_rec,trans_data_send,trans_data_rec,block_text_send,block_text_rec,send,rec,trans_data]) #issue_btn.click(issue_tokens,[send,rec,am],[block_text,out_box_bool,json_out_send]) node_add.click(bc_utils.add_node,[this_space,node_repo,node_space,node_file],block_text) pass_btn.click(checkp,pass_box,[invalid,valida]) in_chain_btn.click(bc_utils.get_chain,[chain_repo,chain_n],[chain_json,block_text]) create_bc.click(bc_utils.create_chain,[chain_n],[block_text,json_out,chain_json]) check.click(bc_utils.valid,None,[block_text,out_box_bool]) #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)