|
|
|
|
|
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) |
|
|
|
|
|
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 |
|
|
) |
|
|
|
|
|
|
|
|
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]) |
|
|
|
