Update app.py

app.py

@@ -62,7 +62,8 @@ def enhance_section_title(title, icon="✨"):
 def update_cleaned_data(df):
     """Updates the cleaned data in session state."""
     st.session_state.cleaned_data = df
-    st.session_state.data_versions.append(df.copy())  # Append to history
+    if 'data_versions' in st.session_state:
+        st.session_state.data_versions.append(df.copy())
     st.success("Action completed successfully!")
 
 def generate_quality_report(df):
@@ -359,9 +360,6 @@ with st.sidebar:
     st.markdown("Created by Calvin Allen-Crawford")
     st.markdown("v1.0 | © 2025")
 
-# --------------------------
-# Main App Pages
-# --------------------------
 # --------------------------
 # Main App Pages
 # --------------------------
@@ -623,6 +621,18 @@ elif app_mode == "EDA":
 
     if st.session_state.cleaned_data is None:
         st.warning("Please clean your data first")
+            # Show Upload Clean Data button
+        uploaded_clean_file = st.file_uploader("Upload your cleaned dataset (CSV/XLSX)", type=["csv", "xlsx"])
+        if uploaded_clean_file:
+            try:
+                if uploaded_clean_file.name.endswith('.csv'):
+                    df = pd.read_csv(uploaded_clean_file)
+                else:
+                    df = pd.read_excel(uploaded_clean_file)
+                st.session_state.cleaned_data = df
+                st.success("Cleaned data uploaded successfully!")
+            except Exception as e:
+                st.error(f"Error loading file: {str(e)}")
         st.stop()
 
     df = st.session_state.cleaned_data
@@ -849,6 +859,18 @@ elif app_mode == "Model Training":
 
     if st.session_state.get("cleaned_data") is None:
         st.warning("Please clean your data first")
+            # Show Upload Clean Data button
+        uploaded_clean_file = st.file_uploader("Upload your cleaned dataset (CSV/XLSX)", type=["csv", "xlsx"])
+        if uploaded_clean_file:
+            try:
+                if uploaded_clean_file.name.endswith('.csv'):
+                    df = pd.read_csv(uploaded_clean_file)
+                else:
+                    df = pd.read_excel(uploaded_clean_file)
+                st.session_state.cleaned_data = df
+                st.success("Cleaned data uploaded successfully!")
+            except Exception as e:
+                st.error(f"Error loading file: {str(e)}")
         st.stop()
 
     df = st.session_state.cleaned_data