fixed buttons

app.py

@@ -2,7 +2,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
+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)
 
 st.set_page_config(
     page_title="Retrieval Augmentation with Haystack",
@@ -17,7 +18,6 @@ st.markdown("Ask a question about the collapse of the Silicon Valley Bank (SVB).
 col_1, col_2 = st.columns([4, 2], gap="small")
 with col_1:
     run_pressed, placeholder_plain_gpt, placeholder_retrieval_augmented = main_column()
-    print(f"Run value: {st.session_state.get('run', 'not found')}")
 
 with col_2:
     right_sidebar()
@@ -31,7 +31,7 @@ if st.session_state.get('query') and run_pressed:
         answers = p1.run(ip)
     placeholder_plain_gpt.markdown(answers['results'][0])
 
-    if st.session_state.get("query_type", "Retrieval Augmented") == "Retrieval Augmented":
+    if st.session_state.get("query_type", BUTTON_LOCAL_RET_AUG) == BUTTON_LOCAL_RET_AUG:
         with st.spinner(
                 'Loading Retrieval Augmented pipeline that can fetch relevant documents from local data store... '
                 '\n This may take a few mins and might also fail if OpenAI API server is down.'):

utils/constants.py

@@ -2,9 +2,13 @@ QUERIES = [
     "Did SVB collapse?",
     "Why did SVB collapse?",
     "What does SVB failure mean for our economy?",
-    "Who is responsible for SVC collapse?",
+    "Who is responsible for SVB collapse?",
     "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)"
+
+
+BUTTON_LOCAL_RET_AUG = "Retrieval Augmented (Static news dataset)"
+BUTTON_WEB_RET_AUG = "Retrieval Augmented with Web Search"

utils/ui.py

@@ -1,7 +1,8 @@
 import streamlit as st
 from PIL import Image
 
-from .constants import QUERIES, PLAIN_GPT_ANS, GPT_WEB_RET_AUG_ANS, GPT_LOCAL_RET_AUG_ANS
+from .constants import (QUERIES, PLAIN_GPT_ANS, GPT_WEB_RET_AUG_ANS, GPT_LOCAL_RET_AUG_ANS,
+                        BUTTON_LOCAL_RET_AUG, BUTTON_WEB_RET_AUG)
 
 
 def set_question():
@@ -27,6 +28,7 @@ def set_q4():
 def set_q5():
     st.session_state['query'] = QUERIES[4]
 
+
 def main_column():
     placeholder = st.empty()
     with placeholder:
@@ -40,7 +42,7 @@ def main_column():
             run_pressed = st.button("Run", key="run")
 
     st.write(" ")
-    st.radio("Answer Type:", ("Retrieval Augmented (Static news dataset)", "Retrieval Augmented with Web Search"), key="query_type")
+    st.radio("Answer Type:", (BUTTON_LOCAL_RET_AUG, BUTTON_WEB_RET_AUG), key="query_type")
 
     # st.sidebar.selectbox(
     #      "Example Questions:",
@@ -62,16 +64,10 @@ def main_column():
 
 def right_sidebar():
     st.markdown("<h5> Example questions </h5>", unsafe_allow_html=True)
-    # c1, c2, c3, c4, c5 = st.columns(5)
-    # with c1:
     st.button(QUERIES[0], on_click=set_q1)
-    # with c2:
     st.button(QUERIES[1], on_click=set_q2)
-    # with c3:
     st.button(QUERIES[2], on_click=set_q3)
-    # with c4:
     st.button(QUERIES[3], on_click=set_q4)
-    # with c5:
     st.button(QUERIES[4], on_click=set_q5)