Update app.py

app.py

@@ -752,6 +752,12 @@ elif app_mode == "Model Training":
         Choose from a wide range of algorithms and configurations.
     """)
 
+    # Initialize session state variables if they don't exist
+    if 'model' not in st.session_state:
+        st.session_state.model = None
+    if 'preprocessor' not in st.session_state:
+        st.session_state.preprocessor = None
+
     if 'cleaned_data' not in st.session_state or st.session_state.cleaned_data is None:
         st.warning("Please clean your data in the Smart Cleaning section first.")
         st.stop()
@@ -963,8 +969,7 @@ elif app_mode == "Model Training":
             except Exception as e:
                 st.error(f"An error occurred during training: {e}")
 
-    # Model Saving
-    if st.session_state.model is not None:
+     if st.session_state.model is not None:
         st.subheader("💾 Save Model")
         model_filename = st.text_input("Enter Model Filename (without extension)", "trained_model")
         if st.button("Save Model"):
@@ -973,6 +978,8 @@ elif app_mode == "Model Training":
                 st.success(f"Model saved as {model_filename}.joblib")
             except Exception as e:
                 st.error(f"Error saving model: {e}")
+    else:
+        st.warning("No trained model available. Train a model first to enable saving.")
 
 # Visualization Lab Section
 elif app_mode == "Visualization Lab":