Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
+from huggingface_hub import InferenceClient, HfApi
 import os
 import requests
 from typing import List, Dict, Union
@@ -14,6 +14,7 @@ import json
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_TOKEN)
+hf_api = HfApi(token=HF_TOKEN)
 
 def get_headers():
     if not HF_TOKEN:
@@ -93,12 +94,36 @@ def get_app_py_content(space_id: str) -> str:
         return f"Error fetching app.py content for space: {space_id}"
 
 def get_space_structure(space_id: str) -> Dict:
-    api_url = f"https://huggingface.co/api/spaces/{space_id}/tree"
     try:
-        response = requests.get(api_url, headers=get_headers())
-        response.raise_for_status()
-        return response.json()
-    except requests.RequestException as e:
+        # space_id에서 owner와 repo_name을 분리
+        owner, repo_name = space_id.split('/')
+        
+        # HfApi를 사용하여 파일 목록을 가져옴
+        files = hf_api.list_repo_files(repo_id=space_id, repo_type="space")
+        
+        # 파일 목록을 트리 구조로 변환
+        tree = {"type": "directory", "path": "", "tree": []}
+        for file in files:
+            path_parts = file.split('/')
+            current = tree
+            for i, part in enumerate(path_parts):
+                if i == len(path_parts) - 1:  # 파일
+                    current["tree"].append({"type": "file", "path": part})
+                else:  # 디렉토리
+                    found = False
+                    for item in current["tree"]:
+                        if item["type"] == "directory" and item["path"] == part:
+                            current = item
+                            found = True
+                            break
+                    if not found:
+                        new_dir = {"type": "directory", "path": part, "tree": []}
+                        current["tree"].append(new_dir)
+                        current = new_dir
+        
+        return tree
+    except Exception as e:
+        print(f"Error in get_space_structure: {str(e)}")
         return {"error": f"API request error: {str(e)}"}
 
 def format_tree_structure(tree_data: Dict, indent: str = "") -> str:
@@ -333,6 +358,4 @@ if __name__ == "__main__":
         demo.launch()
     except Exception as e:
         print(f"Error in main: {str(e)}")
-        print(traceback.format_exc())
-
-                    
+        print(traceback.format_exc())