"""This is an example of a simple chatbot that uses the RAG model to answer questions
about GIZ with Streamlit."""

import streamlit as st

# Here we import the rag_pipeline function from the rag.py file
from rag import rag_pipeline


# We use the st.cache decorator to load the RAG pipeline only once and cache it
@st.cache_resource
def load_rag_pipeline():
    rag = rag_pipeline()
    return rag


rag = load_rag_pipeline()

with st.chat_message(name="ai", avatar=":material/smart_toy:"):
    st.html(
        """
        <div style="display: flex; align-items: center; gap: 10px;">
            <span style="font-size: 36px;">🤖</span>
            <h3 style="margin: 0;">Welcome to the GIZ Gender Strategy Assistant</h3>
        </div>
        <p>This assistant provides <b>reliable answers</b> about the <b>GIZ Gender Strategy (2019)</b>, helping you quickly access key insights from the official document.</p>
        """
    )

with st.expander("📖 Further Information"):
    st.html(
        """
        <h4>💡 How does the app work?</h4>
        <p>The assistant uses a <b>Retrieval-Augmented Generation (RAG) approach</b> to ensure responses are grounded in the content of the <b>GIZ Gender Strategy (2019)</b>:</p>
        <ul>
            <li>Your question is <b>converted into an embedding</b> (numerical representation).</li>
            <li>The system <b>retrieves the most relevant text sections</b> from the strategy document.</li>
            <li>A <b>language model (LLM)</b> generates a response based on the retrieved content.</li>
        </ul>

        <p><b>⚠️ Important:</b> The assistant is <b>limited to the Gender Strategy (2019)</b>. It <b>does not</b> access external sources, additional policies, or updated guidelines beyond the provided document.</p>

        <h4>🎯 Example questions:</h4>
        <ul>
            <li>What are the key objectives of the Gender Strategy?</li>
            <li>How does GIZ define "gender," and what is the conceptual foundation of the strategy?</li>
            <li>How does the strategy align with the <b>2030 Agenda</b>?</li>
            <li>How is the success of the strategy measured and reviewed?</li>
        </ul>
        """
    )

st.html(
    """
    <h4>📌 Further resources:</h4>
    <ul>
        <li>📄 <a href="https://www.giz.de/de/downloads/giz-2019-de-genderstrategie.pdf" target="_blank"><b>GIZ Gender Strategy (2019, PDF)</b></a></li>
        <li>🌍 <a href="https://reporting.giz.de/2022/operating-responsibly/our-gender-strategy/index.html" target="_blank"><b>GIZ Gender Strategy Website</b></a></li>
        <li>📊 <a href="https://www.blog-datalab.com/" target="_blank"><b>GIZ Data Lab Blog</b></a></li>
    </ul>

    <p>Feel free to explore and gain deeper insights into GIZ’s commitment to gender equality! 🚀</p>
    <p><b>Now, go ahead and ask a question related to the GIZ Gender Strategy in the text field below!</b> 📝</p>
    """
)


if "messages" not in st.session_state:
    st.session_state.messages = []

for message in st.session_state.messages:
    with st.chat_message(name=message["role"], avatar=message["avatar"]):
        st.markdown(message["content"])

prompt = st.chat_input("Say something")
if prompt:
    with st.chat_message(name="user", avatar=":material/person:"):
        st.write(prompt)
        st.session_state.messages.append(
            {"role": "user", "content": prompt, "avatar": ":material/person:"}
        )
    with st.chat_message(name="ai", avatar=":material/smart_toy:"):
        result = rag.run(
            {"prompt_builder": {"query": prompt}, "text_embedder": {"text": prompt}},
        )
        result = result["llm"]["replies"][0]
        result = result.split("Question:")[0]
        st.write(result)
        st.session_state.messages.append(
            {"role": "ai", "content": result, "avatar": ":material/smart_toy:"}
        )