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_trending_spaces(limit: int = 300) -> Union[List[Dict], str]:
+def get_popular_spaces(limit: int = 300) -> Union[List[Dict], str]:
     url = "https://huggingface.co/api/spaces"
     params = {
-        "sort": "trending",
+        "sort": "likes",  # Changed from "trending" to "likes"
         "direction": -1,
         "limit": limit,
         "full": "true"
@@ -103,7 +103,6 @@ def on_select(space):
         print(f"Selected space: {json.dumps(space, indent=2)}")
         summary = summarize_space(space)
         app_content = get_app_py_content(space['id'])
-        tree_structure = get_space_structure(space['id'])
         
         info = f"선택된 Space: {space.get('name', 'Unknown')} (ID: {space.get('id', 'Unknown')})\n"
         info += f"Author: {space.get('author', 'Unknown')}\n"
@@ -111,11 +110,8 @@ def on_select(space):
         info += f"URL: {space.get('url', 'Unknown')}\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.get('url', 'Unknown')}")
-        return info, app_content, space.get('url', ''), tree_view, list_view
+        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())
@@ -141,65 +137,12 @@ 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 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):
@@ -246,9 +189,9 @@ def generate_usage_guide(app_content):
 
 def create_ui():
     try:
-        spaces_list = get_trending_spaces()
+        spaces_list = get_popular_spaces()  # Changed from get_trending_spaces
         formatted_spaces = format_spaces(spaces_list)
-        print(f"Total trending spaces loaded: {len(formatted_spaces)}")
+        print(f"Total popular spaces loaded: {len(formatted_spaces)}")  # Changed from "trending" to "popular"
 
         css = """
         footer {visibility: hidden;}
@@ -285,7 +228,7 @@ def create_ui():
         """
 
         with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
-            gr.Markdown("# 300: HuggingFace Trending Spaces")
+            gr.Markdown("# 300: HuggingFace Popular Spaces")  # Changed from "Trending" to "Popular"
             
             with gr.Row():
                 with gr.Column(scale=1):
@@ -305,10 +248,6 @@ def create_ui():
                                 info_output = gr.Textbox(label="Space 정보 및 요약", elem_id="info-output", lines=20, max_lines=30)
                             url_state = gr.State("")
                             last_url_state = gr.State("")
-
-
-
-
                             
                             screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=600)
                             refresh_button = gr.Button("🔄 서비스 화면", elem_id="refresh-button")
@@ -348,6 +287,7 @@ def create_ui():
 
                 def open_space_in_browser(url):
                     if url:
+
                         import webbrowser
                         webbrowser.open(url)
                         return f"'{url}' 주소가 새 탭에서 열렸습니다."
@@ -402,4 +342,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())