Update app.py

app.py

@@ -14,26 +14,40 @@ def load_and_process_data():
         # 30일치 데이터 준비
         thirty_days_ago = datetime.now() - timedelta(days=30)
         df['createdAt'] = pd.to_datetime(df['createdAt'])
+        df = df[df['createdAt'] >= thirty_days_ago].copy()
         
-        # 모든 날짜에 대한 데이터 생성
+        # 날짜별 데이터 처리
         dates = pd.date_range(start=thirty_days_ago, end=datetime.now(), freq='D')
-        all_spaces = df['id'].unique()
-        
-        # 일자별 순위 계산을 위한 데이터프레임 생성
         daily_ranks = []
         
         for date in dates:
-            date_data = df[df['createdAt'].dt.date <= date.date()]
-            date_ranks = date_data.sort_values('trendingScore', ascending=False)
-            date_ranks['rank'] = range(1, len(date_ranks) + 1)
-            date_ranks['date'] = date.date()
-            daily_ranks.append(date_ranks[['id', 'date', 'rank', 'trendingScore', 'createdAt']])
+            # 해당 날짜의 데이터 추출
+            date_data = df[df['createdAt'].dt.date <= date.date()].copy()
+            
+            # trendingScore가 같은 경우 id로 정렬하여 유니크한 순위 보장
+            date_data = date_data.sort_values(['trendingScore', 'id'], ascending=[False, True])
+            
+            # 순위 계산
+            date_data['rank'] = range(1, len(date_data) + 1)
+            date_data['date'] = date.date()
+            
+            # 필요한 컬럼만 선택
+            daily_ranks.append(
+                date_data[['id', 'date', 'rank', 'trendingScore', 'createdAt']]
+            )
         
-        daily_ranks_df = pd.concat(daily_ranks)
+        # 전체 데이터 병합
+        daily_ranks_df = pd.concat(daily_ranks, ignore_index=True)
         
-        # 최신 순위로 정렬된 상위 100개 공간
+        # 최신 날짜의 top 100 추출
         latest_date = daily_ranks_df['date'].max()
-        top_100_spaces = daily_ranks_df[daily_ranks_df['date'] == latest_date].head(100)
+        top_100_spaces = daily_ranks_df[
+            daily_ranks_df['date'] == latest_date
+        ].sort_values('rank').head(100).copy()
+        
+        print(f"Total records: {len(daily_ranks_df)}")
+        print(f"Unique spaces: {len(daily_ranks_df['id'].unique())}")
+        print(f"Date range: {daily_ranks_df['date'].min()} to {daily_ranks_df['date'].max()}")
         
         return daily_ranks_df, top_100_spaces
     except Exception as e:
@@ -43,50 +57,60 @@ def load_and_process_data():
 def create_trend_chart(space_id, daily_ranks_df):
     if space_id is None or daily_ranks_df.empty:
         return None
+    
+    try:
+        # 특정 space의 데이터만 필터링
+        space_data = daily_ranks_df[daily_ranks_df['id'] == space_id].copy()
+        if space_data.empty:
+            return None
+        
+        # 데이터 정렬
+        space_data = space_data.sort_values('date')
         
-    space_data = daily_ranks_df[daily_ranks_df['id'] == space_id].copy()
-    if space_data.empty:
+        fig = px.line(
+            space_data,
+            x='date',
+            y='rank',
+            title=f'Daily Rank Trend for {space_id}',
+            labels={'date': 'Date', 'rank': 'Rank'},
+            markers=True
+        )
+        
+        fig.update_layout(
+            xaxis_title="Date",
+            yaxis_title="Rank",
+            yaxis_autorange="reversed",  # 순위 1이 위쪽에 오도록
+            hovermode='x unified',
+            plot_bgcolor='white',
+            paper_bgcolor='white'
+        )
+        
+        return fig
+    except Exception as e:
+        print(f"Error creating chart: {e}")
         return None
-    
-    # 순위를 역순으로 변경 (차트에서 위로 갈수록 좋은 순위)
-    max_rank = daily_ranks_df.groupby('date')['rank'].transform('max')
-    space_data['rank_reversed'] = max_rank - space_data['rank'] + 1
-    
-    fig = px.line(
-        space_data,
-        x='date',
-        y='rank_reversed',
-        title=f'Daily Rank Trend for {space_id}',
-        labels={'date': 'Date', 'rank_reversed': 'Rank'},
-        markers=True
-    )
-    
-    fig.update_layout(
-        xaxis_title="Date",
-        yaxis_title="Rank",
-        yaxis_autorange="reversed",  # 순위 1이 위쪽에 오도록
-        hovermode='x unified',
-        plot_bgcolor='white',
-        paper_bgcolor='white'
-    )
-    
-    return fig
 
-def update_display(selected_id):
+def update_display(selection):
     global daily_ranks_df
     
-    if selected_id is None:
+    if not selection:
         return None, "Please select a space"
-        
+    
     try:
-        latest_data = daily_ranks_df[daily_ranks_df['id'] == selected_id].sort_values('date').iloc[-1]
+        # 선택된 항목에서 space ID 추출
+        space_id = selection.split(': ')[1].split(' (Score')[0]
+        
+        # 최신 데이터 가져오기
+        latest_data = daily_ranks_df[
+            daily_ranks_df['id'] == space_id
+        ].sort_values('date').iloc[-1]
         
-        info_text = f"""ID: {latest_data['id']}
+        info_text = f"""ID: {space_id}
 Current Rank: {int(latest_data['rank'])}
 Trending Score: {latest_data['trendingScore']:.2f}
 Created At: {latest_data['createdAt'].strftime('%Y-%m-%d')}"""
         
-        chart = create_trend_chart(selected_id, daily_ranks_df)
+        chart = create_trend_chart(space_id, daily_ranks_df)
         
         return chart, info_text
         
@@ -97,7 +121,7 @@ Created At: {latest_data['createdAt'].strftime('%Y-%m-%d')}"""
 # 데이터 로드
 print("Loading initial data...")
 daily_ranks_df, top_100_spaces = load_and_process_data()
-print("Data loaded. Shape:", daily_ranks_df.shape)
+print("Data loaded.")
 
 # Gradio 인터페이스 생성
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -107,8 +131,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=1):
             # 순위가 포함된 리스트로 표시
             space_choices = [
-                f"Rank {idx+1}: {row['id']} (Score: {row['trendingScore']:.2f})"
-                for idx, (_, row) in enumerate(top_100_spaces.iterrows())
+                f"Rank {row['rank']}: {row['id']} (Score: {row['trendingScore']:.2f})"
+                for _, row in top_100_spaces.iterrows()
             ]
             
             space_list = gr.Radio(
@@ -130,15 +154,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 label="Daily Rank Trend"
             )
     
-    def handle_selection(selection):
-        if selection:
-            # 선택된 항목에서 space ID 추출
-            space_id = selection.split(': ')[1].split(' (Score')[0]
-            return update_display(space_id)
-        return None, "Please select a space"
-    
     space_list.change(
-        fn=handle_selection,
+        fn=update_display,
         inputs=[space_list],
         outputs=[trend_plot, info_box]
     )