Update app.py

app.py

@@ -1041,15 +1041,15 @@ elif app_mode == "Model Training":
 
     # In Model Training section - Fix indentation for training logic
    if st.button("Train Model"):
-    if not features:
-        st.error("Please select at least one feature.")
-        st.stop()
-
-    # INDENT ALL THIS CODE UNDER THE BUTTON CLICK
-    # Call the training function
-    model, scaler, label_encoder, imputer_numerical, metrics, column_order, importance, X_train, y_train = train_model(
-        df.copy(), target, features, problem_type, test_size, model_type, model_params, use_grid_search
-    )
+        if not features:
+            st.error("Please select at least one feature.")
+            st.stop()
+
+        # Indent all this code under the button click
+        # Call the training function
+        model, scaler, label_encoder, imputer_numerical, metrics, column_order, importance, X_train, y_train = train_model(
+            df.copy(), target, features, problem_type, test_size, model_type, model_params, use_grid_search
+        )
 
     if model:  # Only proceed if training was successful
         st.success("Model trained successfully!")
@@ -1080,6 +1080,8 @@ elif app_mode == "Model Training":
             st.metric("MSE", f"{metrics['mse']:.2f}")
             st.metric("R2", f"{metrics['r2']:.2f}")
 
+        # A
+
         # Additional model display code...