Update app.py

app.py

@@ -2,16 +2,16 @@ import requests
 from bs4 import BeautifulSoup
 import pandas as pd
 import folium
-from folium.plugins import MarkerCluster
+from folium.plugins import MarkerCluster, HeatMap
 import plotly.graph_objects as go
-import plotly.express as px  # Add this import
+import plotly.express as px
 from geopy.geocoders import Nominatim
 import re
 import streamlit as st
 
 # Streamlit title and description
 st.title("米其林餐廳指南爬蟲")
-st.write("Extract restaurant data, visualize with a pie chart and bar chart, and display locations on a map.")
+st.write("Extract restaurant data, visualize with a Sunburst chart and bar chart, and display locations on a map with heatmap.")
 
 # Read data from Google Sheets
 sheet_id = "1xUfnD1WCF5ldqECI8YXIko1gCpaDDCwTztL17kjI42U"
@@ -113,25 +113,22 @@ if st.button("爬取餐廳資料"):
     # Group the data by region and sum the number of restaurants
     region_group = df.groupby("Region").size().reset_index(name='Count')
 
-    # Plot enlarged pie chart with custom colors and labels
-    pie_chart = go.Figure(go.Pie(
-        labels=region_group["Region"],
-        values=region_group["Count"],
-        textinfo="label+percent",
-        hoverinfo="label+value",
-        textfont=dict(size=18),
-        marker=dict(colors=px.colors.qualitative.Set3, line=dict(color='#000000', width=2))
-    ))
-
-    pie_chart.update_layout(
+    # Plot Sunburst chart
+    sunburst = px.sunburst(
+        region_group,
+        path=['Region'],
+        values='Count',
         title="Restaurant Distribution by Region",
+    )
+
+    sunburst.update_layout(
         title_x=0.5,
         title_font=dict(size=24, family="Arial"),
         height=600,
         margin=dict(t=50, b=50, l=50, r=50)
     )
-    st.subheader("Restaurant Distribution by Region (Enlarged Pie Chart)")
-    st.plotly_chart(pie_chart)
+    st.subheader("Restaurant Distribution by Region (Sunburst Chart)")
+    st.plotly_chart(sunburst)
 
     # Plot bar chart with custom colors and labels
     bar_chart = go.Figure(go.Bar(
@@ -156,13 +153,17 @@ if st.button("爬取餐廳資料"):
     st.plotly_chart(bar_chart)
 
     # Display a map using Folium
-    st.subheader("Restaurant Locations Map")
+    st.subheader("Restaurant Locations Map with Heatmap")
 
     # Create map centered around Tainan
     m = folium.Map(location=[23.0, 120.2], zoom_start=12)
 
     # Add marker cluster to the map
     marker_cluster = MarkerCluster().add_to(m)
+    
+    # Prepare data for heatmap
+    heat_data = []
+    
     for index, row in df.iterrows():
         if pd.notnull(row["Latitude"]) and pd.notnull(row["Longitude"]):
             folium.Marker(
@@ -170,6 +171,10 @@ if st.button("爬取餐廳資料"):
                 popup=f"{row['Store Name']} ({row['Phone']})",
                 tooltip=row["Address"]
             ).add_to(marker_cluster)
+            heat_data.append([row["Latitude"], row["Longitude"]])
+
+    # Add heatmap layer
+    HeatMap(heat_data).add_to(m)
 
     # Display the map in Streamlit
-    st.components.v1.html(m._repr_html_(), height=600)
+    st.components.v1.html(m._repr_html_(), height=600)