import pandas as pd
import streamlit as st
import prediction

# Set the page title and favicon
st.set_page_config(page_title="Notes Text Classification", 
                layout='wide',
                initial_sidebar_state='expanded'
)

# Create a sidebar with a title and a selection box
st.sidebar.title("Choose a page:")
page = st.sidebar.selectbox("", ('Landing Page' , 'Data Prediction'))

# Display different content depending on the selected page
if page == 'Data Prediction':
    prediction.run()
else:
    # Add a header and a subheader with some text
    st.title("What category does this note belong to?")
    st.subheader("Find out the category with this space that uses NLP to do predictions.")

    # Add an image about the case
    st.image("https://imageio.forbes.com/specials-images/imageserve/60808d87824ab7edc3770486/Note-Pad-and-Pen-on-Yellow-background/960x0.jpg?height=474&width=711&fit=bounds")