Update app.py

app.py

@@ -997,13 +997,18 @@ def refresh_data():
 
 def create_registration_bar_chart(data, type_name="Spaces"):
     try:
+        # TOP 기준 설정
+        top_limit = 500 if type_name == "Spaces" else 3000
+        
         # DataFrame인 경우 처리
         if isinstance(data, pd.DataFrame):
             if type_name == "Models":
                 # 3000위 이내의 모델만 필터링
                 data = data[data['Global Rank'].apply(lambda x: isinstance(x, (int, float)) or (isinstance(x, str) and x.startswith('#')))]
-                # '#' 제거하고 숫자로 변환
-                data = data[data['Global Rank'].apply(lambda x: int(str(x).replace('#', '')) if isinstance(x, str) else x) <= 3000]
+                data = data[data['Global Rank'].apply(lambda x: int(str(x).replace('#', '')) if isinstance(x, str) else x) <= top_limit]
+            elif type_name == "Spaces":
+                # 500위 이내의 스페이스만 필터링
+                data = data[data['Rank'].apply(lambda x: isinstance(x, (int, float))) & (data['Rank'] <= top_limit)]
             
             # ID 컬럼 선택
             id_column = 'Space ID' if type_name == "Spaces" else 'Model ID'
@@ -1013,11 +1018,9 @@ def create_registration_bar_chart(data, type_name="Spaces"):
             registrations = {}
             for item in data:
                 if isinstance(item, dict):
-                    # Models의 경우 3000위 이내만 처리
-                    if type_name == "Models":
-                        rank = item.get('global_rank')
-                        if isinstance(rank, str) or rank > 3000:
-                            continue
+                    rank = item.get('global_rank' if type_name == "Models" else 'rank')
+                    if isinstance(rank, str) or rank > top_limit:
+                        continue
                     creator = item.get('id', '').split('/')[0]
                     registrations[creator] = registrations.get(creator, 0) + 1
             registrations = pd.Series(registrations)
@@ -1034,7 +1037,7 @@ def create_registration_bar_chart(data, type_name="Spaces"):
         )])
         
         fig.update_layout(
-            title=f"Korean {type_name} Registrations by Creator (Top 3000)",
+            title=f"Korean {type_name} Registrations by Creator (Top {top_limit})",
             xaxis_title="Creator ID",
             yaxis_title="Number of Registrations",
             showlegend=False,
@@ -1049,25 +1052,30 @@ def create_registration_bar_chart(data, type_name="Spaces"):
 
 def create_pie_chart(data, total_count, type_name="Spaces"):
     try:
+        # TOP 기준 설정
+        top_limit = 500 if type_name == "Spaces" else 3000
+        
         # DataFrame인 경우 처리
         if isinstance(data, pd.DataFrame):
             if type_name == "Models":
                 # 3000위 이내의 모델만 필터링
                 data = data[data['Global Rank'].apply(lambda x: isinstance(x, (int, float)) or (isinstance(x, str) and x.startswith('#')))]
-                data = data[data['Global Rank'].apply(lambda x: int(str(x).replace('#', '')) if isinstance(x, str) else x) <= 3000]
+                data = data[data['Global Rank'].apply(lambda x: int(str(x).replace('#', '')) if isinstance(x, str) else x) <= top_limit]
+            elif type_name == "Spaces":
+                # 500위 이내의 스페이스만 필터링
+                data = data[data['Rank'].apply(lambda x: isinstance(x, (int, float))) & (data['Rank'] <= top_limit)]
             korean_count = len(data)
         else:
             # 리스트나 다른 형태의 데이터인 경우 처리
             if type_name == "Models":
-                # 3000위 이내의 모델만 카운트
-                korean_count = sum(1 for item in data if isinstance(item.get('global_rank'), (int, float)) and item.get('global_rank') <= 3000)
+                korean_count = sum(1 for item in data if isinstance(item.get('global_rank'), (int, float)) and item.get('global_rank') <= top_limit)
             else:
-                korean_count = len(data)
+                korean_count = sum(1 for item in data if isinstance(item.get('rank'), (int, float)) and item.get('rank') <= top_limit)
         
         other_count = total_count - korean_count
         
         fig = go.Figure(data=[go.Pie(
-            labels=[f'Korean {type_name} in Top 3000', f'Other {type_name} in Top 3000'],
+            labels=[f'Korean {type_name} in Top {top_limit}', f'Other {type_name} in Top {top_limit}'],
             values=[korean_count, other_count],
             hole=.3,
             marker_colors=['#FF6B6B', '#4ECDC4'],
@@ -1078,7 +1086,7 @@ def create_pie_chart(data, total_count, type_name="Spaces"):
         )])
         
         fig.update_layout(
-            title=f"Korean vs Other {type_name} Distribution (Top 3000)",
+            title=f"Korean vs Other {type_name} Distribution (Top {top_limit})",
             showlegend=True,
             height=400,
             width=500