from annotated_text import annotation
from json import JSONDecodeError
import logging
from markdown import markdown

import streamlit as st

from utils.haystack import query
from utils.ui import reset_results, set_initial_state

set_initial_state()

st.write("# 🐤 What have they been tweeting about lately?")

search_bar, button = st.columns(2)
# Search bar
with search_bar: 
    username = st.text_input("", value=st.session_state.username, on_change=reset_results, label_visibility="collapsed")

with button: 
    run_pressed = st.button("Seach tweets")

run_query = (
    run_pressed or username != st.session_state.username
)

# Get results for query
if run_query and username:
    reset_results()
    st.session_state.username = username
    with st.spinner("🔎"):
        try:
            st.session_state.result = query(username)
        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:
    voice = st.session_state.result
    st.write(voice[0])