should-i-follow / app.py
Tuana's picture
fixing some bugs and improving the prompt
9e0a453
raw history blame
No virus
1.29 kB
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("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])