Update app.py

app.py

@@ -8,6 +8,7 @@ from sklearn.linear_model import LinearRegression
 from sklearn.tree import DecisionTreeClassifier
 from sklearn.ensemble import RandomForestClassifier
 from sklearn.preprocessing import StandardScaler
+from io import BytesIO
 
 # 讓使用者上傳 CSV 檔案
 uploaded_file = st.file_uploader("上傳一個 CSV 檔案", type="csv")
@@ -95,5 +96,20 @@ if uploaded_file is not None:
         st.write("### 特徵重要性數據表")
         st.dataframe(feature_importance)
 
+        # 讓使用者下載特徵重要性的 Excel 檔案
+        def to_excel(df):
+            output = BytesIO()
+            writer = pd.ExcelWriter(output, engine='xlsxwriter')
+            df.to_excel(writer, index=False, sheet_name='Feature Importance')
+            writer.save()
+            processed_data = output.getvalue()
+            return processed_data
+
+        excel_data = to_excel(feature_importance)
+        st.download_button(label='下載特徵重要性數據為 Excel 檔案',
+                           data=excel_data,
+                           file_name='feature_importance.xlsx',
+                           mime='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
+
     else:
         st.error("上傳的檔案中找不到 'target' 欄位，請確認檔案格式。")