Update app.py

app.py

@@ -52,6 +52,55 @@ def get_all_files(owner, repo, path="", is_hf=False):
     except Exception as e:
         return None
 
+def get_hf_files(repo,name):
+    api = HfApi()
+    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}'
+            
+            else:
+                sf=rem_1.split(".",1)[1]
+                pf=rem_1.split(".",1)[0]
+                f_name=f'{dir_1}/{pf}.{sf}'
+            print(f_name)
+
+        else:
+    
+            sf=d_app.split(".",1)[1]
+            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}','wb')
+        file.write(r.content)
+        file.close()
+        file_list.append(Path(f'{name}/{f_name}'))
+    print(file_list)
+    return file_list
+
 def get_repo_contents(url):
     """Parse URL and fetch repository contents."""
     try:
@@ -67,7 +116,7 @@ def get_repo_contents(url):
                 # Add more known paths or implement HTML scraping if needed
             ]
             #files = get_all_files(owner, repo, "", True) or known_files
-            files = get_all_files(owner, repo, "", True)
+            files = get_hf_files(owner, repo)
             return owner, repo, files, True
         else:  # Assume GitHub URL
             parts = url.rstrip('/').split('/')