Update app.py

app.py

@@ -3,6 +3,7 @@ import streamlit as st
 import folium
 from streamlit_folium import st_folium
 from model import predict_metals
+import pandas as pd
 
 st.set_page_config(page_title="GeoMetals Predictor", layout="wide")
 
@@ -30,7 +31,13 @@ stormwater = st.radio("Stormwater Runoff Present?", ("Yes", "No")) == "Yes"
 
 # Submit and predict
 if st.button("🔍 Predict Heavy Metal Concentrations") and lat and lon:
-    prediction = predict_metals(lat, lon, stormwater)
+    stormwater_type = "stormwater" if stormwater else "no_stormwater"
+    input_data = pd.DataFrame([{
+        "lat": lat,
+        "lon": lon,
+        "type": stormwater_type
+    }])
+    prediction = predict_metals(input_data)
 
     st.subheader("🔬 Predicted Heavy Metal Concentrations (ppm)")
     for metal, value in prediction.items():