Update app.py

app.py

@@ -42,25 +42,27 @@ target_spaces = {
     "NCSOFT/VARCO_Arena": "https://huggingface.co/spaces/NCSOFT/VARCO_Arena"
 }
 
-def get_trending_spaces(progress=gr.Progress()):
-    """트렌딩 스페이스 가져오기"""
-    url = "https://huggingface.co/api/spaces"
+def get_spaces_data(sort_type="trending", progress=gr.Progress()):
+    """스페이스 데이터 가져오기 (trending 또는 modes)"""
+    url = f"https://huggingface.co/api/spaces"
+    params = {
+        'full': 'true',
+        'limit': 300
+    }
+    
+    if sort_type == "modes":
+        params['sort'] = 'likes'  # modes는 좋아요 순으로 정렬
     
     try:
-        progress(0, desc="Fetching spaces data...")
-        params = {
-            'full': 'true',
-            'limit': 300  # 상위 300개 스페이스 가져오기
-        }
-
+        progress(0, desc=f"Fetching {sort_type} spaces data...")
         response = requests.get(url, params=params)
         response.raise_for_status()
         all_spaces = response.json()
         
-        # 모든 스페이스의 순위 정보 저장
+        # 순위 정보 저장
         space_ranks = {space['id']: idx + 1 for idx, space in enumerate(all_spaces)}
         
-        # target_spaces에 있는 스페이스만 필터링하고 실제 순위 정보 포함
+        # target_spaces 필터링 및 순위 정보 포함
         spaces = []
         for space in all_spaces:
             if space.get('id', '') in target_spaces:
@@ -72,18 +74,21 @@ def get_trending_spaces(progress=gr.Progress()):
         
         progress(0.3, desc="Creating visualization...")
         
-        # 트렌드 시각화 생성
+        # 시각화 생성
         fig = go.Figure()
         
-        # 순위 데이터 준비
+        # 데이터 준비
         ids = [space['id'] for space in spaces]
         ranks = [space['rank'] for space in spaces]
         likes = [space.get('likes', 0) for space in spaces]
         
+        # Y축 값을 반전 (300 - rank + 1)
+        y_values = [301 - r for r in ranks]  # 순위를 반전된 값으로 변환
+        
         # 막대 그래프 생성
         fig.add_trace(go.Bar(
             x=ids,
-            y=ranks,
+            y=y_values,
             text=[f"Rank: {r}<br>Likes: {l}" for r, l in zip(ranks, likes)],
             textposition='auto',
             marker_color='rgb(158,202,225)',
@@ -92,7 +97,7 @@ def get_trending_spaces(progress=gr.Progress()):
         
         fig.update_layout(
             title={
-                'text': 'Hugging Face Spaces Rankings (Top 300)',
+                'text': f'Hugging Face Spaces {sort_type.title()} Rankings (Top 300)',
                 'y':0.95,
                 'x':0.5,
                 'xanchor': 'center',
@@ -100,7 +105,11 @@ def get_trending_spaces(progress=gr.Progress()):
             },
             xaxis_title='Space ID',
             yaxis_title='Rank',
-            yaxis_autorange='reversed',  # 순위가 위에서 아래로 내려가도록
+            yaxis=dict(
+                ticktext=[str(i) for i in range(1, 301, 20)],  # 1부터 300까지 20 간격
+                tickvals=[301 - i for i in range(1, 301, 20)],  # 반전된 값
+                range=[0, 300]  # y축 범위 설정
+            ),
             height=800,
             showlegend=False,
             template='plotly_white',
@@ -109,9 +118,10 @@ def get_trending_spaces(progress=gr.Progress()):
         
         progress(0.6, desc="Creating space cards...")
         
-        # 스페이스 카드 HTML 생성
-        html_content = """
+        # HTML 카드 생성
+        html_content = f"""
         <div style='padding: 20px; background: #f5f5f5;'>
+            <h2 style='color: #2c3e50;'>{sort_type.title()} Rankings</h2>
             <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
         """
         
@@ -149,12 +159,10 @@ def get_trending_spaces(progress=gr.Progress()):
                 </a>
             </div>
             """
-            progress((0.6 + 0.3 * spaces.index(space)/len(spaces)), 
-                    desc=f"Loading space {spaces.index(space)+1}/{len(spaces)}...")
         
         html_content += "</div></div>"
         
-        # 데이터 테이블 생성
+        # 데이터프레임 생성
         df = pd.DataFrame([{
             'Rank': space.get('rank', 'N/A'),
             'Space ID': space.get('id', ''),
@@ -171,7 +179,8 @@ def get_trending_spaces(progress=gr.Progress()):
         error_plot = create_error_plot()
         return error_plot, error_html, pd.DataFrame()
 
-# Gradio 인터페이스는 이전과 동일
+
+
 
 def create_trend_visualization(spaces_data):
     if not spaces_data:
@@ -379,35 +388,43 @@ def refresh_data():
     else:
         return create_error_plot(), "<div>API 인증이 필요합니다.</div>", pd.DataFrame()
 
-# Gradio 인터페이스 생성
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
-    # 🤗 HuggingFace Spaces Trending Analysis
-    실시간으로 Hugging Face Spaces의 트렌딩 순위를 분석합니다.
+    # 🤗 HuggingFace Spaces Rankings Analysis
+    실시간으로 Hugging Face Spaces의 순위를 분석합니다.
     """)
     
-    with gr.Tab("Trending Analysis"):
-        plot_output = gr.Plot()
-        info_output = gr.HTML()
+    with gr.Tab("Trending Rankings"):
+        trending_plot = gr.Plot()
+        trending_info = gr.HTML()
+        trending_df = gr.DataFrame()
     
-    with gr.Tab("Export Data"):
-        df_output = gr.DataFrame()
+    with gr.Tab("Most Liked Rankings"):
+        modes_plot = gr.Plot()
+        modes_info = gr.HTML()
+        modes_df = gr.DataFrame()
     
     refresh_btn = gr.Button("🔄 Refresh Data", variant="primary")
     
     def refresh_data():
-        return get_trending_spaces()
+        trending_results = get_spaces_data("trending")
+        modes_results = get_spaces_data("modes")
+        return [*trending_results, *modes_results]
     
     refresh_btn.click(
         refresh_data,
-        outputs=[plot_output, info_output, df_output]
+        outputs=[
+            trending_plot, trending_info, trending_df,
+            modes_plot, modes_info, modes_df
+        ]
     )
     
     # 초기 데이터 로드
-    initial_plot, initial_info, initial_df = get_trending_spaces()
-    plot_output.value = initial_plot
-    info_output.value = initial_info
-    df_output.value = initial_df
+    trending_results = get_spaces_data("trending")
+    modes_results = get_spaces_data("modes")
+    
+    trending_plot.value, trending_info.value, trending_df.value = trending_results
+    modes_plot.value, modes_info.value, modes_df.value = modes_results
 
 # Gradio 앱 실행
 demo.launch(