from json import JSONDecodeError
import logging

import streamlit as st

from utils.haystack import run_agent, start_haystack
from utils.ui import reset_results, set_initial_state, sidebar
from utils.config import OPENAI_API_KEY, SERPER_KEY

set_initial_state()

sidebar()

st.write("# 👩 What would they have posted about this?")
st.write("### This app had to be changed from searching Twitter posts to searching Mastodon posts due to API changes")
st.write("## About this App")
st.write("This app, built with [Haystack](https://github.com/deepset-ai/haystack#readme) uses an [Agent](https://docs.haystack.deepset.ai/docs/agent) (with GPT-4) with a `WebSearch` and `MastodonRetriever` tool")
st.write("The `MastodonRetriever` is set to retrieve the latest 20 mastodon posts by the given Mastodon username to construct an understanding of their style of posting")
st.write("### Instructions")
st.write("For best results, please provide a Mastodon username as it appears on twitter. E.g.: [ivepetthatdog@mastodon.social](https://mastodon.social/@ivepetthatdog) 🐶")
st.write("### Example")
st.image("./images/mastodon_example.png")

# if st.session_state.get("OPENAI_API_KEY") and st.session_state.get("SERPER_KEY"):

agent = start_haystack(openai_key=OPENAI_API_KEY, serper_key=SERPER_KEY)
# st.session_state["api_keys_configured"] = True

# Search bar
question = st.text_input("Example: If the twitter account tuanacelik were to write a post in their style about climate change, what would it be?", on_change=reset_results) 
run_pressed = st.button("Generate post")
# else:
#     st.write("Please provide your OpenAI and SerperDev Keys to start using the application")
#     st.write("If you are using a smaller screen, open the sidebar from the top left to provide your OpenAI Key 🙌")

# if st.session_state.get("api_keys_configured"):
run_query = (
    run_pressed or question != st.session_state.question
)
if run_query and question:
    reset_results()
    st.session_state.question = question
    try:
        st.session_state.result = run_agent(agent, question)
    except JSONDecodeError as je:
        st.error(
            "👓    An error occurred reading the results. Is the document store working?"
        )    
    except Exception as e:
        logging.exception(e)
        st.error("🐞    An error occurred during the request.")            
            
if st.session_state.result:
    result = st.session_state.result
    st.write(result["answers"][0].answer)