Update app.py

app.py

@@ -9,17 +9,17 @@ from pytrends.request import TrendReq
 import plotly.graph_objs as go
 import plotly.express as px
 
-# 新增函數：準備 sunburst 圖的數據
+# 新增函數：準備sunburst圖的數據
 def prepare_sunburst_data(df):
     # 將價格分成不同區間
     df['price_range'] = pd.cut(df['price'], bins=[0, 10000, 20000, 30000, 40000, float('inf')],
                                labels=['0-10k', '10k-20k', '20k-30k', '30k-40k', '40k+'])
     
-    # 計算每個來源、價格區間和商品名稱的商品數量
-    sunburst_data = df.groupby(['source', 'price_range', 'title']).size().reset_index(name='count')
+    # 計算每個來源和價格區間的商品數量
+    sunburst_data = df.groupby(['source', 'price_range']).size().reset_index(name='count')
     
     # 添加一個根節點
-    root = pd.DataFrame({'source': ['Total'], 'price_range': [''], 'title': [''], 'count': [sunburst_data['count'].sum()]})
+    root = pd.DataFrame({'source': ['Total'], 'price_range': [''], 'count': [sunburst_data['count'].sum()]})
     sunburst_data = pd.concat([root, sunburst_data])
     
     return sunburst_data
@@ -234,9 +234,9 @@ if st.button("開始搜索"):
 
     # 準備並繪製 Sunburst 圖
     sunburst_data = prepare_sunburst_data(combined_data)
-    fig5 = px.sunburst(sunburst_data, path=['source', 'price_range', 'title'], values='count',
+    fig5 = px.sunburst(sunburst_data, path=['source', 'price_range'], values='count',
                        title=f'"{search_keyword}" 價格分佈 Sunburst 圖')
     st.plotly_chart(fig5)
 
     # 添加互動性說明
-    st.write("提示：點擊 Sunburst 圖的各個部分可以深入查看細節。雙擊可以返回上一層。")
+    st.write("提示：點擊 Sunburst 圖的各個部分可以深入查看細節。雙擊可以返回上一層。")