Update app.py

app.py

@@ -2,6 +2,11 @@ import gradio as gr
 import chatchain
 import crypt
 import os
+from huggingface_hub import (upload_file,HfApi)
+
+token_self = os.environ['HF_TOKEN']
+api = HfApi(token=token_self)
+
 rec_list = []
 pa=os.environ['PASS']
 
@@ -75,15 +80,28 @@ with gr.Blocks() as app:
                 '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 = chatchain.update_rec_list(main_chain)
+        
+        rec_list,rec_drop = update_rec_list()
         #rec_drop = gr.Dropdown.update(label="Recipient", choices=[f for f in rec_list])
         return response,message,rec_drop      
 
-    def update_rec():
-        _,rec_up = chatchain.update_rec_list(main_chain)
-        return rec_up
+    def update_rec_list():
+        repo = main_chain.split('datasets/',1)[1].split('/raw',1)[0].split('/',1)[0]
+        name = main_chain.split('datasets/',1)[1].split('/raw',1)[0].split('/',1)[1]
+
+        f_ist = (api.list_repo_files(repo_id=f'{repo}/{name}', repo_type="dataset"))
+        rec_list =[]
+        for i,ea in enumerate(f_ist):
+            if "chat/" in ea:
+                try:
+                    if not "__Source__" in ea:
+                        rec_list.append(ea.split("/",1)[1].split("/",1)[0])
+                except Exception:
+                    pass
+        return gr.Dropdown.update(label="Recipient", choices=[f for f in rec_list])
+    rec_list, rec_drop = update_rec_list()
         
-    input_key.change(crypt.address,input_key,sender).then(update_rec,None,rec)
+    input_key.change(crypt.address,input_key,sender).then(update_rec_list,None,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])