broadfield-dev commited on
Commit
a2b08e0
·
verified ·
1 Parent(s): d803c51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -4
app.py CHANGED
@@ -57,7 +57,7 @@ def get_all_files(owner, repo, path="", is_hf=False):
57
  except Exception as e:
58
  return None
59
 
60
- def get_hf_files(repo,name):
61
  api = HfApi()
62
  f_ist = (api.list_repo_files(repo_id=f'{repo}/{name}', repo_type="space"))
63
  print (f_ist)
@@ -69,6 +69,8 @@ def get_hf_files(repo,name):
69
  for d_app in f_ist:
70
  if "/" in d_app:
71
  dir_1=d_app.split("/",1)[0]
 
 
72
  rem_1=d_app.split("/",1)[1]
73
  if not os.path.exists(f'{name}/{dir_1}'):
74
  os.makedirs(f'{name}/{dir_1}')
@@ -87,14 +89,14 @@ def get_hf_files(repo,name):
87
  pf=rem_1.split(".",1)[0]
88
  f_name=f'{dir_1}/{pf}.{sf}'
89
  print(f_name)
90
-
91
  else:
92
 
93
  sf=d_app.split(".",1)[1]
94
  pf=d_app.split(".",1)[0]
95
  f_name=f'{pf}.{sf}'
96
  pass
97
-
98
  r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}')
99
  print(d_app)
100
  #print (r.text)
@@ -102,7 +104,13 @@ def get_hf_files(repo,name):
102
  file = open(f'{name}/{f_name}','wb')
103
  file.write(r.content)
104
  file.close()
105
- file_list.append({"path":f'{f_name}'})
 
 
 
 
 
 
106
  print(file_list)
107
  return file_list
108
 
 
57
  except Exception as e:
58
  return None
59
 
60
+ def get_hf_files(repo,name,path=""):
61
  api = HfApi()
62
  f_ist = (api.list_repo_files(repo_id=f'{repo}/{name}', repo_type="space"))
63
  print (f_ist)
 
69
  for d_app in f_ist:
70
  if "/" in d_app:
71
  dir_1=d_app.split("/",1)[0]
72
+ file_list.extend(get_hf_files(repo,name,dir_1))
73
+ '''
74
  rem_1=d_app.split("/",1)[1]
75
  if not os.path.exists(f'{name}/{dir_1}'):
76
  os.makedirs(f'{name}/{dir_1}')
 
89
  pf=rem_1.split(".",1)[0]
90
  f_name=f'{dir_1}/{pf}.{sf}'
91
  print(f_name)
92
+ '''
93
  else:
94
 
95
  sf=d_app.split(".",1)[1]
96
  pf=d_app.split(".",1)[0]
97
  f_name=f'{pf}.{sf}'
98
  pass
99
+
100
  r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}')
101
  print(d_app)
102
  #print (r.text)
 
104
  file = open(f'{name}/{f_name}','wb')
105
  file.write(r.content)
106
  file.close()
107
+ if item['type'] == 'file':
108
+ files.append(item)
109
+ elif item['type'] == 'dir':
110
+ files.extend(get_all_files(owner, repo, item['path'], is_hf))
111
+
112
+
113
+ file_list.append({"path":f'{d_app}'})
114
  print(file_list)
115
  return file_list
116