import streamlit as st

st.set_page_config(
    page_title = 'Patacognition',
    layout= 'wide',
    initial_sidebar_state = 'collapsed',
    menu_items = {
        "About" : 'Proyecto ideado para la investigación de "Clasificación de imágenes de una sola clase con algortimos de Inteligencia Artificial".',
        "Report a Bug" : 'mailto:contact@patacon.org'
    }
)

with st.sidebar:
    st.write("contact@patacon.org")

st.title("Resultados")
st.markdown(
f"""
### Se usaron 4 carpetas distintas que suman 900 archivos no vistos durante entrenamiento.
""")

cnn, vit, ml = st.tabs(["CNN", "Transformers Visuales y Zero-Shot", "Machine Learning Clásico"])

with cnn:
    st.image('/home/user/app/stats/cnn_stats.png')
with vit:
    st.image('/home/user/app/stats/vit_stats.png')
with ml:
    st.image('/home/user/app/stats/ml_stats.png')