app.py

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("# Ask anything about Milvus")

# Search bar
question = st.text_input("Ask a question", max_chars=100, on_change=reset_results)

run_pressed = st.button("Run")

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

# Get results for query
if run_query and question:
    reset_results()
    st.session_state.question = question
    with st.spinner("🔎    Running your pipeline"):
        try:
            st.session_state.results = query(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.results:
    answers = st.session_state.results
    for count, answer in enumerate(answers):
        if answer:
            st.write(
                markdown(answer.answer),
                unsafe_allow_html=True,
            )
        else:
            st.info(
                "🤔    Haystack is unsure whether any of the documents contain an answer to your question. Try to reformulate it!"
            )