Update app.py

app.py

@@ -60,7 +60,7 @@ coordinates = (selected_farm_data['longitude'].iloc[0], selected_farm_data['lati
 if st.button('نمایش نقشه NDRE'):
     NDRE = calculate_ndre(coordinates, start_date, end_date)
     if NDRE is not None:
-        st.session_state.ndre_value = NDRE
+        st.session_state.ndre_value = NDRE  # Store NDRE in session state
         st.write(f'شاخص NDRE: {NDRE}')  
         
         Map = geemap.Map()  
@@ -73,8 +73,18 @@ if st.button('نمایش نقشه NDRE'):
         st.error("Unable to calculate NDRE.")
 
 if st.button("Predict"):  
-    ndre_value = st.session_state.get('ndre_value', 0)  # Default to 0 if not calculated
+    # Retrieve NDRE value from session state, default to 0 if not set
+    ndre_value = st.session_state.get('ndre_value', 0)  
     user_input = pd.DataFrame({  
         'Farm_Name': [selected_farm],  
         'Farm_Age': [farm_age],
-        'Farm_Variety': [farm_variety],  
+        'Farm_Variety': [farm_variety],  
+        'NDRE': [ndre_value]  # Use the retrieved NDRE value
+    })  
+
+    # Feature Engineering might be needed here depending on your model's input features  
+    prediction = model.predict(user_input)  
+
+    st.write("Predictions:")  
+    st.write(f"Brix: {prediction[0][0]}")  # Assuming model outputs a list of lists  
+    st.write(f"Pol: