Update app.py

app.py

@@ -12,6 +12,9 @@ def load_and_process_data():
         response = requests.get(url)
         df = pd.read_parquet(BytesIO(response.content))
         
+        # 데이터 로딩 확인
+        print("Initial data shape:", df.shape)
+        
         # 30일 전 날짜 계산
         thirty_days_ago = datetime.now() - timedelta(days=30)
         
@@ -21,17 +24,17 @@ def load_and_process_data():
         filtered_df['created'] = filtered_df['createdAt'].dt.date
         filtered_df = filtered_df.sort_values('trendingScore', ascending=False)
         
+        # 필터링된 데이터 확인
+        print("Filtered data shape:", filtered_df.shape)
+        print("Sample data:", filtered_df.head(1))
+        
         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 selected_id is None or df.empty:
-        return None
-        
-    space_data = df[df['id'] == selected_id]
-    if space_data.empty:
+def create_trend_chart(space_data):
+    if space_data is None or space_data.empty:
         return None
         
     # 선택된 space의 트렌드 차트 생성
@@ -39,7 +42,7 @@ def create_trend_chart(selected_id, df):
         space_data,
         x='created',
         y='trendingScore',
-        title=f'Trending Score for {selected_id}',
+        title=f'Trending Score for {space_data.iloc[0]["id"]}',
         labels={'created': 'Date', 'trendingScore': 'Trending Score'}
     )
     
@@ -53,29 +56,40 @@ def create_trend_chart(selected_id, df):
     
     return fig
 
-def update_display(selected_id, df):
-    if selected_id is None or df.empty:
-        return None, "No data selected"
+def update_display(selected_id):
+    global df  # 전역 데이터프레임 사용
     
+    if selected_id is None:
+        return None, "Please select a space"
+        
     try:
+        # 선택된 ID로 데이터 필터링
         space_data = df[df['id'] == selected_id]
+        print(f"Selected ID: {selected_id}")
+        print(f"Found data: {not space_data.empty}")
+        
         if space_data.empty:
-            return None, "Space not found"
+            return None, "Space not found in dataset"
             
+        # 데이터가 존재하는 경우 정보 생성
         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)
+        # 차트 생성
+        chart = create_trend_chart(space_data)
         
         return chart, info_text
+        
     except Exception as e:
-        print(f"Error updating display: {e}")
-        return None, f"Error: {str(e)}"
+        print(f"Error in update_display: {e}")
+        return None, f"Error processing data: {str(e)}"
 
 # 전역 변수로 데이터프레임 로드
+print("Loading initial data...")
 df = load_and_process_data()
+print("Data loaded. Shape:", df.shape)
 
 # Gradio 인터페이스 생성
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -85,19 +99,22 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         # 왼쪽 패널 - 스페이스 리스트
         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_choices = df['id'].tolist()
+                default_value = space_choices[0] if space_choices else None
                 
-            space_list = gr.Dropdown(
-                choices=choices,
-                label="Select a Space",
-                info="Click to select a space and view its trend",
-                value=default_value
-            )
+                space_list = gr.Dropdown(
+                    choices=space_choices,
+                    label="Select a Space",
+                    value=default_value,
+                    info="Select a space to view its trend"
+                )
+            else:
+                space_list = gr.Dropdown(
+                    choices=[],
+                    label="Select a Space",
+                    info="No data available"
+                )
             
             # 스페이스 정보 표시
             info_box = gr.Textbox(
@@ -114,15 +131,12 @@ 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=wrapped_update_display,
+        fn=update_display,
         inputs=[space_list],
         outputs=[trend_plot, info_box]
     )
 
 # 대시보드 실행
 if __name__ == "__main__":
-    demo.launch(share=True)  # share=True 추가
+    demo.launch(share=True)