Update src/streamlit_app.py

src/streamlit_app.py

@@ -483,12 +483,23 @@ with tabs[4]:
     
     # Select only valid feature columns
     cols_needed = [c for c in features if c in df.columns]
+    # --- Build sub_df safely (force exact column match) ---
     if target not in df.columns:
-        st.error(f"Target `{target}` not found in dataframe columns.")
-        st.stop()
+        # try case-insensitive or partial fallback once
+        matches = [c for c in df.columns if target.lower() in c.lower()]
+        if len(matches) == 1:
+            target = matches[0]
+            st.info(f"Auto-corrected target to exact match: `{target}`")
+        elif len(matches) > 1:
+            st.warning(f"Multiple columns match '{target}': {matches}. Using first: {matches[0]}")
+            target = matches[0]
+        else:
+            st.error(f"Target `{target}` not found in dataframe columns.")
+            st.stop()
     
-    # Build sub_df safely
-    sub_df = df[cols_needed + [target]].sample(n=sample_size, random_state=42).reset_index(drop=True)
+    # Now build sub_df strictly
+    sub_df = df.loc[:, cols_needed + [target]].sample(n=sample_size, random_state=42).reset_index(drop=True)
+
     
     # Construct X and y
     X = sub_df[cols_needed].copy()