Update app.py

app.py

@@ -1,9 +1,11 @@
 import gradio as gr
+import chatchain
 import crypt
 import os
 rec_list = []
 pa=os.environ['PASS']
 
+main_chain='https://huggingface.co/datasets/Omnibus/chat-at/raw/main/chat/'
 
 
 def checkp(inp):
@@ -31,7 +33,10 @@ with gr.Blocks() as app:
                             rec=gr.Dropdown(label="Recipient", choices=[f for f in rec_list], allow_custom_value=True)
                             send_mes=gr.Textbox(label="Message", lines=6)
                             send_mes_btn=gr.Button()
-                        rec_mes = gr.Textbox(lines=6)
+                        with gr.Column():
+                            block_text = gr.Textbox(label = "System", interactive = False)
+                            response_json=gr.JSON()
+                            rec_mes = gr.Textbox(lines=6)
             
             
                         
@@ -61,12 +66,39 @@ with gr.Blocks() as app:
         gr.Column()
     def test_fn(im):
         return (im)
+
+
+    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()
+
+
+        
+    def create_new_chain(address):
+        address = str(address.strip("b").strip("'"))
+        mychain_rec = chatchain.MyChainRec(chain_load=main_chain,create=address)
+        response = {'chain': mychain_rec.chain,
+                'length': len(mychain_rec.chain)}
+        
+        message = f"Blockchain loaded from: {main_chain}{address}.json"
+        #send_list,send_drop = update_send_list()
+        rec_list, rec_drop = update_rec_list()
     
+        return response,message,rec_drop       
     input_key.change(crypt.address,input_key,sender)
     
     pass_btn.click(checkp,pass_box,[invalid,valida])
 
-    gen_wal_btn.click(crypt.generate_keys,None,[out2,out1,img3,out3,img1,img2]).then(test_fn,[img1],[input_key])
+    gen_wal_btn.click(crypt.generate_keys,None,[out2,out1,img3,out3,img1,img2]).then(create_new_chain,out1,[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)