Update app.py

app.py

@@ -5,11 +5,12 @@ from bs4 import BeautifulSoup
 from google.oauth2.service_account import Credentials
 import gspread
 import plotly.express as px
+import json  # 用於解析 JSON 文件
 
 # Streamlit UI 標題
 st.title("Booking.com 多項目數據抓取和視覺化展示")
 
-# Step 1: 多 URL 抓取 Booking.com 數據
+# 定義 URL 列表
 urls = [
     "https://www.booking.com/hotel/it/appartamento-via-genova-roma.zh-tw.html",
     "https://www.booking.com/hotel/it/giulia-39-s-coliseum.zh-tw.html",
@@ -32,7 +33,7 @@ headers = {
     "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36"
 }
 
-# 抓取每個 URL 的標題和評分
+# 定義抓取數據函數
 def fetch_booking_data(url, headers):
     try:
         response = requests.get(url, headers=headers)
@@ -45,38 +46,44 @@ def fetch_booking_data(url, headers):
         return None
 
 # 逐一處理 URL 並收集數據
+st.header("抓取數據")
 data = []
 for url in urls:
     result = fetch_booking_data(url, headers)
     if result:
         data.append(result)
 
-# 將數據轉換為 DataFrame
+# 將數據轉換為 DataFrame 並顯示
 if data:
     df = pd.DataFrame(data)
     st.write("抓取到的數據：", df)
 else:
     st.warning("無法從任何 URL 抓取數據。")
 
-# Step 2: 上傳數據至 Google Sheets
+# 定義上傳 Google Sheets 函數
 def upload_to_google_sheet(df, spreadsheet_url, creds_file_content):
     try:
+        # 將金鑰內容轉換為字典
+        creds_dict = json.loads(creds_file_content)
+        # 創建憑據
         scope = ['https://www.googleapis.com/auth/spreadsheets']
-        creds = Credentials.from_service_account_info(creds_file_content, scopes=scope)
+        creds = Credentials.from_service_account_info(creds_dict, scopes=scope)
+        # 授權並打開 Google Sheets
         gs = gspread.authorize(creds)
         sheet = gs.open_by_url(spreadsheet_url)
         worksheet = sheet.get_worksheet(0)
+        # 清除並更新數據
         worksheet.clear()
         worksheet.update([df.columns.values.tolist()] + df.values.tolist())
-        st.success("Data uploaded to Google Sheets successfully!")
+        st.success("數據已成功上傳到 Google Sheets！")
     except Exception as e:
-        st.error(f"Error uploading data: {e}")
+        st.error(f"數據上傳失敗：{e}")
 
 # Google Sheets URL 和金鑰文件上傳
+st.header("上傳到 Google Sheets")
 spreadsheet_url = st.text_input("Google Sheets URL", "https://docs.google.com/spreadsheets/d/1iOzoii9bVAmqlcqnseoqjZBkBuaFcpbIvUxZeRJ2kmk/edit?gid=0#gid=0")
 creds_file = st.file_uploader("上傳 Google API 金鑰檔案", type=["json"])
 
-# 上傳至 Google Sheets
 if st.button("上傳數據至 Google Sheets") and data:
     if creds_file is not None:
         creds_content = creds_file.read()  # 讀取上傳的文件內容
@@ -84,11 +91,11 @@ if st.button("上傳數據至 Google Sheets") and data:
     else:
         st.error("請上傳 Google API 金鑰檔案")
 
-# Step 3: 顯示抓取的數據為動態圖表
+# 顯示動態圖表
+st.header("數據視覺化")
 if not df.empty:
     # 確保評分可以轉換為數字
     df['評分'] = pd.to_numeric(df['評分'], errors='coerce')
-    
     # 使用 Plotly 繪製條形圖
     fig = px.bar(df, x='標題', y='評分', title="Booking.com 評分比較", labels={'標題': '酒店標題', '評分': '評分'})
     fig.update_layout(xaxis_title="酒店標題", yaxis_title="評分", xaxis_tickangle=-45)