Update pages/answers.py

pages/answers.py

@@ -1,7 +1,14 @@
 import random
 import streamlit as st
 from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
+from streamlit_extras.let_it_rain import rain
 
+rain(
+    emoji="❔",
+    font_size=54,
+    falling_speed=5,
+    animation_length="infinite",
+)
 
 model_name = "deepset/roberta-base-squad2"
 model = AutoModelForQuestionAnswering.from_pretrained(model_name)
@@ -50,33 +57,34 @@ def get_answer(context, question):
     answer = res['answer']
     return answer
 
+
 def main():
     st.title("Question Answering App :robot_face:")
-    st.divider()
-    st.sidebar.markdown("### **Enter the context and question:**")
+    st.sidebar.markdown("### **Choose a Pair:**")
+
+    pair_names = [f"Pair {i+1}" for i in range(len(pairs))]
+    selected_pair_index = st.sidebar.selectbox("Select Pair", pair_names, key='pair_selection')
 
-    context_placeholder = "Enter the context here..."
-    question_placeholder = "Enter the question here..."
-    
-    context = st.sidebar.text_area("Context", value=context_placeholder)
-    question = st.sidebar.text_input("Question", value=question_placeholder)
+    context_placeholder = pairs[selected_pair_index]['context']
+    question_placeholder = pairs[selected_pair_index]['question']
 
-    if st.sidebar.button("Random Example"):
+    context = st.text_area("**:blue[Context]**", context_placeholder)
+    question = st.text_input("**:blue[Question]**", question_placeholder)
+
+    if st.button(":blue[**Random Example**]"):
         pair = random.choice(pairs)
         context = pair['context']
         question = pair['question']
+        
+        st.text_area("**:blue[Context]**", context)
+        st.text_input("**:blue[Question]**", question)
 
-    if st.button("Get Answer"):
+    if st.button(":blue[**Get Answer**]"):
         if context.strip() == "" or question.strip() == "":
             st.warning("Please enter the context and question.")
         else:
             answer = get_answer(context, question)
             st.success(f"Answer: {answer}")
 
-
 if __name__ == "__main__":
     main()
-
-
-
-