Update app.py

app.py

@@ -17,17 +17,16 @@ target_models = {
     "seawolf2357/flux-lora-car-rolls-royce": "https://huggingface.co/seawolf2357/flux-lora-car-rolls-royce"
 }
 
-def get_models_data(sort_type="trending", progress=gr.Progress()):
+def get_models_data(progress=gr.Progress()):
     """모델 데이터 가져오기"""
-    url = "https://huggingface.co/api/models"
+    url = "https://huggingface.co/api/models/sort/trending"  # trending API 엔드포인트로 변경
     params = {
-        'full': 'true',
-        'limit': 1000,  # 상위 1000개 모델
-        'sort': 'downloads' if sort_type == "modes" else 'lastModified'
+        'limit': 1000,
+        'full': 'true'
     }
     
     try:
-        progress(0, desc=f"Fetching {sort_type} models data...")
+        progress(0, desc="Fetching models data...")
         response = requests.get(url, params=params)
         response.raise_for_status()
         all_models = response.json()
@@ -53,24 +52,24 @@ def get_models_data(sort_type="trending", progress=gr.Progress()):
         # 데이터 준비
         ids = [model['id'] for model in models]
         ranks = [model['rank'] for model in models]
-        downloads = [model.get('downloads', 0) for model in models]
+        likes = [model.get('likes', 0) for model in models]
         
-        # Y축 값을 반전 (1000 - rank + 1)
+        # Y축 값을 반전
         y_values = [1001 - r for r in ranks]
         
         # 막대 그래프 생성
         fig.add_trace(go.Bar(
             x=ids,
             y=y_values,
-            text=[f"Rank: {r}<br>Downloads: {d}" for r, d in zip(ranks, downloads)],
+            text=[f"Rank: {r}<br>Likes: {l}" for r, l in zip(ranks, likes)],
             textposition='auto',
-            marker_color='rgb(255,127,14)',
+            marker_color='rgb(158,202,225)',
             opacity=0.8
         ))
         
         fig.update_layout(
             title={
-                'text': f'Hugging Face Models {sort_type.title()} Rankings (Top 1000)',
+                'text': 'Hugging Face Models Trending Rankings',
                 'y':0.95,
                 'x':0.5,
                 'xanchor': 'center',
@@ -92,17 +91,17 @@ def get_models_data(sort_type="trending", progress=gr.Progress()):
         progress(0.6, desc="Creating model cards...")
         
         # HTML 카드 생성
-        html_content = f"""
+        html_content = """
         <div style='padding: 20px; background: #f5f5f5;'>
-            <h2 style='color: #2c3e50;'>Model {sort_type.title()} 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 models:
             model_id = model.get('id', '')
             rank = model.get('rank', 'N/A')
-            downloads = model.get('downloads', 0)
             likes = model.get('likes', 0)
+            downloads = model.get('downloads', 0)
             
             html_content += f"""
             <div style='
@@ -113,14 +112,14 @@ def get_models_data(sort_type="trending", progress=gr.Progress()):
                 transition: transform 0.2s;
             '>
                 <h3 style='color: #34495e;'>Rank #{rank} - {model_id}</h3>
-                <p style='color: #7f8c8d;'>⬇️ Downloads: {downloads}</p>
                 <p style='color: #7f8c8d;'>👍 Likes: {likes}</p>
+                <p style='color: #7f8c8d;'>⬇️ Downloads: {downloads}</p>
                 <a href='{target_models[model_id]}' 
                    target='_blank' 
                    style='
                     display: inline-block;
                     padding: 8px 16px;
-                    background: #e67e22;
+                    background: #3498db;
                     color: white;
                     text-decoration: none;
                     border-radius: 5px;
@@ -137,8 +136,8 @@ def get_models_data(sort_type="trending", progress=gr.Progress()):
         df = pd.DataFrame([{
             'Rank': model.get('rank', 'N/A'),
             'Model ID': model.get('id', ''),
-            'Downloads': model.get('downloads', 'N/A'),
             'Likes': model.get('likes', 'N/A'),
+            'Downloads': model.get('downloads', 'N/A'),
             'URL': target_models[model.get('id', '')]
         } for model in models])
         
@@ -538,17 +537,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     실시간으로 Hugging Face의 Spaces와 Models 순위를 분석합니다.
     """)
     
-    with gr.Tab("Trending Rankings"):
+    with gr.Tab("Spaces Trending"):
         trending_plot = gr.Plot()
         trending_info = gr.HTML()
         trending_df = gr.DataFrame()
     
-    with gr.Tab("Most Liked Rankings"):
-        modes_plot = gr.Plot()
-        modes_info = gr.HTML()
-        modes_df = gr.DataFrame()
-    
-    with gr.Tab("Models Rankings"):
+    with gr.Tab("Models Trending"):
         models_plot = gr.Plot()
         models_info = gr.HTML()
         models_df = gr.DataFrame()
@@ -557,26 +551,22 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     
     def refresh_all_data():
         trending_results = get_spaces_data("trending")
-        modes_results = get_spaces_data("modes")
-        models_results = get_models_data("trending")
-        return [*trending_results, *modes_results, *models_results]
+        models_results = get_models_data()
+        return [*trending_results, *models_results]
     
     refresh_btn.click(
         refresh_all_data,
         outputs=[
             trending_plot, trending_info, trending_df,
-            modes_plot, modes_info, modes_df,
             models_plot, models_info, models_df
         ]
     )
     
     # 초기 데이터 로드
     trending_results = get_spaces_data("trending")
-    modes_results = get_spaces_data("modes")
-    models_results = get_models_data("trending")
+    models_results = get_models_data()
     
     trending_plot.value, trending_info.value, trending_df.value = trending_results
-    modes_plot.value, modes_info.value, modes_df.value = modes_results
     models_plot.value, models_info.value, models_df.value = models_results
     
 # Gradio 앱 실행