Update app.py

app.py

@@ -6,24 +6,28 @@ import requests
 from io import BytesIO
 
 def load_and_process_data():
-    # Hugging Face 데이터셋에서 parquet 파일 다운로드
-    url = "https://huggingface.co/datasets/cfahlgren1/hub-stats/resolve/main/spaces.parquet"
-    response = requests.get(url)
-    df = pd.read_parquet(BytesIO(response.content))
-    
-    # 30일 전 날짜 계산
-    thirty_days_ago = datetime.now() - timedelta(days=30)
-    
-    # SQL 쿼리와 동일한 처리
-    df['createdAt'] = pd.to_datetime(df['createdAt'])
-    filtered_df = df[df['createdAt'] >= thirty_days_ago].copy()
-    filtered_df['created'] = filtered_df['createdAt'].dt.date
-    filtered_df = filtered_df.sort_values('trendingScore', ascending=False)
-    
-    return filtered_df.head(100)
+    try:
+        # Hugging Face 데이터셋에서 parquet 파일 다운로드
+        url = "https://huggingface.co/datasets/cfahlgren1/hub-stats/resolve/main/spaces.parquet"
+        response = requests.get(url)
+        df = pd.read_parquet(BytesIO(response.content))
+        
+        # 30일 전 날짜 계산
+        thirty_days_ago = datetime.now() - timedelta(days=30)
+        
+        # SQL 쿼리와 동일한 처리
+        df['createdAt'] = pd.to_datetime(df['createdAt'])
+        filtered_df = df[df['createdAt'] >= thirty_days_ago].copy()
+        filtered_df['created'] = filtered_df['createdAt'].dt.date
+        filtered_df = filtered_df.sort_values('trendingScore', ascending=False)
+        
+        return filtered_df.head(100)
+    except Exception as e:
+        print(f"Error loading data: {e}")
+        return pd.DataFrame()
 
 def create_trend_chart(selected_id, df):
-    if not selected_id:
+    if selected_id is None or df.empty:
         return None
         
     space_data = df[df['id'] == selected_id]
@@ -50,19 +54,27 @@ def create_trend_chart(selected_id, df):
     return fig
 
 def update_display(selected_id, df):
-    if selected_id is None:
-        return None, ""
+    if selected_id is None or df.empty:
+        return None, "No data selected"
     
-    space_data = df[df['id'] == selected_id].iloc[0]
-    info_text = f"""ID: {space_data['id']}
-Created At: {space_data['createdAt'].strftime('%Y-%m-%d')}
-Trending Score: {space_data['trendingScore']:.2f}"""
-    
-    chart = create_trend_chart(selected_id, df)
-    
-    return chart, info_text
+    try:
+        space_data = df[df['id'] == selected_id]
+        if space_data.empty:
+            return None, "Space not found"
+            
+        space_info = space_data.iloc[0]
+        info_text = f"""ID: {space_info['id']}
+Created At: {space_info['createdAt'].strftime('%Y-%m-%d')}
+Trending Score: {space_info['trendingScore']:.2f}"""
+        
+        chart = create_trend_chart(selected_id, df)
+        
+        return chart, info_text
+    except Exception as e:
+        print(f"Error updating display: {e}")
+        return None, f"Error: {str(e)}"
 
-# 데이터 로드
+# 전역 변수로 데이터프레임 로드
 df = load_and_process_data()
 
 # Gradio 인터페이스 생성
@@ -72,12 +84,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     with gr.Row():
         # 왼쪽 패널 - 스페이스 리스트
         with gr.Column(scale=1):
+            if not df.empty:
+                choices = [(row['id'], f"{row['id']} (Score: {row['trendingScore']:.2f})") 
+                          for _, row in df.iterrows()]
+                default_value = df['id'].iloc[0]
+            else:
+                choices = []
+                default_value = None
+                
             space_list = gr.Dropdown(
-                choices=[(row['id'], f"{row['id']} (Score: {row['trendingScore']:.2f})") 
-                        for _, row in df.iterrows()],
+                choices=choices,
                 label="Select a Space",
                 info="Click to select a space and view its trend",
-                value=df['id'].iloc[0] if not df.empty else None
+                value=default_value
             )
             
             # 스페이스 정보 표시
@@ -95,12 +114,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             )
     
     # 이벤트 핸들러
+    def wrapped_update_display(selected_id):
+        return update_display(selected_id, df)
+    
     space_list.change(
-        fn=update_display,
+        fn=wrapped_update_display,
         inputs=[space_list],
         outputs=[trend_plot, info_box]
     )
 
 # 대시보드 실행
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)  # share=True 추가