import streamlit as st
from langchain_groq import ChatGroq
from apps.agent.constant import GROQ_API_KEY, MODEL_GROQ, CONFIG
from apps.agent.graph import Agent


llm = ChatGroq(model=MODEL_GROQ, api_key=GROQ_API_KEY, temperature=0.1)


agent = Agent(llm=llm)


def get_response(query: str):
    response = agent.graph.invoke({"messages": ("user", query)}, CONFIG)
    return response["messages"][-1].content

with st.sidebar:
    st.header("Prof of Concept")
    st.markdown(
        """
        This is just a prototype chatbot, the data taken is based on the following sites:
        Xano Documentation
        - https://docs.xano.com/about
        - https://releases.xano.com/?_gl=1*sifgtw*_ga*MTI5NTY3MTk5NS4xNzMwNjMzNjY3*_ga_EJWDZRK3CG*MTczMDgwNjg3Mi43LjEuMTczMDgwNjkyMy45LjAuODUyNzA5OTA4
        - https://docs.xano.com/onboarding-tutorial-reference
        - https://docs.xano.com/faq
        - https://docs.xano.com/about
        - https://docs.xano.com/what-xano-includes
        - https://docs.xano.com/what-xano-includes/instance
        - https://docs.xano.com/what-xano-includes/workspace
        - https://docs.xano.com/database/triggers
        - https://docs.xano.com/fundamentals/the-development-life-cycle    

        WeWeb Documentation
        - https://docs.weweb.io/start-here/welcome.html
        - https://docs.weweb.io/start-here/frequently-asked-questions.html
        - https://docs.weweb.io/editor/intro-to-the-editor.html
        - https://docs.weweb.io/editor/intro-to-html-css.html
        - https://docs.weweb.io/editor/how-to-use-the-add-panel.html
        - https://docs.weweb.io/editor/logs.html
        - https://docs.weweb.io/editor/copilot/import-figma-designs.html
        - https://docs.weweb.io/editor/app-settings/app-settings.html
        - https://docs.weweb.io/editor/app-settings/pwa.html
"""
    )

    st.header("Example Question")
    st.markdown(
        """
        Note: When asking a question, always add the word **xeno** or **weweb** so that the agent can easily find an accurate answer.

        - What is PWA? and how enabling mobile app features in Weweb?
        - How installing a PWA on a phone in WeWeb?
        - Will the Marketplace have templates that I can use to start my backend with?
        - Can I scale my backend with Xano?
        """
    )

st.title("AI Agent Assistance")

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

for message in st.session_state.messages:
    role = message.get("role", "assistant")
    with st.chat_message(role):
        if "output" in message:
            st.markdown(message["output"])


if prompt := st.chat_input("What do you want to know?"):
    st.chat_message("user").markdown(prompt)
    st.session_state.messages.append({"role": "user", "output": prompt})

    with st.spinner("Searching for an answer..."):
        output_text = get_response(prompt)
        print("Output", output_text)

    # Display assistant response and SQL query
    st.chat_message("assistant").markdown(output_text)  # Kenapa ini tidak muncul di UI?
    

    # Append assistant response to session state
    st.session_state.messages.append(
        {
            "role": "assistant",
            "output": output_text,
        }
    )