Update app.py

app.py

@@ -303,7 +303,7 @@ def get_models_data(progress=gr.Progress()):
         progress(0, desc="Fetching models data...")
         params = {
             'full': 'true',
-            'limit': 1000,
+            'limit': 3000,  # 3000개로 증가
             'sort': 'downloads',
             'direction': -1
         }
@@ -320,21 +320,30 @@ def get_models_data(progress=gr.Progress()):
         
         # 전체 순위 정보 저장 (다운로드 수 기준)
         model_ranks = {}
+        model_data = {}  # 모든 모델의 상세 데이터 저장
+        
         for idx, model in enumerate(models, 1):
             model_id = normalize_model_id(model.get('id', ''))
-            model_ranks[model_id] = {
+            model_data[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')
             }
         
         # target_models 중 순위권 내 모델 필터링
         filtered_models = []
-        for model in models:
-            model_id = normalize_model_id(model.get('id', ''))
-            if model_id in [normalize_model_id(tid) for tid in target_models.keys()]:
-                model['rank'] = model_ranks[model_id]['rank']
-                filtered_models.append(model)
+        for target_id in target_models.keys():
+            normalized_target_id = normalize_model_id(target_id)
+            if normalized_target_id in model_data:
+                model_info = {
+                    'id': target_id,
+                    'rank': model_data[normalized_target_id]['rank'],
+                    'downloads': model_data[normalized_target_id]['downloads'],
+                    'likes': model_data[normalized_target_id]['likes'],
+                    'title': model_data[normalized_target_id]['title']
+                }
+                filtered_models.append(model_info)
         
         # 순위로 정렬
         filtered_models.sort(key=lambda x: x['rank'])
@@ -350,11 +359,11 @@ def get_models_data(progress=gr.Progress()):
         # 데이터 준비
         ids = [model['id'] for model in filtered_models]
         ranks = [model['rank'] for model in filtered_models]
-        likes = [model.get('likes', 0) for model in filtered_models]
-        downloads = [model.get('downloads', 0) for model in filtered_models]
+        likes = [model['likes'] for model in filtered_models]
+        downloads = [model['downloads'] for model in filtered_models]
         
         # Y축 값을 반전
-        y_values = [1001 - r for r in ranks]
+        y_values = [3001 - r for r in ranks]  # 3000으로 변경
         
         # 막대 그래프 생성
         fig.add_trace(go.Bar(
@@ -369,7 +378,7 @@ def get_models_data(progress=gr.Progress()):
         
         fig.update_layout(
             title={
-                'text': 'Hugging Face Models Global Download Rankings (Top 1000)',
+                'text': 'Hugging Face Models Global Download Rankings (Top 3000)',
                 'y':0.95,
                 'x':0.5,
                 'xanchor': 'center',
@@ -378,9 +387,9 @@ def get_models_data(progress=gr.Progress()):
             xaxis_title='Model ID',
             yaxis_title='Global Rank',
             yaxis=dict(
-                ticktext=[str(i) for i in range(1, 1001, 50)],
-                tickvals=[1001 - i for i in range(1, 1001, 50)],
-                range=[0, 1000]
+                ticktext=[str(i) for i in range(1, 3001, 150)],  # 간격 조정
+                tickvals=[3001 - i for i in range(1, 3001, 150)],
+                range=[0, 3000]
             ),
             height=800,
             showlegend=False,
@@ -393,7 +402,7 @@ def get_models_data(progress=gr.Progress()):
         # HTML 카드 생성
         html_content = """
         <div style='padding: 20px; background: #f5f5f5;'>
-            <h2 style='color: #2c3e50;'>Models Global Download Rankings</h2>
+            <h2 style='color: #2c3e50;'>Models Global Download Rankings (Top 3000)</h2>
             <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
         """
         
@@ -401,8 +410,9 @@ def get_models_data(progress=gr.Progress()):
         for model in filtered_models:
             model_id = model['id']
             rank = model['rank']
-            likes = model.get('likes', 0)
-            downloads = model.get('downloads', 0)
+            likes = model['likes']
+            downloads = model['downloads']
+            title = model.get('title', 'No Title')
             
             html_content += f"""
             <div style='
@@ -413,6 +423,7 @@ def get_models_data(progress=gr.Progress()):
                 transition: transform 0.2s;
             '>
                 <h3 style='color: #34495e;'>Global Rank #{rank} - {model_id}</h3>
+                <p style='color: #2c3e50;'>{title}</p>
                 <p style='color: #7f8c8d;'>👍 Likes: {likes:,}</p>
                 <p style='color: #7f8c8d;'>⬇️ Downloads: {downloads:,}</p>
                 <a href='{target_models[model_id]}' 
@@ -442,7 +453,7 @@ def get_models_data(progress=gr.Progress()):
                     box-shadow: 0 2px 4px rgba(0,0,0,0.1);
                 '>
                     <h3 style='color: #34495e;'>{model_id}</h3>
-                    <p style='color: #7f8c8d;'>Not in top 1000 by downloads</p>
+                    <p style='color: #7f8c8d;'>Not in top 3000 by downloads</p>
                     <a href='{target_models[model_id]}' 
                        target='_blank' 
                        style='
@@ -467,16 +478,18 @@ def get_models_data(progress=gr.Progress()):
             df_data.append({
                 'Global Rank': model['rank'],
                 'Model ID': model['id'],
-                'Likes': f"{model.get('likes', 0):,}",
-                'Downloads': f"{model.get('downloads', 0):,}",
+                'Title': model.get('title', 'No Title'),
+                'Likes': f"{model['likes']:,}",
+                'Downloads': f"{model['downloads']:,}",
                 'URL': target_models[model['id']]
             })
         # 순위권 밖 모델
         for model_id in target_models:
             if model_id not in [m['id'] for m in filtered_models]:
                 df_data.append({
-                    'Global Rank': 'Not in top 1000',
+                    'Global Rank': 'Not in top 3000',
                     'Model ID': model_id,
+                    'Title': 'N/A',
                     'Likes': 'N/A',
                     'Downloads': 'N/A',
                     'URL': target_models[model_id]