Update app.py

app.py

@@ -173,10 +173,28 @@ rag = RAG(file_path, device)
 
 st.title("RAG Model Query Interface")
 
-# offer to ask a question and get an answer. make it so they can ask as many questions as they want
+# Initialize session state to keep track of the list of answers and questions
+if 'questions' not in st.session_state:
+    st.session_state.questions = []
+if 'answers' not in st.session_state:
+    st.session_state.answers = []
 
-question = st.text_input("Ask a question", "What is another name for self-attention?")
+question = st.text_input("Ask a question", "")
 
 if st.button("Ask"):
-  answer = rag.extractive_query(question)
-  st.write(answer)
+    # Fetch the answer for the question
+    answer = rag.extractive_query(question)
+    
+    # Store the question and its answer in session state
+    st.session_state.questions.append(question)
+    st.session_state.answers.append(answer)
+
+# Display the questions and their answers
+for q, a in zip(st.session_state.questions, st.session_state.answers):
+    st.write(f"Q: {q}\nA: {a}")
+
+# Button to ask another question which simply refreshes the page to get a new question input
+if st.button("Ask another question"):
+    st.session_state.questions = []
+    st.session_state.answers = []
+    st.experimental_rerun()