Update dl.py

dl.py

@@ -58,7 +58,40 @@ def show_f(repo,name,token):
 
     return(gr.Dropdown.update(label="Files", choices=[f for f in f_ist],interactive=True), file_list)
 
+def show_f2(repo,name,token):
+    api = HfApi(token=token)
+    f_ist = snapshot_download(repo_id=f'{repo}/{name}', repo_type="space")
+
+    #f_ist = (api.list_repo_files(repo_id=f'{repo}/{name}', repo_type="space"))
+    print (f_ist)
+    file_list = []
+    file_out = []
+    if not os.path.exists(name):
+           os.makedirs(name)
+      
+    for d_app in f_ist:
+        
+        r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}')
+        #print (r.text)
+        uid = uuid.uuid4()
+
+        sf=d_app.split(".",1)[1]
+        pf=d_app.split(".",1)[0]
+        f_name=f'{pf}.{sf}'
+        file = open(f'{name}/{f_name}','w')
+        file.writelines(r.text)
+        file.close()
+        file_list.append(Path(f'{name}/{f_name}'))
+        file_out.append(d_app)
 
+    with ZipFile(f"{name}.zip", "w") as zipObj:
+        for idx, file in enumerate(f_ist):
+            zipObj.write(f'{name}/{file}')         
+    file_list.append(f'{name}.zip')
+
+    return(gr.Dropdown.update(label="Files", choices=[f for f in f_ist],interactive=True), file_list)
+
+    
 def show_f_cont(repo,name,file,token):
     r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{file}')
     return(r.text)
@@ -183,7 +216,7 @@ with gr.Blocks() as build:
     s_btn.click(show_s,[r_name,token],s_name)    
     f_name.change(show_f_cont,[r_name,s_name,f_name],r_t)
     #s_name.change(show_f,[r_name,s_name,token],[d_app,files])    
-    l_btn.click(show_f,[r_name,s_name,token], [f_name, files])
+    l_btn.click(show_f2,[r_name,s_name,token], [f_name, files])
     
     pass_btn.click(checkp,pass_box,[no,yes])
     #inbut.click(build_space,[token,t_name,s_type,m_type,r_type,d_app,d_css],output_html)