import streamlit as st

# Import functions from eda.py and predict.py
from prediction import make_prediction

def main():
    st.sidebar.title("Navigation")
    selection = st.sidebar.radio("Select", ["Home", "EDA", "Prediction"])

    if selection == "Home":
        st.title("Welcome to our Laptop Price Predictor!")
        st.subheader("",divider = 'gray')
        st.write(
            "This application is build so you don't have to do too much time on researching laptop prices. "
            "Just tell me the specs you want and we'll tell you the price estimation!"
        )

        st.subheader("Prediction Model")
        st.write(
            "The Prediction section allows you to input your dream laptop specs. "
            "By entering details about the laptop, we will give you a price estimation!"
        )
        st.markdown(
            """
            <style>
            .highlight {
                color: #FF5733;
                font-weight: bold;
            }
            </style>
            """,
            unsafe_allow_html=True
        )
        st.markdown('<p class="highlight">Make informed decisions to enhance your hotel management!</p>', unsafe_allow_html=True)

    elif selection == "Prediction":
        make_prediction()  # Call prediction function

if __name__ == "__main__":
    main()