Update app.py

app.py

@@ -177,18 +177,7 @@ def send_message(send,rec,message):
         print(f'response2:: {response}')
         print(f'message2:: {message_out}')
         lod_rec = response
-        '''
-        mes1,val,mes2,ind = chatchain.valid_rec()
-        print (f'val:: {val}')
-        if val == False:
-            p=False 
-            mes = f"Recipient: {rec} Blockchain is not valid. {mes2} at Index {ind}"
-        print(mes)
-        p=True
-        #if z==False:
-        #    mes = b_mes
-        #    p=False
-        '''
+     
         p=True
     except Exception as e:
         mes = f"An Error Occured: {e}" 
@@ -273,9 +262,9 @@ with gr.Blocks() as app:
         gr.Column()
     def test_fn(im):
         return (im)
-    def create_new_chain(address,im):
+    def create_new_chain(address,pub_key,im):
         address = str(address.strip("b").strip("'"))
-        mychain_rec = chatchain.MyChainRec(chain_load=f'{main_chain}{address}/',create=address)
+        mychain_rec = chatchain.MyChainRec(chain_load=f'{main_chain}{address}/',create=address,pub_key=pub_key)
         chatchain.store_image(img=im,chain_load=f'{main_chain}',address=address)
         response = {'chain': mychain_rec.chain,
                 'length': len(mychain_rec.chain)}
@@ -294,7 +283,7 @@ with gr.Blocks() as app:
     input_key.change(crypt.address,input_key,sender).then(rec_list_up,in_chain,rec)
     pass_btn.click(checkp,pass_box,[invalid,valida])
 
-    gen_wal_btn.click(crypt.generate_keys,None,[out2,out1,img3,out3,img1,img2]).then(create_new_chain,[out3,img1],[response_json,block_text,rec]).then(test_fn,[img1],[input_key])
+    gen_wal_btn.click(crypt.generate_keys,None,[out2,out1,img3,out3,img1,img2]).then(create_new_chain,[out3,out2,img1],[response_json,block_text,rec]).then(test_fn,[img1],[input_key])
     rsa_enc_btn.click(crypt.encrypt_text,[rsa_to_enc,pub_key_in,priv_key_in1,out3],[rsa_enc_mes,qr_enc_mes])
     rsa_dec_btn.click(crypt.decrypt_text,[mes_in,priv_key_in],rsa_dec_mes)