Update app.py

app.py

@@ -141,10 +141,10 @@ target_models = {
 
 def get_models_data(progress=gr.Progress()):
     """모델 데이터 가져오기"""
-    url = "https://huggingface.co/api/models"
+    url = "https://huggingface.co/api/models/sort/trending"  # trending 정렬 사용
     params = {
         'full': 'true',
-        'limit': 300  # 스페이스와 동일하게 300개로 설정
+        'limit': 1000  # 1000위까지 가져오기
     }
     
     try:
@@ -159,13 +159,19 @@ def get_models_data(progress=gr.Progress()):
         
         all_models = response.json()
         
-        # target_models에 있는 모델만 필터링
+        # 전체 모델의 순위 정보 저장 (1000위까지)
+        model_ranks = {model['id']: idx + 1 for idx, model in enumerate(all_models)}
+        
+        # target_models에 있는 모델만 필터링하고 실제 순위 포함
         filtered_models = []
         for model in all_models:
             if model.get('id', '') in target_models:
-                model['rank'] = len(filtered_models) + 1
+                model['rank'] = model_ranks[model['id']]  # 실제 순위 사용
                 filtered_models.append(model)
         
+        # 순위로 정렬
+        filtered_models.sort(key=lambda x: x['rank'])
+        
         if not filtered_models:
             return create_error_plot(), "<div>선택된 모델의 데이터를 찾을 수 없습니다.</div>", pd.DataFrame()
         
@@ -178,15 +184,17 @@ 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]
         
-        # Y축 값을 반전 (300 - rank + 1)
-        y_values = [301 - r for r in ranks]
+        # Y축 값을 반전 (1000 - rank + 1)
+        y_values = [1001 - r for r in ranks]
         
         # 막대 그래프 생성
         fig.add_trace(go.Bar(
             x=ids,
             y=y_values,
-            text=[f"Rank: {r}<br>Likes: {l}" for r, l in zip(ranks, likes)],
+            text=[f"Rank: {r}<br>Likes: {l}<br>Downloads: {d}" 
+                  for r, l, d in zip(ranks, likes, downloads)],
             textposition='auto',
             marker_color='rgb(158,202,225)',
             opacity=0.8
@@ -194,7 +202,7 @@ def get_models_data(progress=gr.Progress()):
         
         fig.update_layout(
             title={
-                'text': 'Hugging Face Models Rankings (Top 300)',
+                'text': 'Hugging Face Models Trending Rankings (Top 1000)',
                 'y':0.95,
                 'x':0.5,
                 'xanchor': 'center',
@@ -203,9 +211,9 @@ def get_models_data(progress=gr.Progress()):
             xaxis_title='Model ID',
             yaxis_title='Rank',
             yaxis=dict(
-                ticktext=[str(i) for i in range(1, 301, 20)],
-                tickvals=[301 - i for i in range(1, 301, 20)],
-                range=[0, 300]
+                ticktext=[str(i) for i in range(1, 1001, 50)],
+                tickvals=[1001 - i for i in range(1, 1001, 50)],
+                range=[0, 1000]
             ),
             height=800,
             showlegend=False,
@@ -218,10 +226,11 @@ def get_models_data(progress=gr.Progress()):
         # HTML 카드 생성
         html_content = """
         <div style='padding: 20px; background: #f5f5f5;'>
-            <h2 style='color: #2c3e50;'>Models Rankings</h2>
+            <h2 style='color: #2c3e50;'>Models Trending Rankings</h2>
             <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
         """
         
+        # 순위권 내 모델 카드 생성
         for model in filtered_models:
             model_id = model.get('id', '')
             rank = model.get('rank', 'N/A')
@@ -255,16 +264,58 @@ def get_models_data(progress=gr.Progress()):
             </div>
             """
         
+        # 순위권 밖 모델 카드 생성
+        for model_id in target_models:
+            if model_id not in [m['id'] for m in filtered_models]:
+                html_content += f"""
+                <div style='
+                    background: #f8f9fa;
+                    padding: 20px;
+                    border-radius: 10px;
+                    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</p>
+                    <a href='{target_models[model_id]}' 
+                       target='_blank' 
+                       style='
+                        display: inline-block;
+                        padding: 8px 16px;
+                        background: #95a5a6;
+                        color: white;
+                        text-decoration: none;
+                        border-radius: 5px;
+                       '>
+                       Visit Model 🔗
+                    </a>
+                </div>
+                """
+        
         html_content += "</div></div>"
         
         # 데이터프레임 생성
-        df = pd.DataFrame([{
-            'Rank': model.get('rank', 'N/A'),
-            'Model ID': model.get('id', ''),
-            'Likes': model.get('likes', 'N/A'),
-            'Downloads': model.get('downloads', 'N/A'),
-            'URL': target_models[model.get('id', '')]
-        } for model in filtered_models])
+        df_data = []
+        # 순위권 내 모델
+        for model in filtered_models:
+            df_data.append({
+                'Rank': model.get('rank', 'N/A'),
+                'Model ID': model.get('id', ''),
+                'Likes': model.get('likes', 'N/A'),
+                'Downloads': model.get('downloads', 'N/A'),
+                'URL': target_models[model.get('id', '')]
+            })
+        # 순위권 밖 모델
+        for model_id in target_models:
+            if model_id not in [m['id'] for m in filtered_models]:
+                df_data.append({
+                    'Rank': 'Not in top 1000',
+                    'Model ID': model_id,
+                    'Likes': 'N/A',
+                    'Downloads': 'N/A',
+                    'URL': target_models[model_id]
+                })
+        
+        df = pd.DataFrame(df_data)
         
         progress(1.0, desc="Complete!")
         return fig, html_content, df