Update app.py

app.py

@@ -41,8 +41,12 @@ selected_df = load_data(urls[dataset_choice])
 st.write("### 爬取的資料預覽")
 st.dataframe(selected_df.head())
 
+# 過濾出數值類型的列，排除 '出表日期' 和 '報告年度'
+numeric_columns = selected_df.select_dtypes(include=['float64', 'int64']).columns
+numeric_columns = [col for col in numeric_columns if col not in ['出表日期', '報告年度']]
+
 # 允許用戶選擇用於繪製圖表的列
-column_choice = st.selectbox("選擇欄位來繪製圖表", selected_df.columns)
+column_choice = st.selectbox("選擇欄位來繪製圖表", numeric_columns)
 
 # 添加一個生成圖表的按鈕
 if st.button("生成圖表"):
@@ -52,39 +56,37 @@ if st.button("生成圖表"):
         time.sleep(0.01)
         progress_bar.progress(i + 1)
     
-    # 檢查該列是否為數值類型
-    if pd.api.types.is_numeric_dtype(selected_df[column_choice]):
-        # 創建一個標籤頁布局
-        tab1, tab2 = st.tabs(["圓餅圖", "長條圖"])
-        
-        with tab1:
-            # 使用 plotly 創建圓餅圖
-            fig_pie = px.pie(selected_df, names=selected_df.index, values=column_choice, 
-                             title=f"{dataset_choice} - {column_choice} 圓餅圖",
-                             color_discrete_sequence=px.colors.qualitative.Pastel)
-            fig_pie.update_traces(textposition='inside', textinfo='percent+label')
-            fig_pie.update_layout(
-                font=dict(size=14),
-                legend=dict(orientation="h", yanchor="bottom", y=1.02, xanchor="right", x=1)
-            )
-            st.plotly_chart(fig_pie, use_container_width=True)
-        
-        with tab2:
-            # 使用 plotly 創建長條圖
-            fig_bar = px.bar(selected_df, x=selected_df.index, y=column_choice, 
-                             title=f"{dataset_choice} - {column_choice} 長條圖",
-                             color_discrete_sequence=px.colors.qualitative.Pastel)
-            fig_bar.update_layout(
-                xaxis_title="企業",
-                yaxis_title=column_choice,
-                font=dict(size=14),
-                xaxis_tickangle=-45
-            )
-            st.plotly_chart(fig_bar, use_container_width=True)
-        
-        st.success("圖表生成完成！")
-    else:
-        st.error("選定的欄位不是數值類型，無法繪製圖表。")
+    # 創建一個標籤頁布局
+    tab1, tab2 = st.tabs(["圓餅圖", "長條圖"])
+    
+    with tab1:
+        # 使用 plotly 創建圓餅圖
+        fig_pie = px.pie(selected_df, names=selected_df['公司名稱'], values=column_choice, 
+                         title=f"{dataset_choice} - {column_choice} 圓餅圖",
+                         color_discrete_sequence=px.colors.qualitative.Pastel)
+        fig_pie.update_traces(textposition='inside', textinfo='percent+label')
+        fig_pie.update_layout(
+            font=dict(size=14),
+            legend=dict(orientation="h", yanchor="bottom", y=1.02, xanchor="right", x=1)
+        )
+        st.plotly_chart(fig_pie, use_container_width=True)
+    
+    with tab2:
+        # 使用 plotly 創建長條圖
+        fig_bar = px.bar(selected_df, x='公司名稱', y=column_choice, 
+                         title=f"{dataset_choice} - {column_choice} 長條圖",
+                         color='公司名稱',  # 為每個公司使用不同的顏色
+                         color_discrete_sequence=px.colors.qualitative.Pastel)
+        fig_bar.update_layout(
+            xaxis_title="企業",
+            yaxis_title=column_choice,
+            font=dict(size=14),
+            xaxis_tickangle=-45,
+            showlegend=False  # 隱藏圖例，因為顏色已經用於區分不同的公司
+        )
+        st.plotly_chart(fig_bar, use_container_width=True)
+    
+    st.success("圖表生成完成！")
 
 # 下載並設置自定義字體以顯示中文字符
 font_url = "https://drive.google.com/uc?id=1eGAsTN1HBpJAkeVM57_C7ccp7hbgSz3_&export=download"