Update app.py

app.py

@@ -25,7 +25,8 @@ from stqdm import stqdm
 def enhance_section_title(title, icon="✨"):
     """Helper function to create a styled section title with an icon."""
     st.markdown(f"<h2 style='border-bottom: 2px solid #ccc; padding-bottom: 5px;'>{icon} {title}</h2>", unsafe_allow_html=True)
-
+    
+@st.cache_data
 def update_cleaned_data(df):
     """Updates the cleaned data in session state."""
     st.session_state.cleaned_data = df
@@ -330,43 +331,6 @@ with st.sidebar:
 if app_mode == "Data Upload":
     st.title("📤 Data Upload & Profiling")
 
-    uploaded_file = st.file_uploader("Upload your dataset (CSV/XLSX)", type=["csv", "xlsx"])
-
-    if uploaded_file:
-        try:
-            if uploaded_file.name.endswith('.csv'):
-                df = pd.read_csv(uploaded_file)
-            else:
-                df = pd.read_excel(uploaded_file)
-
-            st.session_state.raw_data = df
-
-            col1, col2, col3 = st.columns(3)
-            with col1:
-                st.metric("Rows", df.shape[0])
-            with col2:
-                st.metric("Columns", df.shape[1])
-            with col3:
-                st.metric("Missing Values", df.isna().sum().sum())
-
-            with st.expander("Data Preview", expanded=True):
-                st.dataframe(df.head(10), use_container_width=True)
-
-            if st.button("Generate Full Profile Report"):
-                with st.spinner("Generating comprehensive analysis..."):
-                    pr = ProfileReport(df, explorative=True)
-                    st_profile_report(pr)
-
-        except Exception as e:
-            st.error(f"Error loading file: {str(e)}")
-
-# --------------------------
-# Page Content
-# --------------------------
-# Data Upload Section
-if app_mode == "Data Upload":
-    st.title("📤 Data Upload & Profiling")
-
     uploaded_file = st.file_uploader("Upload your dataset (CSV/XLSX)", type=["csv", "xlsx"], key="file_uploader")
 
     if uploaded_file:
@@ -413,10 +377,8 @@ elif app_mode == "Data Cleaning":
         st.warning("No cleaned data available. Please clean your data first.")
         st.stop()  # Stop execution if no cleaned data is available
 
-
     # Data Health Dashboard
     enhance_section_title("Data Health Dashboard", "📊")
-
     with st.expander("📊 Data Health Dashboard", expanded=True):
         col1, col2, col3 = st.columns(3)
         with col1:
@@ -432,14 +394,6 @@ elif app_mode == "Data Cleaning":
                 profile = ProfileReport(df, minimal=True)
                 st_profile_report(profile)
 
-    # Undo Functionality
-    if len(st.session_state.data_versions) > 1:
-        if st.button("⏮️ Undo Last Action"):
-            st.session_state.data_versions.pop()  # Remove current version
-            st.session_state.cleaned_data = st.session_state.data_versions[-1].copy()  # Set data
-            st.success("Last action undone!")
-            st.experimental_rerun()  # Force re-run after undo
-
     # Missing Value Handling
     enhance_section_title("Missing Values Treatment", "🔍")
     with st.expander("🔍 Missing Values Treatment", expanded=True):