Update app.py

app.py

@@ -729,6 +729,17 @@ with st.expander("🚀 Bulk Actions"):
         enhance_section_title("Cleaned Data Preview", "✨")
         with st.expander("✨ Cleaned Data Preview", expanded=True):
             st.dataframe(st.session_state.cleaned_data.head(), use_container_width=True)
+    
+                                                       enhance_section_title("Export Cleaned Data", "💾")
+    with st.expander("💾 Export Cleaned Data", expanded=True):
+        st.write("You can download the cleaned data as a CSV file.")
+        csv = df.to_csv(index=False).encode('utf-8')
+        st.download_button(
+            label="📥 Download Cleaned Data as CSV",
+            data=csv,
+            file_name='cleaned_data.csv',
+            mime='text/csv',
+        )
 
 
 
@@ -1117,27 +1128,39 @@ elif app_mode == "EDA":
             except Exception as e:
             st.error(f"Transformation failed: {str(e)}")
 
-    # --------------------------
-    # Export & Save
-    # --------------------------
     enhance_section_title("Export Options", "💾")
     st.subheader("💾 Export Options")
     col1, col2 = st.columns(2)
+    
+    # Download Current Visualization
     with col1:
-        if st.button("📥 Download Current Visualization"):
-            try:
-                fig.write_image("visualization.png")
-                st.success("Image saved!")
-            except NameError:
-                st.error("No visualization to download. Please create a chart first.")
+        if 'fig' in locals() or 'fig' in globals():
+            img_buffer = io.BytesIO()
+            fig.savefig(img_buffer, format='png')
+            img_buffer.seek(0)
+            st.download_button(
+                label="📥 Download Current Visualization",
+                data=img_buffer,
+                file_name="visualization.png",
+                mime="image/png",
+            )
+        else:
+            st.warning("No visualization to download. Please create a chart first.")
+    
+    # Export Analysis Report
     with col2:
-        if st.button("📊 Export Analysis Report"):
-            try:
-                profile = ProfileReport(df, minimal=True)
-                profile.to_file("analysis_report.html")
-                st.success("Report generated!")
-            except Exception as e:
-                st.error(f"Could not generate analysis report. Ensure pandas-profiling is installed correctly.")
+        try:
+            profile = ProfileReport(df, minimal=True)
+            profile_html = profile.to_html()
+            st.download_button(
+                label="📊 Export Analysis Report",
+                data=profile_html,
+                file_name="analysis_report.html",
+                mime="text/html",
+            )
+        except Exception as e:
+            st.error(f"Could not generate analysis report. Ensure pandas-profiling is installed correctly.")
+
 
 elif app_mode == "Model Training":
     st.title("🧠 Advanced Model Architect")