Update app.py

app.py

@@ -301,36 +301,26 @@ def get_models_data(progress=gr.Progress()):
     try:
         progress(0, desc="Fetching global rankings...")
         
-        # 전체 모델 목록을 여러 번 나눠서 가져오기
-        all_global_models = []
-        page_size = 10000
-        max_models = 50000  # 충분히 큰 수로 설정
-        
-        for offset in range(0, max_models, page_size):
-            global_params = {
-                'full': 'true',
-                'limit': page_size,
-                'offset': offset,
-                'sort': 'downloads',
-                'direction': -1
-            }
-            
-            global_response = requests.get(
-                "https://huggingface.co/api/models",
-                headers={'Accept': 'application/json'},
-                params=global_params
-            )
+        # 전체 모델 목록 가져오기 (좋아요 순으로 정렬)
+        global_params = {
+            'full': 'true',
+            'limit': 3000,
+            'sort': 'likes',  # 좋아요 순으로 변경
+            'direction': -1
+        }
+        
+        global_response = requests.get(
+            "https://huggingface.co/api/models",
+            headers={'Accept': 'application/json'},
+            params=global_params
+        )
+        
+        if global_response.status_code != 200:
+            print(f"Failed to fetch global rankings: {global_response.status_code}")
+            return create_error_plot(), "<div>전체 순위 데이터를 가져오는데 실패했습니다.</div>", pd.DataFrame()
             
-            if global_response.status_code != 200:
-                print(f"Failed to fetch global rankings page {offset//page_size + 1}")
-                continue
-                
-            page_models = global_response.json()
-            if not page_models:  # 더 이상 모델이 없으면 중단
-                break
-                
-            all_global_models.extend(page_models)
-            print(f"Fetched {len(all_global_models)} models so far...")
+        all_global_models = global_response.json()
+        print(f"Fetched {len(all_global_models)} models")
         
         # 전체 순위 맵 생성
         global_ranks = {}
@@ -339,35 +329,51 @@ def get_models_data(progress=gr.Progress()):
             global_ranks[model_id] = {
                 'rank': idx,
                 'downloads': model.get('downloads', 0),
-                'likes': model.get('likes', 0)
+                'likes': model.get('likes', 0),
+                'title': model.get('title', 'No Title')
             }
         
-        print(f"Total models fetched: {len(global_ranks)}")
-        
-        # target_models의 순위 확인 및 저장
+        # target_models의 상세 정보 가져오기
         filtered_models = []
         for model_id in target_models.keys():
-            normalized_id = normalize_model_id(model_id)
-            if normalized_id in global_ranks:
-                rank_info = global_ranks[normalized_id]
-                model_info = {
-                    'id': model_id,
-                    'global_rank': rank_info['rank'],
-                    'downloads': rank_info['downloads'],
-                    'likes': rank_info['likes'],
-                    'title': 'No Title'  # 필요한 경우 개별 API 호출로 가져올 수 있음
-                }
-                print(f"Found {model_id} at rank {rank_info['rank']}")
-            else:
-                model_info = {
+            try:
+                # 개별 모델 API 호출로 정확한 정보 가져오기
+                model_url_api = f"https://huggingface.co/api/models/{model_id}"
+                response = requests.get(model_url_api, headers={'Accept': 'application/json'})
+                
+                if response.status_code == 200:
+                    model_data = response.json()
+                    normalized_id = normalize_model_id(model_id)
+                    
+                    # 전체 순위 정보와 개별 모델 정보 결합
+                    rank_info = global_ranks.get(normalized_id, {})
+                    model_info = {
+                        'id': model_id,
+                        'global_rank': rank_info.get('rank', 'Not in top 3000'),
+                        'downloads': model_data.get('downloads', 0),
+                        'likes': model_data.get('likes', 0),
+                        'title': model_data.get('title', 'No Title')
+                    }
+                    filtered_models.append(model_info)
+                    print(f"Model {model_id}: Rank={model_info['global_rank']}, Likes={model_info['likes']}, Downloads={model_info['downloads']}")
+                else:
+                    print(f"Failed to fetch data for {model_id}: {response.status_code}")
+                    filtered_models.append({
+                        'id': model_id,
+                        'global_rank': 'Not in top 3000',
+                        'downloads': 0,
+                        'likes': 0,
+                        'title': 'No Title'
+                    })
+            except Exception as e:
+                print(f"Error fetching data for {model_id}: {str(e)}")
+                filtered_models.append({
                     'id': model_id,
-                    'global_rank': 'Not found',
+                    'global_rank': 'Not in top 3000',
                     'downloads': 0,
                     'likes': 0,
                     'title': 'No Title'
-                }
-                print(f"Model {model_id} not found in rankings")
-            filtered_models.append(model_info)
+                })
         
         # 순위로 정렬
         filtered_models.sort(key=lambda x: float('inf') if isinstance(x['global_rank'], str) else x['global_rank'])
@@ -408,7 +414,7 @@ def get_models_data(progress=gr.Progress()):
             
             fig.update_layout(
                 title={
-                    'text': 'Hugging Face Models Global Rankings',
+                    'text': 'Hugging Face Models Global Rankings (by Likes)',
                     'y':0.95,
                     'x':0.5,
                     'xanchor': 'center',
@@ -428,7 +434,7 @@ def get_models_data(progress=gr.Progress()):
         # HTML 카드 생성
         html_content = """
         <div style='padding: 20px; background: #f5f5f5;'>
-            <h2 style='color: #2c3e50;'>Models Global Rankings</h2>
+            <h2 style='color: #2c3e50;'>Models Global Rankings (by Likes)</h2>
             <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
         """