Update dl.py

dl.py

@@ -32,7 +32,7 @@ def show_f(repo,name,token):
     f_ist = (api.list_repo_files(repo_id=f'{repo}/{name}', repo_type="space"))
     print (f_ist)
     file_list = []
-    file_out = []
+    #file_out = []
     if not os.path.exists(name):
            os.makedirs(name)
       
@@ -64,17 +64,31 @@ def show_f(repo,name,token):
             pf=d_app.split(".",1)[0]
             f_name=f'{pf}.{sf}'            
             pass
-        r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}')
-        print(d_app)
-        #print (r.text)
-        uid = uuid.uuid4()
-
-
-        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)
+        
+        
+        
+        if ".png" or ".jpg" or ".gif" or ".webm" or ".mp4" in f_name:
+            #html_text += f'<object data="https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}"'
+            #out_text = "Image File"
+            sav_im = Image.open(f"https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}")
+            sav_im.save(f'{name}/{f_name}')
+            file_list.append(Path(f'{name}/{f_name}'))
+            #file_out.append(f"https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}")
+        
+        
+        
+        else:
+            r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}')
+            print(d_app)
+            #print (r.text)
+            uid = uuid.uuid4()
+    
+    
+            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):