import streamlit as st
from chroma_utils import ask_chroma

st.title("Encuentra en el Manual")

question = st.text_input("Haz una pregunta para encontrar en el manual")

if question:
    results = ask_chroma(question, k=5)
    for i, result in enumerate(results):
        # st.page_link(result.metadata['url'],
        #              label=f"{result.metadata['title']}", help=f"{result.page_content[:250]} ...", )
        with st.container(border=True):
            st.markdown(
                f"## [{result.metadata['title']}]({result.metadata['url']}) ")
            if len(result.page_content) > 500:
                st.markdown(f"{result.page_content[:500]}...")
            else:
                st.markdown(result.page_content)
        # with st.expander(f"**Titulo:** {result.metadata['title']}"):
        #     st.markdown(f"**Resumen:** {result.page_content[:250]} ...")
        #     st.markdown(
        #         f"**Enlace:** [{result.metadata['title']}]({result.metadata['url']})")