Update app.py

app.py

@@ -149,7 +149,6 @@ st.markdown("""
         transform: translateY(-2px);
         box-shadow: 0 6px 16px rgba(59, 130, 246, 0.4);
     }
-
     /* Card styles */
     .card {
         background: var(--white);
@@ -1123,6 +1122,16 @@ def model_training_page():
                 st.session_state.selected_columns = df.columns.tolist()
         else:
             # For clustering, all columns are features
+            available_features = df.columns.tolist()
+            selected_features = st.multiselect(
+                "Select features for clustering", 
+                options=available_features,
+                default=available_features[:min(5, len(available_features))]
+            )
+            st.session_state.selected_columns = selected_features
+    
+    # Rest of the model_training_page function would go here...
+    st.markdown('</div>', unsafe_allow_html=True)
 
 def visualization_page():
     """Visualization and evaluation page for trained models"""