import streamlit as st
import eda
import prediction
from streamlit_option_menu import option_menu

with open("./style.css") as f:
    st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)

st.sidebar.header("Heart Failure Classification")

with st.sidebar:
    st.write("Ediashta Revindra - FTDS-020")
    selected = option_menu(
        "Menu",
        [
            "Dataset",
            "Distribution",
            "Boxplot",
            "Correlation Matrix",
            "Prediksi Klasifikasi",
        ],
        icons=["database", "bar-chart", "box", "link-45deg", "code-square"],
        menu_icon="cast",
        default_index=1,
    )

if selected == "Distribution":
    eda.distribution()
elif selected == "Boxplot":
    eda.boxplot()
elif selected == "Dataset":
    eda.dataset()
elif selected == "Correlation Matrix":
    eda.corr_matrix()
elif selected == "Prediksi Klasifikasi":
    prediction.run()