Update app.py

app.py

@@ -19,16 +19,32 @@ target_models = {
 
 def get_models_data(progress=gr.Progress()):
     """모델 데이터 가져오기 (스페이스와 동일한 방식)"""
-    url = "https://huggingface.co/api/models/sort/trending"  # trending API 엔드포인트
+    url = "https://huggingface.co/api/models"  # 기본 API 엔드포인트
     params = {
-        'limit': 1000,  # 상위 1000개 모델
-        'full': 'true'
+        'full': 'true',
+        'limit': 1000,
+        'sort': 'lastModified',  # 최신 순으로 정렬
+        'direction': -1
+    }
+    
+    headers = {
+        'Accept': 'application/json',
+        'User-Agent': 'Mozilla/5.0'
     }
     
+    # API 토큰이 있는 경우 헤더에 추가
+    if HF_TOKEN:
+        headers['Authorization'] = f'Bearer {HF_TOKEN}'
+    
     try:
         progress(0, desc="Fetching models data...")
-        response = requests.get(url, params=params)
-        response.raise_for_status()
+        response = requests.get(url, params=params, headers=headers)
+        
+        if response.status_code != 200:
+            print(f"API 요청 실패: {response.status_code}")
+            print(f"Response: {response.text}")
+            return create_error_plot(), "<div>모델 데이터를 가져오는데 실패했습니다.</div>", pd.DataFrame()
+        
         all_models = response.json()
         
         # 순위 정보 저장
@@ -41,6 +57,9 @@ def get_models_data(progress=gr.Progress()):
                 model['rank'] = model_ranks.get(model['id'], 'N/A')
                 models.append(model)
         
+        if not models:
+            return create_error_plot(), "<div>선택된 모델의 데이터를 찾을 수 없습니다.</div>", pd.DataFrame()
+        
         # 순위별로 정렬
         models.sort(key=lambda x: x['rank'])
         
@@ -53,6 +72,7 @@ def get_models_data(progress=gr.Progress()):
         ids = [model['id'] for model in models]
         ranks = [model['rank'] for model in models]
         likes = [model.get('likes', 0) for model in models]
+        downloads = [model.get('downloads', 0) for model in models]
         
         # Y축 값을 반전 (1000 - rank + 1)
         y_values = [1001 - r for r in ranks]
@@ -61,7 +81,8 @@ def get_models_data(progress=gr.Progress()):
         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
@@ -69,7 +90,7 @@ def get_models_data(progress=gr.Progress()):
         
         fig.update_layout(
             title={
-                'text': 'Hugging Face Models Trending Rankings',
+                'text': 'Hugging Face Models Rankings',
                 'y':0.95,
                 'x':0.5,
                 'xanchor': 'center',
@@ -93,7 +114,7 @@ def get_models_data(progress=gr.Progress()):
         # HTML 카드 생성
         html_content = """
         <div style='padding: 20px; background: #f5f5f5;'>
-            <h2 style='color: #2c3e50;'>Models Trending Rankings</h2>
+            <h2 style='color: #2c3e50;'>Models Rankings</h2>
             <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
         """
         
@@ -102,6 +123,7 @@ def get_models_data(progress=gr.Progress()):
             rank = model.get('rank', 'N/A')
             likes = model.get('likes', 0)
             downloads = model.get('downloads', 0)
+            description = model.get('description', 'No description available')[:200]
             
             html_content += f"""
             <div style='
@@ -114,6 +136,7 @@ def get_models_data(progress=gr.Progress()):
                 <h3 style='color: #34495e;'>Rank #{rank} - {model_id}</h3>
                 <p style='color: #7f8c8d;'>👍 Likes: {likes}</p>
                 <p style='color: #7f8c8d;'>⬇️ Downloads: {downloads}</p>
+                <p style='color: #7f8c8d; font-size: 0.9em;'>{description}...</p>
                 <a href='{target_models[model_id]}' 
                    target='_blank' 
                    style='
@@ -145,9 +168,8 @@ def get_models_data(progress=gr.Progress()):
         return fig, html_content, df
         
     except Exception as e:
-        error_html = f'<div style="color: red; padding: 20px;">Error: {str(e)}</div>'
-        error_plot = create_error_plot()
-        return error_plot, error_html, pd.DataFrame()
+        print(f"Error in get_models_data: {str(e)}")
+        return create_error_plot(), f"<div>에러 발생: {str(e)}</div>", pd.DataFrame()
         
 # 관심 스페이스 URL 리스트와 정보
 target_spaces = {