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


st.title("Customer Churn Classification")

st.sidebar.header("Final Project - Hacktiv8")
with st.sidebar:
    st.write("Ediashta Revindra - FTDS-020")
    selected = option_menu(
        "Menu",
        [
            "Distribution",
            "Classification",
            "Cluster"
        ],
        icons=["bar-chart", "link-45deg", "code-square"],
        menu_icon="cast",
        default_index=0,
    )

if selected == "Distribution":
    eda.distribution()
elif selected == "Classification":
    prediction.predict()
elif selected == "Cluster":
    prediction.cluster()