Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import requests
 from bs4 import BeautifulSoup
 import pandas as pd
+import plotly.express as px
 
 # 定義爬取數據的函數
 def fetch_data(hospital_url, table_id, hospital_name):
@@ -35,6 +36,11 @@ def fetch_data(hospital_url, table_id, hospital_name):
 
     return pd.DataFrame(data, columns=columns)
 
+# 定義繪圖函數
+def plot_data(df):
+    fig = px.bar(df, x='病床種類', y='床位別數', color='醫院', barmode='group', title="各醫院床位分配情況")
+    st.plotly_chart(fig)
+
 # Streamlit UI
 st.title("醫院床位分配表爬取工具")
 
@@ -88,5 +94,10 @@ if st.button("爬取資料"):
             file_name='合併的床位分配資料.csv',
             mime='text/csv'
         )
+        
+        # 添加繪製按鈕
+        if st.button("繪製圖表"):
+            plot_data(all_data)
+
     else:
         st.error("沒有成功爬取任何數據。")