Update dl.py

dl.py

@@ -223,6 +223,33 @@ def show_all_z(author,token):
     file_list.append(f'{author}.zip')
     yield file_list
 
+def dl_checked_fn(author,checklist,token):
+    spaces=[]
+    api = HfApi(token=token)
+    #author=name
+    s_ist = checklist
+    file_list = []
+    file_list_ea=[]
+    for i,space in enumerate(s_ist):
+        space_ea = space.id.split("/",1)[1]
+        spaces.append(space_ea)
+        #print (space_ea)
+        f_ist = (api.list_repo_files(repo_id=f'{author}/{space_ea}', repo_type="space"))
+        #print (f_ist)
+        if not os.path.exists(space_ea):
+               os.makedirs(space_ea)
+        file= snapshot_download(repo_id=f'{author}/{space_ea}', repo_type="space")
+        shutil.make_archive(f"{space_ea}", 'zip', file)
+        file_list.append(f'{space_ea}.zip')
+        yield file_list
+    with ZipFile(f"{author}.zip", "w") as zipObj:
+        for idx, file in enumerate(file_list):
+            try:
+                zipObj.write(f'{file}')   
+            except Exception:
+                pass
+    file_list.append(f'{author}.zip')
+    yield file_list
     
         
 def checkp(p):
@@ -274,12 +301,14 @@ with gr.Blocks(css=css) as build:
             with gr.Row():
                 with gr.Column(scale=1):
                     with gr.Group():
-                        space_radio=gr.Radio(label="Spaces",choices=[])
+                        space_check=gr.Checkbox(label="Spaces",choices=[])
                 with gr.Column(scale=1):
-                    file_radio=gr.Radio(label="Files",choices=[])
+                    dl_checked_btn=gr.Button("Download Checked")
                 with gr.Column(scale=2):
-                    file_contents=gr.HTML()
+                    file_contents_2=gr.HTML()
                     
+    dl_checked_btn.click(dl_checked_fn,[r_name,space_check,token],files)
+    
     show_all_btn1.click(show_all,[r_name,token],files)
     show_all_btn2.click(show_all_z,[r_name,token],files)