Update app.py

app.py

@@ -3,19 +3,19 @@ import requests
 from bs4 import BeautifulSoup
 import pandas as pd
 import plotly.express as px
+import plotly.graph_objects as go
 
 # 定義爬取數據的函數
 def fetch_data(hospital_url, table_id, hospital_name):
     response = requests.get(hospital_url)
     soup = BeautifulSoup(response.text, 'html.parser')
     table = soup.find('table', {'id': table_id})
-    
     if not table:
         return pd.DataFrame()  # 若表格未找到，返回空的DataFrame
     
     rows = table.find_all('tr')
     data = []
-
+    
     if hospital_name == "成大醫院":
         # 成大醫院的表格結構不同
         columns = [th.text.strip() for th in rows[0].find_all("th")]
@@ -33,19 +33,9 @@ def fetch_data(hospital_url, table_id, hospital_name):
                 row_data = [col.get_text(strip=True) for col in cols]
                 if all(row_data):  # 確保數據不為空
                     data.append(row_data)
-
+    
     return pd.DataFrame(data, columns=columns)
 
-# 定義繪製圓餅圖的函數
-def plot_pie_chart(df):
-    fig = px.pie(df, names='病床種類', values='床位別數', title='各類病床床位佔比')
-    st.plotly_chart(fig)
-
-# 定義繪製柱狀圖的函數
-def plot_bar_chart(df):
-    fig = px.bar(df, x='病床種類', y='床位別數', color='醫院', barmode='group', title="各醫院病床分佈")
-    st.plotly_chart(fig)
-
 # Streamlit UI
 st.title("醫院床位分配表爬取工具")
 
@@ -71,26 +61,24 @@ selected_hospitals = st.multiselect("選擇醫院", list(hospital_options.keys()
 if st.button("爬取資料"):
     st.write("正在爬取資料...")
     progress_bar = st.progress(0)
-
     all_data = pd.DataFrame()
 
     for i, hospital_name in enumerate(selected_hospitals):
         hospital_data = hospital_options[hospital_name]
         df = fetch_data(hospital_data["url"], hospital_data["table_id"], hospital_name)
-
         if df.empty:
             st.warning(f"{hospital_name} 的數據爬取結果為空，請檢查是否存在問題。")
         else:
             df['醫院'] = hospital_name
             all_data = pd.concat([all_data, df], ignore_index=True)
-
+        
         # 更新進度條
         progress_bar.progress((i + 1) / len(selected_hospitals))
 
     if not all_data.empty:
         st.write("爬取完成，合併的數據如下：")
         st.dataframe(all_data)
-        
+
         # 下載數據為 CSV
         csv = all_data.to_csv(index=False).encode('utf-8-sig')
         st.download_button(
@@ -99,11 +87,22 @@ if st.button("爬取資料"):
             file_name='合併的床位分配資料.csv',
             mime='text/csv'
         )
-        
-        # 添加繪製按鈕
+
+        # 添加繪製圖表的按鈕
         if st.button("繪製圖表"):
-            plot_pie_chart(all_data)
-            plot_bar_chart(all_data)
+            # 準備數據
+            all_data['床位別數'] = pd.to_numeric(all_data['床位別數'], errors='coerce')
+            bed_counts = all_data.groupby(['醫院', '病床種類'])['床位別數'].sum().reset_index()
+            
+            # 繪製圓餅圖
+            fig_pie = px.pie(bed_counts, values='床位別數', names='病床種類', title='各類型病床分佈',
+                             hover_data=['醫院'], labels={'床位別數':'床位數'})
+            st.plotly_chart(fig_pie)
+
+            # 繪製柱狀圖
+            fig_bar = px.bar(bed_counts, x='醫院', y='床位別數', color='病床種類', title='醫院病床分佈',
+                             labels={'床位別數':'床位數'}, barmode='group')
+            st.plotly_chart(fig_bar)
 
     else:
-        st.error("沒有成功爬取任何數據。")
+        st.error("沒有成功爬取任何數據。")