Update app.py

app.py

@@ -14,9 +14,9 @@ import crypt
 
 from huggingface_hub import (create_repo,get_full_repo_name,upload_file,CommitOperationAdd,HfApi)
 
-main_chain='https://huggingface.co/datasets/Omnibus/blockchain-sim-2/raw/main/chains/'
-main_balance='https://huggingface.co/datasets/Omnibus/blockchain-sim-2/raw/main/balance/'
-main_trans='https://huggingface.co/datasets/Omnibus/blockchain-sim-2/raw/main/transact/'
+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'
@@ -442,6 +442,19 @@ with gr.Blocks() as bc:
         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.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)
+
             with gr.Tab("OG"):
                 with gr.Row():
                     with gr.Column():
@@ -552,6 +565,123 @@ with gr.Blocks() as bc:
             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:
+                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)
+    
+    
+    
+        #os.environ['PRIV'] = ""
+
+
+    send_mes_btn.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)