Update app.py

app.py

@@ -161,30 +161,25 @@ question_model_name = "facebook/dpr-question_encoder-single-nq-base"
 
 rag = RAG(file_path, device)
 
-st.title("RAG Model Query Interface")
+st.title("RAG Model Query Interface Chatbot")
 
 # 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 = []
+if 'history' not in st.session_state:
+    st.session_state.history = []
 
-question = st.text_input("Ask a question", "")
+question = st.text_input("Enter your question:")
 
 if st.button("Ask"):
     # 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()
+    # Add the question and its answer to the history
+    st.session_state.history.append({"type": "question", "content": question})
+    st.session_state.history.append({"type": "answer", "content": answer})
+
+# Display the chat history
+for item in st.session_state.history:
+    if item["type"] == "question":
+        st.write(f"🧑 You: {item['content']}")
+    else:
+        st.write(f"🤖 Bot: {item['content']}")