wissamantoun's picture
ui fixes
f34fa3a
import streamlit as st
from .qa_utils import annotate_answer
from .services import get_qa_answers
def write():
_, col1, _ = st.columns(3)
with col1:
st.image("images/is2alni_logo.png", width=200)
st.title("إسألني أي شيء")
st.markdown(
"""
<style>
p, div, input, label {
text-align: right;
}
</style>
""",
unsafe_allow_html=True,
)
st.sidebar.header("Info")
st.sidebar.image("images/AraELECTRA.png", width=150)
st.sidebar.write("Powered by [AraELECTRA](https://github.com/aub-mind/arabert)")
st.sidebar.write("\n")
n_answers = st.sidebar.slider(
"Max. number of answers", min_value=1, max_value=10, value=2, step=1
)
question = st.text_input("", value="من هو جو بايدن؟")
if "؟" not in question:
question += "؟"
run_query = st.button("أجب")
if run_query:
# https://discuss.streamlit.io/t/showing-a-gif-while-st-spinner-runs/5084
with st.spinner("... جاري البحث "):
results_dict = get_qa_answers(question)
if len(results_dict) > 0:
st.write("## :الأجابات هي")
for result in results_dict["results"][:n_answers]:
annotate_answer(result)
f"[**المصدر**](<{result['link']}>)"
else:
st.write("## 😞 ليس لدي جواب")