Update app.py

app.py

@@ -72,13 +72,17 @@ if uploaded_file is not None:
         st.write("Cleaned Dataset:")
         st.dataframe(df)
 
+        # Add clean data download option
+        st.subheader("Download Cleaned Dataset")
+
+        # Provide a download button for the cleaned dataset
         st.download_button(
-    label="Download Cleaned Dataset",
-    data=to_excel(df),  # The cleaned dataset is 'df'
-    file_name="cleaned_dataset.xlsx",
-    mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
-)
-        
+        label="Download Cleaned Dataset (CSV)",
+        data=df.to_csv(index=False),  # Convert the cleaned dataset to CSV
+        file_name="cleaned_dataset.csv",  # Specify the file name
+        mime="text/csv"  # Specify the MIME type for CSV
+        )
+
         target = st.selectbox("Select Target Variable", df.columns)
         features = [col for col in df.columns if col != target]
         X = df[features]