Update app.py

app.py

@@ -75,14 +75,25 @@ if st.button('نمایش نقشه NDRE'):
 if st.button("Predict"):  
     # Retrieve NDRE value from session state, default to 0 if not set
     ndre_value = st.session_state.get('ndre_value', 0)  
+    
+    # Prepare the user input DataFrame
     user_input = pd.DataFrame({  
-        'Farm_Name': [selected_farm],  
-        'Farm_Age': [farm_age],
-        'Farm_Variety': [farm_variety],  
+        'Age': [farm_age],  # Rename to match model's expected feature
+        'Variety': [farm_variety],  # Rename to match model's expected feature
         'NDRE': [ndre_value]  # Use the retrieved NDRE value
-    })  
-
-    # Feature Engineering might be needed here depending on your model's input features  
+    })
+    
+    # Additional features: calculate DayOfYear and Month from the start date
+    if start_date:
+        day_of_year = start_date.timetuple().tm_yday
+        month = start_date.month
+        user_input['DayOfYear'] = [day_of_year]
+        user_input['Month'] = [month]
+
+    # Reorder the columns to match the order expected by the model
+    user_input = user_input[['Age', 'DayOfYear', 'Month', 'Variety', 'NDRE']]
+
+    # Make predictions
     prediction = model.predict(user_input)  
 
     st.write("Predictions:")