Update app.py

app.py

@@ -42,11 +42,18 @@ def generate_plots(df, df_name, selected_columns, chart_type):
         st.error("❌ 缺少 公司名稱 欄位，請確認數據格式！")
         return
 
+    # **檢查選擇的列是否存在於 DataFrame 中**
+    valid_columns = [col for col in selected_columns if col in df.columns]
+
+    # **如果有缺失的列，顯示錯誤信息**
+    missing_columns = set(selected_columns) - set(valid_columns)
+    if missing_columns:
+        st.error(f"❌ 缺少以下欄位，請確認數據格式！: {missing_columns}")
+        return
+
     # ✅ **確保所有 y 軸數據與 x 軸 (公司名稱) 長度一致**
-    df = df.dropna(subset=["公司名稱"] + selected_columns)  # **移除 NaN 確保長度相同**
-    df[selected_columns] = df[selected_columns].apply(pd.to_numeric, errors='coerce')  # **轉換數據類型**
-    
-    valid_columns = [col for col in selected_columns if col in df.columns and len(df[col]) == len(df["公司名稱"])]
+    df = df.dropna(subset=["公司名稱"] + valid_columns)  # **移除 NaN 確保長度相同**
+    df[valid_columns] = df[valid_columns].apply(pd.to_numeric, errors='coerce')  # **轉換數據類型**
     
     if not valid_columns:
         st.error(f"❌ 選擇的欄位長度與 公司名稱 不匹配，請檢查數據！\n"