Update app.py

app.py

@@ -4,6 +4,9 @@ from bs4 import BeautifulSoup
 import pandas as pd
 import plotly.express as px
 import base64
+import folium
+from streamlit_folium import st_folium
+from geopy.geocoders import Nominatim
 
 # Function to set background image
 def set_background(png_file):
@@ -35,6 +38,7 @@ min_rating = st.slider("請輸入想查詢的最低評分:", 1.0, 5.0, 3.5)
 urls = [
     "https://www.tw-animal.com/pet/171211/c000196.html",
     "https://www.tw-animal.com/pet/171211/c000186.html",
+    "https://www.tw-animal.com/pet/171211/c000186.html",
     "https://www.tw-animal.com/pet/171211/c000081.html",
     "https://www.tw-animal.com/pet/171211/c001166.html",
     "https://www.tw-animal.com/pet/171211/c000773.html",
@@ -56,9 +60,13 @@ urls = [
     "https://www.tw-animal.com/pet/171211/c001252.html"
 ]
 
+
 # Create an empty list to store the extracted data
 data_list = []
 
+# Initialize the geolocator
+geolocator = Nominatim(user_agent="geoapiExercises")
+
 # Scrape data when the button is pressed
 if st.button('開始爬取資料'):
     st.write("正在爬取資料，請稍候...")
@@ -76,12 +84,20 @@ if st.button('開始爬取資料'):
 
         # Append the data to the list if rating meets the threshold
         if rating >= min_rating:
-            data_list.append({
-                "標題": title,
-                "手機": phone,
-                "地址": address,
-                "評分": rating
-            })
+            try:
+                # Geocode the address to get the latitude and longitude
+                location = geolocator.geocode(address)
+                if location:
+                    data_list.append({
+                        "標題": title,
+                        "手機": phone,
+                        "地址": address,
+                        "評分": rating,
+                        "經度": location.longitude,
+                        "緯度": location.latitude
+                    })
+            except Exception as e:
+                st.warning(f"無法獲取經緯度：{e}")
 
     # If data was scraped successfully
     if data_list:
@@ -109,6 +125,23 @@ if st.button('開始爬取資料'):
         pie_fig = px.pie(grouped_df, names='區域', values='評分', title="各區域寵物醫院比例")
         st.plotly_chart(pie_fig)
 
+        # Display the map
+        if st.button('顯示地圖'):
+            # Create a folium map centered around the average location
+            map_center = [df1['緯度'].mean(), df1['經度'].mean()]
+            pet_map = folium.Map(location=map_center, zoom_start=12)
+
+            # Add markers for each hospital
+            for index, row in df1.iterrows():
+                folium.Marker(
+                    location=[row['緯度'], row['經度']],
+                    popup=f"{row['標題']} (評分: {row['評分']})",
+                    tooltip=row['標題']
+                ).add_to(pet_map)
+
+            # Render the map using streamlit_folium
+            st_folium(pet_map, width=700, height=500)
+
         # Sending notification to LINE
         if st.button('發送前五筆資料到Line'):
             msg = df1[:5].to_string(index=False)
@@ -128,4 +161,3 @@ if st.button('開始爬取資料'):
             st.success('資料已成功發送到 Line!')
     else:
         st.warning('沒有符合條件的資料。')
-