Update app.py

app.py

@@ -375,6 +375,24 @@ def get_all_models(limit=3000):
     return all_models[:limit]
 
 def get_models_data(progress=gr.Progress()):
+    def calculate_rank(model_id, all_global_models, korea_models):
+        # 글로벌 순위 확인
+        global_rank = next((idx for idx, m in enumerate(all_global_models, 1) 
+                          if m.get('id', '').strip() == model_id.strip()), None)
+        
+        # Korea 모델인 경우
+        is_korea = any(m.get('id', '').strip() == model_id.strip() for m in korea_models)
+        
+        if is_korea:
+            # Korea 모델 중에서의 순위 확인
+            korea_rank = next((idx for idx, m in enumerate(korea_models, 1)
+                             if m.get('id', '').strip() == model_id.strip()), None)
+            
+            if korea_rank:
+                return min(global_rank or 3001, korea_rank + 1000), True
+        
+        return global_rank if global_rank else 'Not in top 3000', is_korea
+
     try:
         progress(0, desc="Fetching models...")
         
@@ -393,14 +411,22 @@ def get_models_data(progress=gr.Progress()):
         all_global_models = get_all_models(limit=3000)
         korea_models = get_korea_models()
         
+        print(f"Total global models fetched: {len(all_global_models)}")
+        print(f"Total Korea models fetched: {len(korea_models)}")
+        
         # 모든 모델 통합 (중복 제거)
         all_models = all_global_models.copy()
         existing_ids = {model.get('id', '') for model in all_global_models}
         
+        added_korea_models = 0
         for korea_model in korea_models:
             if korea_model.get('id', '') not in existing_ids:
                 all_models.append(korea_model)
                 existing_ids.add(korea_model.get('id', ''))
+                added_korea_models += 1
+        
+        print(f"Added {added_korea_models} unique Korea models")
+        print(f"Total combined models: {len(all_models)}")
         
         # 시각화를 위한 Figure 생성
         fig = go.Figure()
@@ -418,15 +444,18 @@ def get_models_data(progress=gr.Progress()):
                 
                 if response.status_code == 200:
                     model_data = response.json()
+                    rank, is_korea = calculate_rank(model_id, all_global_models, korea_models)
+                    
                     filtered_models.append({
                         'id': model_id,
-                        'global_rank': next((idx for idx, m in enumerate(all_models, 1) 
-                                          if m.get('id', '').strip() == model_id.strip()), 'Not in top 3000'),
+                        'global_rank': rank,
                         'downloads': model_data.get('downloads', 0),
                         'likes': model_data.get('likes', 0),
                         'title': model_data.get('title', 'No Title'),
-                        'is_korea': any(m.get('id', '').strip() == model_id.strip() for m in korea_models)
+                        'is_korea': is_korea
                     })
+                    
+                    print(f"Model {model_id}: Rank={rank}, Is Korea={is_korea}")
                 else:
                     filtered_models.append({
                         'id': model_id,