import streamlit as st
import pandas as pd
import joblib

pipe = joblib.load('wine_quality_prediction.pkl')

def get_user_input():
    input_dict = {}
    for feat in selected_features:
        input_value = st.number_input(f"Enter {feat} value", value=0.0, step=0.01)
        input_dict[feat] = input_value
    return pd.DataFrame([input_dict])


st.title("Wine Quality Prediction")

user_input = get_user_input()

prediction = pipe.predict(user_input)

st.subheader("Prediction")
st.write(prediction[0])