Update app.py

app.py

@@ -21,10 +21,10 @@ def get_headers():
         raise ValueError("Hugging Face token not found in environment variables")
     return {"Authorization": f"Bearer {HF_TOKEN}"}
 
-def get_popular_spaces(limit: int = 300) -> Union[List[Dict], str]:
+def get_most_liked_spaces(limit: int = 300) -> Union[List[Dict], str]:
     url = "https://huggingface.co/api/spaces"
     params = {
-        "sort": "likes",  # Changed from "trending" to "likes"
+        "sort": "likes",
         "direction": -1,
         "limit": limit,
         "full": "true"
@@ -33,93 +33,40 @@ def get_popular_spaces(limit: int = 300) -> Union[List[Dict], str]:
     try:
         response = requests.get(url, params=params, headers=get_headers())
         response.raise_for_status()
-        data = response.json()
-        print(f"API Response (first 2 items): {json.dumps(data[:2], indent=2)}")
-        return data
+        return response.json()
     except requests.RequestException as e:
-        print(f"API request error: {str(e)}")
-        return get_fallback_spaces()
-
-def get_fallback_spaces() -> List[Dict]:
-    return [
-        {
-            "id": "facebook/llama-2-7b-chat",
-            "name": "LLaMA 2 7B Chat",
-            "author": "Meta AI",
-            "likes": 1000,
-            "url": "https://huggingface.co/spaces/facebook/llama-2-7b-chat"
-        },
-        {
-            "id": "openai/whisper-large-v3",
-            "name": "Whisper Large V3",
-            "author": "OpenAI",
-            "likes": 800,
-            "url": "https://huggingface.co/spaces/openai/whisper-large-v3"
-        },
-        # Add more fallback spaces as needed
-    ]
+        return f"API request error: {str(e)}"
+    except ValueError as e:
+        return f"JSON decoding error: {str(e)}"
 
 def format_space(space: Dict) -> Dict:
-    print(f"Formatting space: {json.dumps(space, indent=2)}")
-    
     space_id = space.get('id', 'Unknown')
-    space_name = space.get('title', space.get('name', space_id.split('/')[-1] if '/' in space_id else space_id))
+    space_name = space_id.split('/')[-1] if '/' in space_id else space_id
     
     space_author = space.get('author', 'Unknown')
     if isinstance(space_author, dict):
         space_author = space_author.get('user', space_author.get('name', 'Unknown'))
     
     space_likes = space.get('likes', 'N/A')
-    space_url = space.get('url', f"https://huggingface.co/spaces/{space_id}")
+    space_url = f"https://huggingface.co/spaces/{space_id}"
     
-    formatted_space = {
+    return {
         "id": space_id,
         "name": space_name,
         "author": space_author,
         "likes": space_likes,
         "url": space_url,
     }
-    print(f"Formatted space: {json.dumps(formatted_space, indent=2)}")
-    return formatted_space
 
 def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
     if isinstance(spaces, str):
-        print(f"Error in spaces data: {spaces}")
-        return get_fallback_spaces()
+        return [{"error": spaces}]
     
-    formatted = []
-    for space in spaces:
-        if isinstance(space, dict):
-            try:
-                formatted.append(format_space(space))
-            except Exception as e:
-                print(f"Error formatting space: {str(e)}")
-                print(f"Problematic space data: {json.dumps(space, indent=2)}")
-    print(f"Total formatted spaces: {len(formatted)}")
-    return formatted if formatted else get_fallback_spaces()
-
-def on_select(space):
-    try:
-        print(f"Selected space: {json.dumps(space, indent=2)}")
-        summary = summarize_space(space)
-        app_content = get_app_py_content(space['id'])
-        
-        info = f"선택된 Space: {space.get('name', 'Unknown')} (ID: {space.get('id', 'Unknown')})\n"
-        info += f"Author: {space.get('author', 'Unknown')}\n"
-        info += f"Likes: {space.get('likes', 'N/A')}\n"
-        info += f"URL: {space.get('url', 'Unknown')}\n\n"
-        info += f"요약:\n{summary}"
-        
-        print(f"Returning URL: {space.get('url', 'Unknown')}")
-        return info, app_content, space.get('url', ''), "", ""  # Returning empty strings for tree and list views
-    except Exception as e:
-        print(f"Error in on_select: {str(e)}")
-        print(traceback.format_exc())
-        return f"오류가 발생했습니다: {str(e)}", "", "", "", ""
+    return [format_space(space) for space in spaces if isinstance(space, dict)]
 
 def summarize_space(space: Dict) -> str:
     system_message = "당신은 Hugging Face Space의 내용을 요약하는 AI 조수입니다. 주어진 정보를 바탕으로 간결하고 명확한 요약을 제공해주세요."
-    user_message = f"다음 Hugging Face Space를 요약해주세요: {space.get('name', 'Unknown')} by {space.get('author', 'Unknown')}. 좋아요 수: {space.get('likes', 'N/A')}. URL: {space.get('url', 'Unknown')}"
+    user_message = f"다음 Hugging Face Space를 요약해주세요: {space['name']} by {space['author']}. 좋아요 수: {space['likes']}. URL: {space['url']}"
     
     messages = [
         {"role": "system", "content": system_message},
@@ -137,12 +84,89 @@ def get_app_py_content(space_id: str) -> str:
     try:
         response = requests.get(app_py_url, headers=get_headers())
         if response.status_code == 200:
-            return response.text
+            return response.text  # 전체 내용을 반환
         else:
             return f"app.py file not found or inaccessible for space: {space_id}"
     except requests.RequestException:
         return f"Error fetching app.py content for space: {space_id}"
 
+
+def get_space_structure(space_id: str) -> Dict:
+    try:
+        # 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:
+    formatted = ""
+    for item in tree_data.get("tree", []):
+        if item["type"] == "file":
+            formatted += f"{indent}├── {item['path']}\n"
+        elif item["type"] == "directory":
+            formatted += f"{indent}├── {item['path']}/\n"
+            formatted += format_tree_structure(item, indent + "│   ")
+    return formatted
+
+def format_list_structure(tree_data: Dict) -> List[str]:
+    formatted = []
+    for item in tree_data.get("tree", []):
+        if item["type"] == "file":
+            formatted.append(item["path"])
+        elif item["type"] == "directory":
+            formatted.append(f"{item['path']}/")
+            formatted.extend(format_list_structure(item))
+    return formatted
+
+def on_select(space):
+    try:
+        print(f"Selected space: {space['name']}")
+        summary = summarize_space(space)
+        app_content = get_app_py_content(space['id'])
+        tree_structure = get_space_structure(space['id'])
+        
+        info = f"선택된 Space: {space['name']} (ID: {space['id']})\n"
+        info += f"Author: {space['author']}\n"
+        info += f"Likes: {space['likes']}\n"
+        info += f"URL: {space['url']}\n\n"
+        info += f"요약:\n{summary}"
+        
+        tree_view = format_tree_structure(tree_structure)
+        list_view = "\n".join(format_list_structure(tree_structure))
+        
+        print(f"Returning URL: {space['url']}")
+        return info, app_content, space['url'], tree_view, list_view
+    except Exception as e:
+        print(f"Error in on_select: {str(e)}")
+        print(traceback.format_exc())
+        return f"오류가 발생했습니다: {str(e)}", "", "", "", ""
+
 def update_screenshot(url, last_url, force_update=False):
     print(f"Updating screenshot. Current URL: {url}, Last URL: {last_url}, Force update: {force_update}")
     if url and (url != last_url or force_update):
@@ -189,9 +213,9 @@ def generate_usage_guide(app_content):
 
 def create_ui():
     try:
-        spaces_list = get_popular_spaces()  # Changed from get_trending_spaces
+        spaces_list = get_most_liked_spaces()
         formatted_spaces = format_spaces(spaces_list)
-        print(f"Total popular spaces loaded: {len(formatted_spaces)}")  # Changed from "trending" to "popular"
+        print(f"Total spaces loaded: {len(formatted_spaces)}")
 
         css = """
         footer {visibility: hidden;}
@@ -228,7 +252,7 @@ def create_ui():
         """
 
         with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
-            gr.Markdown("# 300: HuggingFace Popular Spaces")  # Changed from "Trending" to "Popular"
+            gr.Markdown("# 300: HuggingFace Most Liked Spaces")
             
             with gr.Row():
                 with gr.Column(scale=1):
@@ -236,8 +260,7 @@ def create_ui():
                     for space in formatted_spaces:
                         with gr.Row(elem_classes="space-row") as space_row:
                             with gr.Column():
-                                space_info = f"{space.get('name', 'Unknown')} by {space.get('author', 'Unknown')}"
-                                gr.Markdown(space_info, elem_classes="space-info")
+                                gr.Markdown(f"{space['name']} by {space['author']} (Likes: {space['likes']})", elem_classes="space-info")
                                 button = gr.Button("클릭", elem_classes="minimal-button")
                         space_rows.append((space_row, button, space))
 
@@ -269,6 +292,7 @@ def create_ui():
                     
                     update_trigger = gr.Button("Update Screenshot", visible=False)
 
+
                 def on_select_with_link(space):
                     info, app_content, url, tree, list_structure = on_select(space)
                     info += f"\n\n선택한 Space URL: {url}"
@@ -287,7 +311,6 @@ def create_ui():
 
                 def open_space_in_browser(url):
                     if url:
-
                         import webbrowser
                         webbrowser.open(url)
                         return f"'{url}' 주소가 새 탭에서 열렸습니다."
@@ -342,4 +365,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())