Update app.py

app.py

@@ -7,48 +7,64 @@ from io import BytesIO
 
 def load_and_process_data():
     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))
         
-        # 데이터 로딩 확인
-        print("Initial data shape:", df.shape)
-        
-        # 30일 전 날짜 계산
+        # 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)
         
-        # 필터링된 데이터 확인
-        print("Filtered data shape:", filtered_df.shape)
-        print("Sample data:", filtered_df.head(1))
+        # 모든 날짜에 대한 데이터 생성
+        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']])
         
-        return filtered_df.head(100)
+        daily_ranks_df = pd.concat(daily_ranks)
+        
+        # 최신 순위로 정렬된 상위 100개 공간
+        latest_date = daily_ranks_df['date'].max()
+        top_100_spaces = daily_ranks_df[daily_ranks_df['date'] == latest_date].head(100)
+        
+        return daily_ranks_df, top_100_spaces
     except Exception as e:
         print(f"Error loading data: {e}")
-        return pd.DataFrame()
+        return pd.DataFrame(), pd.DataFrame()
 
-def create_trend_chart(space_data):
-    if space_data is None or space_data.empty:
+def create_trend_chart(space_id, daily_ranks_df):
+    if space_id is None or daily_ranks_df.empty:
         return None
         
-    # 선택된 space의 트렌드 차트 생성
+    space_data = daily_ranks_df[daily_ranks_df['id'] == space_id].copy()
+    if space_data.empty:
+        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='created',
-        y='trendingScore',
-        title=f'Trending Score for {space_data.iloc[0]["id"]}',
-        labels={'created': 'Date', 'trendingScore': 'Trending Score'}
+        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="Trending Score",
+        yaxis_title="Rank",
+        yaxis_autorange="reversed",  # 순위 1이 위쪽에 오도록
         hovermode='x unified',
         plot_bgcolor='white',
         paper_bgcolor='white'
@@ -57,28 +73,20 @@ def create_trend_chart(space_data):
     return fig
 
 def update_display(selected_id):
-    global df  # 전역 데이터프레임 사용
+    global daily_ranks_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}")
+        latest_data = daily_ranks_df[daily_ranks_df['id'] == selected_id].sort_values('date').iloc[-1]
         
-        if space_data.empty:
-            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}"""
+        info_text = f"""ID: {latest_data['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(space_data)
+        chart = create_trend_chart(selected_id, daily_ranks_df)
         
         return chart, info_text
         
@@ -86,57 +94,54 @@ Trending Score: {space_info['trendingScore']:.2f}"""
         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)
+daily_ranks_df, top_100_spaces = load_and_process_data()
+print("Data loaded. Shape:", daily_ranks_df.shape)
 
 # Gradio 인터페이스 생성
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# Trending Spaces Dashboard")
     
     with gr.Row():
-        # 왼쪽 패널 - 스페이스 리스트
         with gr.Column(scale=1):
-            if not df.empty:
-                # 드롭다운 선택지 생성
-                space_choices = df['id'].tolist()
-                default_value = space_choices[0] if space_choices else None
-                
-                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"
-                )
+            # 순위가 포함된 리스트로 표시
+            space_choices = [
+                f"Rank {idx+1}: {row['id']} (Score: {row['trendingScore']:.2f})"
+                for idx, (_, row) in enumerate(top_100_spaces.iterrows())
+            ]
+            
+            space_list = gr.Radio(
+                choices=space_choices,
+                label="Top 100 Trending Spaces",
+                info="Select a space to view its rank trend",
+                value=space_choices[0] if space_choices else None
+            )
             
-            # 스페이스 정보 표시
             info_box = gr.Textbox(
                 label="Space Details",
                 value="",
                 interactive=False,
-                lines=3
+                lines=4
             )
         
-        # 오른쪽 패널 - 트렌드 차트
         with gr.Column(scale=2):
             trend_plot = gr.Plot(
-                label="Trending Score Over Time"
+                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=update_display,
+        fn=handle_selection,
         inputs=[space_list],
         outputs=[trend_plot, info_box]
     )
 
-# 대시보드 실행
 if __name__ == "__main__":
     demo.launch(share=True)