Update app.py

app.py

@@ -36,14 +36,13 @@ def get_trending_spaces(limit: int = 300) -> Union[List[Dict], str]:
         response = requests.get(url, params=params, headers=get_headers())
         response.raise_for_status()
         data = response.json()
-        print(f"API Response: {json.dumps(data[:2], indent=2)}")  # 처음 두 항목만 출력
+        print(f"API Response (first 2 items): {json.dumps(data[:2], indent=2)}")
         return data
     except requests.RequestException as e:
         return f"API request error: {str(e)}"
     except ValueError as e:
         return f"JSON decoding error: {str(e)}"
 
-
         
 def get_most_liked_spaces(limit: int = 300) -> Union[List[Dict], str]:
     url = "https://huggingface.co/api/spaces"
@@ -64,11 +63,10 @@ def get_most_liked_spaces(limit: int = 300) -> Union[List[Dict], str]:
         return f"JSON decoding error: {str(e)}"
 
 def format_space(space: Dict) -> Dict:
-    # 디버깅을 위해 원본 데이터 출력
-    print(f"Original space data: {json.dumps(space, indent=2)}")
+    print(f"Formatting space: {json.dumps(space, indent=2)}")
     
     space_id = space.get('id', 'Unknown')
-    space_name = space.get('title', space_id.split('/')[-1] if '/' in space_id else space_id)
+    space_name = space.get('title', space.get('name', space_id.split('/')[-1] if '/' in space_id else space_id))
     
     space_author = space.get('author', 'Unknown')
     if isinstance(space_author, dict):
@@ -77,16 +75,19 @@ def format_space(space: Dict) -> Dict:
     space_likes = space.get('likes', 'N/A')
     space_url = f"https://huggingface.co/spaces/{space_id}"
     
-    return {
+    formatted_space = {
         "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 [{"error": spaces}]
     
     formatted = []
@@ -97,8 +98,31 @@ def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
             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
 
+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'])
+        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"
+        info += f"Likes: {space.get('likes', 'N/A')}\n"
+        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
+    except Exception as e:
+        print(f"Error in on_select: {str(e)}")
+        print(traceback.format_exc())
+        return f"오류가 발생했습니다: {str(e)}", "", "", "", ""
 
 def summarize_space(space: Dict) -> str:
     system_message = "당신은 Hugging Face Space의 내용을 요약하는 AI 조수입니다. 주어진 정보를 바탕으로 간결하고 명확한 요약을 제공해주세요."
@@ -180,28 +204,7 @@ def format_list_structure(tree_data: Dict) -> List[str]:
             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}")
@@ -287,6 +290,7 @@ def create_ui():
         }
         """
 
+
         with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
             gr.Markdown("# 300: HuggingFace Trending Spaces")
             
@@ -301,7 +305,6 @@ def create_ui():
                                 button = gr.Button("클릭", elem_classes="minimal-button")
                         space_rows.append((space_row, button, space))
 
-
                 with gr.Column(scale=2):
                     with gr.Tabs():
                         with gr.TabItem("기본 정보"):
@@ -330,7 +333,6 @@ 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}"
@@ -347,6 +349,7 @@ def create_ui():
                         outputs=[screenshot_output, last_url_state]
                     )
 
+
                 def open_space_in_browser(url):
                     if url:
                         import webbrowser