Update app.py

app.py

@@ -7,6 +7,149 @@ import os
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 
+target_models = {
+    "openfree/flux-lora-korea-palace": "https://huggingface.co/openfree/flux-lora-korea-palace",
+    "seawolf2357/hanbok": "https://huggingface.co/seawolf2357/hanbok",
+    "LGAI-EXAONE/EXAONE-3.5-32B-Instruct": "https://huggingface.co/LGAI-EXAONE/EXAONE-3.5-32B-Instruct",
+    "LGAI-EXAONE/EXAONE-3.5-2.4B-Instruct": "https://huggingface.co/LGAI-EXAONE/EXAONE-3.5-2.4B-Instruct",
+    "LGAI-EXAONE/EXAONE-3.5-7.8B-Instruct": "https://huggingface.co/LGAI-EXAONE/EXAONE-3.5-7.8B-Instruct",
+    "ginipick/flux-lora-eric-cat": "https://huggingface.co/ginipick/flux-lora-eric-cat",
+    "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()):
+    """모델 데이터 가져오기"""
+    url = "https://huggingface.co/api/models"
+    params = {
+        'full': 'true',
+        'limit': 1000,  # 상위 1000개 모델
+        'sort': 'downloads' if sort_type == "modes" else 'lastModified'
+    }
+    
+    try:
+        progress(0, desc=f"Fetching {sort_type} models data...")
+        response = requests.get(url, params=params)
+        response.raise_for_status()
+        all_models = response.json()
+        
+        # 순위 정보 저장
+        model_ranks = {model['id']: idx + 1 for idx, model in enumerate(all_models)}
+        
+        # target_models 필터링 및 순위 정보 포함
+        models = []
+        for model in all_models:
+            if model.get('id', '') in target_models:
+                model['rank'] = model_ranks.get(model['id'], 'N/A')
+                models.append(model)
+        
+        # 순위별로 정렬
+        models.sort(key=lambda x: x['rank'])
+        
+        progress(0.3, desc="Creating visualization...")
+        
+        # 시각화 생성
+        fig = go.Figure()
+        
+        # 데이터 준비
+        ids = [model['id'] for model in models]
+        ranks = [model['rank'] 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]
+        
+        # 막대 그래프 생성
+        fig.add_trace(go.Bar(
+            x=ids,
+            y=y_values,
+            text=[f"Rank: {r}<br>Downloads: {d}" for r, d in zip(ranks, downloads)],
+            textposition='auto',
+            marker_color='rgb(255,127,14)',
+            opacity=0.8
+        ))
+        
+        fig.update_layout(
+            title={
+                'text': f'Hugging Face Models {sort_type.title()} Rankings (Top 1000)',
+                'y':0.95,
+                'x':0.5,
+                'xanchor': 'center',
+                'yanchor': 'top'
+            },
+            xaxis_title='Model ID',
+            yaxis_title='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]
+            ),
+            height=800,
+            showlegend=False,
+            template='plotly_white',
+            xaxis_tickangle=-45
+        )
+        
+        progress(0.6, desc="Creating model cards...")
+        
+        # HTML 카드 생성
+        html_content = f"""
+        <div style='padding: 20px; background: #f5f5f5;'>
+            <h2 style='color: #2c3e50;'>Model {sort_type.title()} 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)
+            
+            html_content += f"""
+            <div style='
+                background: white;
+                padding: 20px;
+                border-radius: 10px;
+                box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+                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>
+                <a href='{target_models[model_id]}' 
+                   target='_blank' 
+                   style='
+                    display: inline-block;
+                    padding: 8px 16px;
+                    background: #e67e22;
+                    color: white;
+                    text-decoration: none;
+                    border-radius: 5px;
+                    transition: background 0.3s;
+                   '>
+                   Visit Model 🔗
+                </a>
+            </div>
+            """
+        
+        html_content += "</div></div>"
+        
+        # 데이터프레임 생성
+        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'),
+            'URL': target_models[model.get('id', '')]
+        } for model in models])
+        
+        progress(1.0, desc="Complete!")
+        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()
+        
 # 관심 스페이스 URL 리스트와 정보
 target_spaces = {
     "ginipick/FLUXllama": "https://huggingface.co/spaces/ginipick/FLUXllama",
@@ -388,10 +531,11 @@ def refresh_data():
     else:
         return create_error_plot(), "<div>API 인증이 필요합니다.</div>", pd.DataFrame()
 
+
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
-    # 🤗 HuggingFace Spaces Rankings Analysis
-    실시간으로 Hugging Face Spaces의 순위를 분석합니다.
+    # 🤗 HuggingFace Rankings Analysis
+    실시간으로 Hugging Face의 Spaces와 Models 순위를 분석합니다.
     """)
     
     with gr.Tab("Trending Rankings"):
@@ -404,28 +548,37 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         modes_info = gr.HTML()
         modes_df = gr.DataFrame()
     
+    with gr.Tab("Models Rankings"):
+        models_plot = gr.Plot()
+        models_info = gr.HTML()
+        models_df = gr.DataFrame()
+    
     refresh_btn = gr.Button("🔄 Refresh Data", variant="primary")
     
-    def refresh_data():
+    def refresh_all_data():
         trending_results = get_spaces_data("trending")
         modes_results = get_spaces_data("modes")
-        return [*trending_results, *modes_results]
+        models_results = get_models_data("trending")
+        return [*trending_results, *modes_results, *models_results]
     
     refresh_btn.click(
-        refresh_data,
+        refresh_all_data,
         outputs=[
             trending_plot, trending_info, trending_df,
-            modes_plot, modes_info, modes_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")
     
     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 앱 실행
 demo.launch(
     server_name="0.0.0.0",