v1 ready

app.py

@@ -1,9 +1,8 @@
 import streamlit as st
 from utils.backend import (get_plain_pipeline, get_retrieval_augmented_pipeline,
                            get_web_retrieval_augmented_pipeline)
-from utils.ui import set_q1, set_q2, set_q3, set_q4, set_q5, left_sidebar, right_sidebar, main_column
-from utils.constants import (QUERIES, PLAIN_GPT_ANS, GPT_WEB_RET_AUG_ANS, GPT_LOCAL_RET_AUG_ANS,
-                             BUTTON_LOCAL_RET_AUG, BUTTON_WEB_RET_AUG)
+from utils.ui import left_sidebar, right_sidebar, main_column
+from utils.constants import BUTTON_LOCAL_RET_AUG
 
 st.set_page_config(
     page_title="Retrieval Augmentation with Haystack",
@@ -13,7 +12,7 @@ left_sidebar()
 
 st.markdown("<center> <h2> Reduce Hallucinations with Retrieval Augmentation </h2> </center>", unsafe_allow_html=True)
 
-st.markdown("Ask a question about the collapse of the Silicon Valley Bank (SVB).", unsafe_allow_html=True)
+st.markdown("<center>Ask a question about the collapse of the Silicon Valley Bank (SVB).</center>", unsafe_allow_html=True)
 
 col_1, col_2 = st.columns([4, 2], gap="small")
 with col_1:

utils/constants.py

@@ -6,9 +6,9 @@ QUERIES = [
     "When did SVB collapse?"
 ]
 PLAIN_GPT_ANS = "Answer with plain GPT"
-GPT_LOCAL_RET_AUG_ANS = "Answer with Retrieval Augmented GPT (Static news dataset)"
-GPT_WEB_RET_AUG_ANS = "Answer with Retrieval Augmented GPT (Web Search)"
+GPT_LOCAL_RET_AUG_ANS = "Answer with Retrieval augmented GPT (static news dataset)"
+GPT_WEB_RET_AUG_ANS = "Answer with Retrieval augmented GPT (web search)"
 
 
-BUTTON_LOCAL_RET_AUG = "Retrieval Augmented (Static news dataset)"
-BUTTON_WEB_RET_AUG = "Retrieval Augmented with Web Search"
+BUTTON_LOCAL_RET_AUG = "Retrieval augmented (static news dataset)"
+BUTTON_WEB_RET_AUG = "Retrieval augmented with web search"

utils/ui.py

@@ -49,16 +49,16 @@ def main_column():
     #      QUERIES,
     #      key='q_drop_down', on_change=set_question)
 
-    st.markdown(f"<h5> {PLAIN_GPT_ANS} </h5>", unsafe_allow_html=True)
+    st.markdown(f"<h5>{PLAIN_GPT_ANS}</h5>", unsafe_allow_html=True)
     placeholder_plain_gpt = st.empty()
-    placeholder_plain_gpt.text_area(f" ", placeholder="The answer will appear here.",
+    placeholder_plain_gpt.text_area(f" ", placeholder="The answer will appear here.", disabled=True,
                                     key=PLAIN_GPT_ANS, height=1, label_visibility='collapsed')
     if st.session_state.get("query_type", BUTTON_LOCAL_RET_AUG) == BUTTON_LOCAL_RET_AUG:
-        st.markdown(f"<h5> {GPT_LOCAL_RET_AUG_ANS} </h5>", unsafe_allow_html=True)
+        st.markdown(f"<h5>{GPT_LOCAL_RET_AUG_ANS}</h5>", unsafe_allow_html=True)
     else:
-        st.markdown(f"<h5>{GPT_WEB_RET_AUG_ANS} </h5>", unsafe_allow_html=True)
+        st.markdown(f"<h5>{GPT_WEB_RET_AUG_ANS}</h5>", unsafe_allow_html=True)
     placeholder_retrieval_augmented = st.empty()
-    placeholder_retrieval_augmented.text_area(f" ", placeholder="The answer will appear here.",
+    placeholder_retrieval_augmented.text_area(f" ", placeholder="The answer will appear here.", disabled=True,
                                               key=GPT_LOCAL_RET_AUG_ANS, height=1, label_visibility='collapsed')
 
     return run_pressed, placeholder_plain_gpt, placeholder_retrieval_augmented
@@ -66,11 +66,11 @@ def main_column():
 
 def right_sidebar():
     st.markdown("<h5> Example questions </h5>", unsafe_allow_html=True)
-    st.button(QUERIES[0], on_click=set_q1)
-    st.button(QUERIES[1], on_click=set_q2)
-    st.button(QUERIES[2], on_click=set_q3)
-    st.button(QUERIES[3], on_click=set_q4)
-    st.button(QUERIES[4], on_click=set_q5)
+    st.button(QUERIES[0], on_click=set_q1, use_container_width=True)
+    st.button(QUERIES[1], on_click=set_q2, use_container_width=True)
+    st.button(QUERIES[2], on_click=set_q3, use_container_width=True)
+    st.button(QUERIES[3], on_click=set_q4, use_container_width=True)
+    st.button(QUERIES[4], on_click=set_q5, use_container_width=True)
 
 
 def left_sidebar():