import streamlit as st
import pandas as pd

data_df = pd.DataFrame(
    {
        "apps": [
            "https://d1afc1j4569hs1.cloudfront.net/Autobahn.jpeg",
        ],
    }
)


st.set_page_config(
    page_title="P&ID Object Detection ",
    layout="wide",
    initial_sidebar_state="expanded"
)

st.title('P&ID Object Detection App')
st.subheader(' Identify valves and pumps with deep learning model', divider='rainbow')
st.caption('Developed by Deep Drawings Co.')

col1, col2, col3 = st.columns([10, 10, 10])
with col2:
    with st.expander('How to use it'):
        st.markdown(
        '''
        1) Upload your P&ID or Select Test Diagrams 📬
        2) Set Confidence Threshold 📈
        3) Press to Perform Inference 🚀
        4) Visualize Model Predictions 🔎
        '''
    )
        
    st.data_editor(
        data_df,
        column_config={
            "apps": st.column_config.ImageColumn(
                "Preview Image", help="Streamlit app preview screenshots"
            )
        },
        hide_index=True,
    )   

st.write('##')
    
col1, col2, col3 = st.columns([10, 10, 10])
with col1:
    st.markdown(f"##### Upload your File:")
    # set input image by upload
    image_file = st.file_uploader("Upload your diagram:", type=["pdf"])