Update app.py

app.py

@@ -543,54 +543,31 @@ target_spaces = {
 
 def get_spaces_data(sort_type="trending", progress=gr.Progress()):
     """스페이스 데이터 가져오기 (trending 또는 modes)"""
-    url = f"https://huggingface.co/api/spaces"
-    params = {
-        'full': 'true',
-        'limit': 300
-    }
+    url = "https://huggingface.co/api/spaces"
     
     try:
         progress(0, desc=f"Fetching {sort_type} spaces data...")
+        params = {
+            'full': 'true',
+            'limit': 300
+        }
+        
         response = requests.get(url, params=params)
-        if response.status_code != 200:
-            print(f"API 요청 실패: {response.status_code}")
-            print(f"Response: {response.text}")
-            return create_error_plot(), "<div>스페이스 데이터를 가져오는데 실패했습니다.</div>", pd.DataFrame()
-            
+        response.raise_for_status()
         all_spaces = response.json()
         
-        # API 응답 구조 확인을 위한 로깅
-        print("Sample space data:", all_spaces[0] if all_spaces else "No data")
-        
         # 순위 정보 저장
         space_ranks = {}
-        space_details = {}
-        
         for idx, space in enumerate(all_spaces, 1):
             space_id = space.get('id', '')
-            space_ranks[space_id] = idx
-            space_details[space_id] = {
-                'title': space.get('title', 'No Title'),
-                'cardData': space.get('cardData', {}),
-                'likes': space.get('likes', 0),
-                'description': space.get('description', 'No Description')
-            }
+            if space_id in target_spaces:
+                # 전체 space 정보 저장
+                space['rank'] = idx
+                space_ranks[space_id] = space
         
-        # target_spaces 필터링 및 순위 정보 포함
-        spaces = []
-        for space_id in target_spaces.keys():
-            if space_id in space_ranks:
-                space_info = {
-                    'id': space_id,
-                    'rank': space_ranks[space_id],
-                    'title': space_details[space_id]['title'],
-                    'description': space_details[space_id]['description'],
-                    'likes': space_details[space_id]['likes']
-                }
-                spaces.append(space_info)
-        
-        # 순위별로 정렬
-        spaces.sort(key=lambda x: x['rank'])
+        # target_spaces 중 순위권 내 space 필터링
+        spaces = [space_ranks[space_id] for space_id in space_ranks.keys()]
+        spaces.sort(key=lambda x: x['rank'])  # 순위별로 정렬
         
         progress(0.3, desc="Creating visualization...")
         
@@ -600,8 +577,8 @@ def get_spaces_data(sort_type="trending", progress=gr.Progress()):
         # 데이터 준비
         ids = [space['id'] for space in spaces]
         ranks = [space['rank'] for space in spaces]
-        likes = [space['likes'] for space in spaces]
-        titles = [space['title'] for space in spaces]
+        likes = [space.get('likes', 0) for space in spaces]
+        titles = [space.get('title', 'No Title') for space in spaces]
         
         # Y축 값을 반전
         y_values = [301 - r for r in ranks]
@@ -650,9 +627,14 @@ def get_spaces_data(sort_type="trending", progress=gr.Progress()):
         for space in spaces:
             space_id = space['id']
             rank = space['rank']
-            title = space['title']
-            description = space['description']
-            likes = space['likes']
+            title = space.get('title', 'No Title')
+            likes = space.get('likes', 0)
+            description = space.get('description', '')
+            
+            # cardData에서 추가 정보 가져오기
+            card_data = space.get('cardData', {})
+            if not description and card_data:
+                description = card_data.get('description', 'No Description')
             
             # description이 너무 길면 자르기
             short_description = description[:150] + '...' if description and len(description) > 150 else description
@@ -691,9 +673,9 @@ def get_spaces_data(sort_type="trending", progress=gr.Progress()):
         df = pd.DataFrame([{
             'Rank': space['rank'],
             'Space ID': space['id'],
-            'Title': space['title'],
-            'Description': space['description'][:100] + '...' if space['description'] else 'No Description',
-            'Likes': space['likes'],
+            'Title': space.get('title', 'No Title'),
+            'Description': (space.get('description', '') or space.get('cardData', {}).get('description', 'No Description'))[:100] + '...',
+            'Likes': space.get('likes', 0),
             'URL': target_spaces[space['id']]
         } for space in spaces])