Upload 1_Data_Validation.py

pages/1_Data_Validation.py

@@ -6,6 +6,7 @@ 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
@@ -15,6 +16,8 @@ from st_aggrid import GridOptionsBuilder,GridUpdateMode
 from st_aggrid import GridOptionsBuilder
 from st_aggrid import AgGrid
 import base64
+import os
+import tempfile
 
 st.set_page_config(
   page_title="Data Validation",
@@ -95,6 +98,14 @@ if "selected_feature" not in st.session_state:
     st.session_state['selected_feature']=None
 
 
+def generate_report_with_target(channel_data, target_feature):
+        report = sv.analyze([channel_data, "Dataset"], target_feat=target_feature)
+        temp_dir = tempfile.mkdtemp()
+        report_path = os.path.join(temp_dir, "report.html")
+        report.show_html(filepath=report_path, open_browser=False)  # Generate the report as an HTML file
+        return report_path
+
+
 st.header('Univariate and Bivariate Analysis')
 eda_columns=st.columns(2)
 with eda_columns[0]:
@@ -111,13 +122,21 @@ with eda_columns[0]:
 
 with eda_columns[1]:
     if st.button('Generate Sweetviz Report'):
-    
-        def generate_report_with_target(df, target_feature):
-            report = sv.analyze([df, "Dataset"], target_feat=target_feature)
-            return report
-
-        report = generate_report_with_target(st.session_state['cleaned_data'], target_feature=target_column)
-        report.show_html()
+       with st.spinner('Generating Report'):     
+            report_file = generate_report_with_target(st.session_state['cleaned_data'], target_column)
+
+            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="report.html",
+                        mime="text/html"
+                    )
+            else:
+                st.warning("Report generation failed. Unable to find the report file.")
+            
 
 
 st.warning('Work in Progress')