Update app.py

app.py

@@ -40,156 +40,136 @@ target_spaces = {
     "NCSOFT/VARCO_Arena": "https://huggingface.co/spaces/NCSOFT/VARCO_Arena"
 }
 
-
-def get_trending_spaces(date):
+def get_trending_spaces():
     try:
-        url = f"https://huggingface.co/api/spaces/trending?date={date}&limit=300"
-        response = requests.get(url)
+        url = "https://huggingface.co/api/spaces/sort/trending"
+        headers = {
+            'Accept': 'application/json',
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
+        }
+        response = requests.get(url, headers=headers)
+        
         if response.status_code == 200:
             return response.json()
         else:
             print(f"API 요청 실패: {response.status_code}")
+            print(f"Response: {response.text}")
             return None
     except Exception as e:
         print(f"API 호출 중 에러 발생: {str(e)}")
         return None
 
-def get_space_rank(spaces, space_id):
-    if not spaces:
-        return None
-    for idx, space in enumerate(spaces, 1):
-        if space.get('id', '') == space_id:
-            return idx
-    return None
-
-def fetch_and_analyze_data():
-    try:
-        start_date = datetime(2023, 12, 1)
-        end_date = datetime(2023, 12, 31)
-        dates = [(start_date + timedelta(days=x)).strftime('%Y-%m-%d') 
-                for x in range((end_date - start_date).days + 1)]
-        
-        trending_data = {}
-        target_space_ranks = {space: [] for space in target_spaces.keys()}
-        
-        for date in dates:
-            spaces = get_trending_spaces(date)
-            if spaces:
-                trending_data[date] = spaces
-                for space_id in target_spaces.keys():
-                    rank = get_space_rank(spaces, space_id)
-                    target_space_ranks[space_id].append(rank)
-            time.sleep(0.5)  # API 요청 간 딜레이 추가
-        
-        if not trending_data:
-            return create_error_plot(), "데이터를 불러오는데 실패했습니다.", pd.DataFrame()
-        
-        return trending_data, target_space_ranks, dates
-    except Exception as e:
-        print(f"데이터 수집 중 에러 발생: {str(e)}")
-        return {}, {}, []
-
 def create_error_plot():
     fig = go.Figure()
     fig.add_annotation(
-        text="데이터를 불러오는데 실패했습니다.",
+        text="데이터를 불러올 수 없습니다.\n잠시 후 다시 시도해주세요.",
         xref="paper",
         yref="paper",
         x=0.5,
         y=0.5,
-        showarrow=False
+        showarrow=False,
+        font=dict(size=20)
+    )
+    fig.update_layout(
+        title="Error Loading Data",
+        height=400
     )
     return fig
 
-def create_trend_plot(trending_data, target_space_ranks, dates):
-    if not trending_data or not target_space_ranks or not dates:
+def create_trend_visualization(spaces_data):
+    if not spaces_data:
         return create_error_plot()
     
     fig = go.Figure()
     
-    for space_id, ranks in target_space_ranks.items():
-        if ranks:  # 데이터가 있는 경우만 플롯
-            fig.add_trace(go.Scatter(
-                x=dates,
-                y=ranks,
-                name=space_id,
-                mode='lines+markers'
-            ))
+    # 타겟 스페이스들의 순위 찾기
+    ranks = []
+    for idx, space in enumerate(spaces_data, 1):
+        space_id = space.get('id', '')
+        if space_id in target_spaces:
+            ranks.append({
+                'id': space_id,
+                'rank': idx,
+                'likes': space.get('likes', 0)
+            })
+    
+    # 순위별로 정렬
+    ranks.sort(key=lambda x: x['rank'])
+    
+    # 플롯 데이터 생성
+    ids = [r['id'] for r in ranks]
+    rank_values = [r['rank'] for r in ranks]
+    likes = [r['likes'] for r in ranks]
+    
+    fig.add_trace(go.Bar(
+        x=ids,
+        y=rank_values,
+        text=likes,
+        textposition='auto',
+        name='Rank'
+    ))
     
     fig.update_layout(
-        title='Trending Ranks Over Time',
-        xaxis_title='Date',
+        title='Current Trending Ranks',
+        xaxis_title='Space ID',
         yaxis_title='Rank',
         yaxis_autorange='reversed',
-        height=800
+        height=600,
+        showlegend=False
     )
     
     return fig
 
-def create_space_info_html(trending_data):
-    if not trending_data:
+def create_space_info_html(spaces_data):
+    if not spaces_data:
         return "<div style='padding: 20px;'><h2>데이터를 불러오는데 실패했습니다.</h2></div>"
     
-    try:
-        latest_date = max(trending_data.keys())
-        latest_spaces = trending_data[latest_date]
-        
-        html_content = "<div style='padding: 20px;'>"
-        html_content += f"<h2>Latest Rankings ({latest_date})</h2>"
-        
-        for space_id, url in target_spaces.items():
-            rank = get_space_rank(latest_spaces, space_id)
-            if rank:
-                space_info = next((s for s in latest_spaces if s['id'] == space_id), None)
-                if space_info:
-                    html_content += f"""
-                    <div style='margin: 20px 0; padding: 15px; border: 1px solid #ddd; border-radius: 8px;'>
-                        <h3>#{rank} - {space_id}</h3>
-                        <p>👍 Likes: {space_info.get('likes', 'N/A')}</p>
-                        <p>📝 {space_info.get('title', 'N/A')}</p>
-                        <p>{space_info.get('description', 'N/A')[:100]}...</p>
-                        <a href='{url}' target='_blank' style='color: blue;'>Visit Space 🔗</a>
-                    </div>
-                    """
-        
-        html_content += "</div>"
-        return html_content
-    except Exception as e:
-        return f"<div style='padding: 20px;'><h2>데이터 처리 중 에러 발생: {str(e)}</h2></div>"
+    html_content = "<div style='padding: 20px;'>"
+    html_content += "<h2>Current Rankings</h2>"
+    
+    for idx, space in enumerate(spaces_data, 1):
+        space_id = space.get('id', '')
+        if space_id in target_spaces:
+            html_content += f"""
+            <div style='margin: 20px 0; padding: 15px; border: 1px solid #ddd; border-radius: 8px;'>
+                <h3>#{idx} - {space_id}</h3>
+                <p>👍 Likes: {space.get('likes', 'N/A')}</p>
+                <p>📝 {space.get('title', 'N/A')}</p>
+                <p>{space.get('description', 'N/A')[:100]}...</p>
+                <a href='{target_spaces[space_id]}' target='_blank' style='color: blue;'>Visit Space 🔗</a>
+            </div>
+            """
+    
+    html_content += "</div>"
+    return html_content
 
-def export_data(trending_data, dates):
-    if not trending_data or not dates:
+def create_data_table(spaces_data):
+    if not spaces_data:
         return pd.DataFrame()
     
-    try:
-        df_data = []
-        for date in dates:
-            spaces = trending_data.get(date, [])
-            for space_id in target_spaces.keys():
-                rank = get_space_rank(spaces, space_id)
-                if rank:
-                    space_info = next((s for s in spaces if s['id'] == space_id), None)
-                    if space_info:
-                        df_data.append({
-                            'Date': date,
-                            'Space ID': space_id,
-                            'Rank': rank,
-                            'Likes': space_info.get('likes', 'N/A'),
-                            'Title': space_info.get('title', 'N/A'),
-                            'URL': target_spaces[space_id]
-                        })
-        
-        return pd.DataFrame(df_data)
-    except Exception as e:
-        print(f"데이터 익스포트 중 에러 발생: {str(e)}")
-        return pd.DataFrame()
+    rows = []
+    for idx, space in enumerate(spaces_data, 1):
+        space_id = space.get('id', '')
+        if space_id in target_spaces:
+            rows.append({
+                'Rank': idx,
+                'Space ID': space_id,
+                'Likes': space.get('likes', 'N/A'),
+                'Title': space.get('title', 'N/A'),
+                'URL': target_spaces[space_id]
+            })
+    
+    return pd.DataFrame(rows)
 
-def main_interface():
-    trending_data, target_space_ranks, dates = fetch_and_analyze_data()
-    plot = create_trend_plot(trending_data, target_space_ranks, dates)
-    info = create_space_info_html(trending_data)
-    df = export_data(trending_data, dates)
-    return plot, info, df
+def refresh_data():
+    spaces_data = get_trending_spaces()
+    if spaces_data:
+        plot = create_trend_visualization(spaces_data)
+        info = create_space_info_html(spaces_data)
+        df = create_data_table(spaces_data)
+        return plot, info, df
+    else:
+        return create_error_plot(), "<div>데이터를 불러오는데 실패했습니다.</div>", pd.DataFrame()
 
 # Gradio 인터페이스 생성
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -203,32 +183,16 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         df_output = gr.DataFrame()
     
     refresh_btn = gr.Button("Refresh Data")
-    
-    def refresh_data():
-        try:
-            return main_interface()
-        except Exception as e:
-            return (
-                create_error_plot(),
-                f"<div style='padding: 20px;'><h2>데이터 새로고침 중 에러 발생: {str(e)}</h2></div>",
-                pd.DataFrame()
-            )
-    
     refresh_btn.click(
         refresh_data,
         outputs=[plot_output, info_output, df_output]
     )
     
     # 초기 데이터 로드
-    try:
-        plot, info, df = main_interface()
-        plot_output.update(value=plot)
-        info_output.update(value=info)
-        df_output.update(value=df)
-    except Exception as e:
-        plot_output.update(value=create_error_plot())
-        info_output.update(value=f"<div style='padding: 20px;'><h2>초기 데이터 로드 중 에러 발생: {str(e)}</h2></div>")
-        df_output.update(value=pd.DataFrame())
+    initial_plot, initial_info, initial_df = refresh_data()
+    plot_output.value = initial_plot
+    info_output.value = initial_info
+    df_output.value = initial_df
 
 # Gradio 앱 실행
 demo.launch()