Update app.py

app.py

@@ -205,6 +205,25 @@ st.pyplot(plt)
 # Prediction
 st.subheader('Predict Median House Value')
 
+# Input values for prediction
+input_values = {}
+for feature in X.columns:
+    input_values[feature] = st.number_input(f'Enter {feature}', value=float(df[feature].mean()))
+
+# Show regression line if selected
+show_regression = st.checkbox('Show Regression Line')
+if show_regression and selected_feature in X.columns and selected_target == 'MedHouseVal':
+    X_feature = df[[selected_feature]]
+    model_feature = LinearRegression()
+    model_feature.fit(X_feature, y)
+    line = model_feature.predict(X_feature)
+    ax.plot(df[selected_feature], line, color='red', linewidth=2)
+
+st.pyplot(fig)
+
+# Prediction
+st.subheader('Predict Median House Value')
+
 # Input values for prediction
 input_values = {}
 for feature in X.columns:
@@ -215,6 +234,4 @@ if st.button('Predict'):
     prediction = model.predict(input_data)
     st.write(f'Predicted Median House Value: {prediction[0]}')
 
-# Display data
-if st.checkbox('Show raw data'):
-    st.write(df)
+