Update app.py

app.py

@@ -1,12 +1,9 @@
 import gradio as gr
 import requests
 import pandas as pd
-import matplotlib.pyplot as plt
-import seaborn as sns
-from datetime import datetime, timedelta
 import plotly.graph_objects as go
-import numpy as np
-import json
+from datetime import datetime, timedelta
+import time
 
 # 관심 스페이스 URL 리스트와 정보
 target_spaces = {
@@ -43,48 +40,81 @@ target_spaces = {
     "NCSOFT/VARCO_Arena": "https://huggingface.co/spaces/NCSOFT/VARCO_Arena"
 }
 
+
 def get_trending_spaces(date):
-    url = f"https://huggingface.co/api/spaces/trending?date={date}&limit=300"
-    response = requests.get(url)
-    if response.status_code == 200:
-        return response.json()
-    return None
+    try:
+        url = f"https://huggingface.co/api/spaces/trending?date={date}&limit=300"
+        response = requests.get(url)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            print(f"API 요청 실패: {response.status_code}")
+            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():
-    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)
-    
-    return trending_data, target_space_ranks, dates
+    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="데이터를 불러오는데 실패했습니다.",
+        xref="paper",
+        yref="paper",
+        x=0.5,
+        y=0.5,
+        showarrow=False
+    )
+    return fig
 
 def create_trend_plot(trending_data, target_space_ranks, dates):
+    if not trending_data or not target_space_ranks or not dates:
+        return create_error_plot()
+    
     fig = go.Figure()
     
     for space_id, ranks in target_space_ranks.items():
-        fig.add_trace(go.Scatter(
-            x=dates,
-            y=ranks,
-            name=space_id,
-            mode='lines+markers'
-        ))
+        if ranks:  # 데이터가 있는 경우만 플롯
+            fig.add_trace(go.Scatter(
+                x=dates,
+                y=ranks,
+                name=space_id,
+                mode='lines+markers'
+            ))
     
     fig.update_layout(
         title='Trending Ranks Over Time',
@@ -97,64 +127,69 @@ def create_trend_plot(trending_data, target_space_ranks, dates):
     return fig
 
 def create_space_info_html(trending_data):
-    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
-
-def export_data(trending_data, dates):
-    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 not trending_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 spaces if s['id'] == space_id), None)
+                space_info = next((s for s in latest_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]
-                    })
+                    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>"
+
+def export_data(trending_data, dates):
+    if not trending_data or not dates:
+        return pd.DataFrame()
     
-    df = pd.DataFrame(df_data)
-    return df
+    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()
 
 def main_interface():
     trending_data, target_space_ranks, dates = fetch_and_analyze_data()
-    
-    # 트렌드 플롯 생성
     plot = create_trend_plot(trending_data, target_space_ranks, dates)
-    
-    # 스페이스 정보 HTML 생성
-    space_info = create_space_info_html(trending_data)
-    
-    # 데이터 익스포트
+    info = create_space_info_html(trending_data)
     df = export_data(trending_data, dates)
-    
-    return plot, space_info, df
+    return plot, info, df
 
 # Gradio 인터페이스 생성
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -168,16 +203,32 @@ 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(
-        main_interface,
+        refresh_data,
         outputs=[plot_output, info_output, df_output]
     )
     
     # 초기 데이터 로드
-    plot, info, df = main_interface()
-    plot_output.update(value=plot)
-    info_output.update(value=info)
-    df_output.update(value=df)
+    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())
 
 # Gradio 앱 실행
 demo.launch()