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, sidebar | |
set_initial_state() | |
sidebar() | |
st.write("# π€ What have they been tweeting about lately?") | |
search_bar, button = st.columns(2) | |
# Search bar | |
with search_bar: | |
username = st.text_input("Please provide a twitter username", on_change=reset_results) | |
with button: | |
st.write("") | |
st.write("") | |
run_pressed = st.button("Search 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]) | |