Update pages/1_Data_Validation.py

pages/1_Data_Validation.py

@@ -4,9 +4,7 @@ import plotly.express as px
 import plotly.graph_objects as go
 from Eda_functions import *
 import numpy as np
-import re
 import pickle
-from ydata_profiling import ProfileReport
 from streamlit_pandas_profiling import st_profile_report
 import streamlit as st
 import streamlit.components.v1 as components
@@ -18,6 +16,9 @@ from st_aggrid import AgGrid
 import base64
 import os
 import tempfile
+from ydata_profiling import ProfileReport
+
+from streamlit_pandas_profiling import st_profile_report
 
 st.set_page_config(
   page_title="Data Validation",
@@ -105,20 +106,32 @@ def generate_report_with_target(channel_data, target_feature):
         report.show_html(filepath=report_path, open_browser=False)  # Generate the report as an HTML file
         return report_path
 
+def generate_profile_report(df):
+    pr = df.profile_report()
+    temp_dir = tempfile.mkdtemp()
+    report_path = os.path.join(temp_dir, "report.html")
+    pr.to_file(report_path)
+    return report_path
+
 
 st.header('Univariate and Bivariate Analysis')
 eda_columns=st.columns(2)
 with eda_columns[0]:
     if st.button('Generate Profile Report'):
-        pr = st.session_state['cleaned_data'].profile_report()
+        with st.spinner('Generating Report'):     
+            report_file = generate_profile_report(st.session_state['cleaned_data'])
 
-        pr.to_file("Profile_Report.html")
-
-        with open("Profile_Report.html", "rb") as f:
-            profile_report_html = f.read()
-        b64 = base64.b64encode(profile_report_html).decode()
-        href = f'<a href="data:text/html;base64,{b64}" download="Profile_Report.html">Download Profile Report</a>'
-        st.markdown(href, unsafe_allow_html=True)
+            if os.path.exists(report_file):
+                with open(report_file, 'rb') as f:
+                    st.success('Report Generated')
+                    st.download_button(
+                        label="Download EDA Report",
+                        data=f.read(),
+                        file_name="pandas_profiling_report.html",
+                        mime="text/html"
+                    )
+            else:
+                st.warning("Report generation failed. Unable to find the report file.")
 
 with eda_columns[1]:
     if st.button('Generate Sweetviz Report'):
@@ -265,3 +278,4 @@ st.warning('Work in Progress')
 # #st.write('You selected:', selected_options)
 # st.pyplot(correlation_plot(df,selected_options,target_column))
 
+