Update app.py

app.py

@@ -35,72 +35,64 @@ def show_s(name,token):
     return(gr.update(label="Spaces", choices=[s for s in spaces]))
            #gr.update(choices=[s for s in spaces],interactive=True))
 
-def show_f(repo,name,token):
-    api = HfApi(token=token)
-    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:
-        if "/" in d_app:
-            dir_1=d_app.split("/",1)[0]
-            rem_1=d_app.split("/",1)[1]
-            if not os.path.exists(f'{name}/{dir_1}'):
-                   os.makedirs(f'{name}/{dir_1}')
-            
-            if "/" in rem_1:
-                dir_2=rem_1.split("/",1)[0]
-                rem_2=rem_1.split("/",1)[1]
-                if not os.path.exists(f'{name}/{dir_1}/{dir_2}'):
-                       os.makedirs(f'{name}/{dir_1}/{dir_2}')                
-                #sf=rem_2.split(".",1)[1]
-                #pf=rem_2.split(".",1)[0]
-                #f_name=f'{dir_1}/{dir2}/{pf}.{sf}'
-                f_name=f'{dir_1}/{dir2}/{rem_2}'
-            
+def show_f(repo,name,token,tog):
+    if tog:
+        api = HfApi(token=token)
+        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:
+            if "/" in d_app:
+                dir_1=d_app.split("/",1)[0]
+                rem_1=d_app.split("/",1)[1]
+                if not os.path.exists(f'{name}/{dir_1}'):
+                       os.makedirs(f'{name}/{dir_1}')
+                if "/" in rem_1:
+                    dir_2=rem_1.split("/",1)[0]
+                    rem_2=rem_1.split("/",1)[1]
+                    if not os.path.exists(f'{name}/{dir_1}/{dir_2}'):
+                           os.makedirs(f'{name}/{dir_1}/{dir_2}')                
+                    f_name=f'{dir_1}/{dir2}/{rem_2}'
+                
+                else:
+                    f_name=f'{dir_1}/{rem_1}'
+                print(f_name)
             else:
-                #sf=rem_1.split(".",1)[1]
-                #pf=rem_1.split(".",1)[0]
-                #f_name=f'{dir_1}/{pf}.{sf}'
-                f_name=f'{dir_1}/{rem_1}'
-            print(f_name)
-
-        else:
+                f_name=f'{d_app}'            
+                pass
     
-            #sf=d_app.split(".",1)[1]
-            #pf=d_app.split(".",1)[0]
-            #f_name=f'{pf}.{sf}'            
-            f_name=f'{d_app}'            
-            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}','wb')
-        #file = open(f'{name}/{f_name}','wb')
-        file.write(r.content)
-        file.close()
-        file_list.append(f'{name}/{f_name}')
-
-    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')
-
-    s_info=api.space_info(f'{repo}/{name}',files_metadata=True)
-    '''
-    with open(f'{uid}-tmp.json','w') as f:
-        json.dump(s_info,f,indent=4)
-    f.close()
-    '''
-    return(file_list,
-          gr.update(choices=[f for f in f_ist],interactive=True),
-          api.get_space_runtime(f'{repo}/{name}'))
-
+            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}','wb')
+            #file = open(f'{name}/{f_name}','wb')
+            file.write(r.content)
+            file.close()
+            file_list.append(f'{name}/{f_name}')
+    
+        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')
+    
+        s_info=api.space_info(f'{repo}/{name}',files_metadata=True)
+        '''
+        with open(f'{uid}-tmp.json','w') as f:
+            json.dump(s_info,f,indent=4)
+        f.close()
+        '''
+        return(file_list,
+              gr.update(choices=[f for f in f_ist],interactive=True),
+              api.get_space_runtime(f'{repo}/{name}'))
+    else:
+        return(None,
+              None,
+              api.get_space_runtime(f'{repo}/{name}'))        
 def show_f_cont(repo,name,file,token):
     html_text = '<html>\n<body>\n<div>\n'
     images=[".png" , ".jpg" , ".gif" , ".webm" , ".mp4"]
@@ -142,6 +134,7 @@ with gr.Blocks(css=css) as build:
                     token = gr.Textbox(label="auth")        
                     s_btn = gr.Button("Show Spaces")
                 with gr.Column(scale=1):
+                    tog=gr.Checkbox(label="Show Files",value=True)
                     files=gr.File(file_count="directory")
         with gr.Tab("View 1"):
             with gr.Row():
@@ -201,6 +194,6 @@ with gr.Blocks(css=css) as build:
 
     move_btn.click(move,[r_name,space_radio,dest,token]).then(show_s,[r_name,token],[space_radio])
     s_btn.click(show_s,[r_name,token],[space_radio])    
-    space_radio.change(show_f,[r_name,space_radio,token],[files,file_radio,space_info_json])    
+    space_radio.change(show_f,[r_name,space_radio,token,tog],[files,file_radio,space_info_json])    
     file_radio.change(show_f_cont,[r_name,space_radio,file_radio,token],[file_contents])
 build.queue(default_concurrency_limit=10).launch(show_api=False)