Update app.py

app.py

@@ -20,7 +20,7 @@ df1 = pd.read_csv(f"https://docs.google.com/spreadsheets/d/{sheet_id}/export?for
 urls = df1["網址"].tolist()
 
 # Create a DataFrame to store all restaurant data
-df = pd.DataFrame(columns=["Store Name", "Address", "Latitude", "Longitude", "Region"])
+df = pd.DataFrame(columns=["Store Name", "Address", "Phone", "Latitude", "Longitude", "Region"])
 
 # Initialize Nominatim geocoder
 geolocator = Nominatim(user_agent="my_app")
@@ -56,6 +56,12 @@ def fetch_data():
         except AttributeError:
             address = None
             region = "Unknown"
+        
+        # Try to extract phone number
+        try:
+            phone = soup.find("a", {"data-event": "CTA_tel"}).get("href").replace("tel:", "")
+        except AttributeError:
+            phone = None
 
         try:
             location = geolocator.geocode(address)
@@ -72,6 +78,7 @@ def fetch_data():
         new_row = pd.DataFrame({
             "Store Name": [store_name],
             "Address": [address],
+            "Phone": [phone],  # Add phone number to the DataFrame
             "Latitude": [latitude],
             "Longitude": [longitude],
             "Region": [region]
@@ -133,7 +140,7 @@ if st.button("Fetch Restaurant Data"):
         if pd.notnull(row["Latitude"]) and pd.notnull(row["Longitude"]):
             folium.Marker(
                 location=[row["Latitude"], row["Longitude"]],
-                popup=row["Store Name"],
+                popup=f"{row['Store Name']} ({row['Phone']})",  # Show phone in the popup
                 tooltip=row["Address"]
             ).add_to(marker_cluster)