Update app.py

app.py

@@ -102,31 +102,41 @@ precomputed_faq_answers = get_precomputed_answers()
 # Streamlit UI
 st.title('Mining Minerals Expert RAG App')
 
+# Sidebar with FAQ Section
+st.sidebar.subheader("Frequently Asked Questions")
+
+# Initialize session state to store the selected question and answer
+if "faq_answer" not in st.session_state:
+    st.session_state.faq_answer = ""
+if "custom_answer" not in st.session_state:
+    st.session_state.custom_answer = ""
+
 # Custom Question Section
 st.subheader("Ask Your Own Question")
 question = st.text_input('Write your question below:')
 if st.button('Enter'):
     if question:
         answer = get_answer(question)
-        st.write(f"**Answer:** {answer}")
+        st.session_state.custom_answer = f"**Answer:** {answer}"
+        st.session_state.faq_answer = ""  # Clear FAQ answer when a custom question is asked
     else:
         st.write("Please enter a question.")
-        
-# Sidebar with FAQ Section
-st.sidebar.subheader("Frequently Asked Questions")
-# Initialize session state to store the selected question and answer
-if "faq_answer" not in st.session_state:
-    st.session_state.faq_answer = ""
 
+# Display the answer to the custom question in the main window
+if st.session_state.custom_answer:
+    st.write(st.session_state.custom_answer)
+    
 # Show initial 10 questions
 faq_expander = st.sidebar.expander("FAQs")
 with faq_expander:
     for i, faq in enumerate(faq_questions):
         if st.sidebar.button(f"Q{i+1}: {faq}", key=f"faq{i+1}"):
             st.session_state.faq_answer = f"**Answer to Q{i+1}:** {precomputed_faq_answers[faq]}"
+            st.session_state.custom_answer = ""  # Clear custom answer when FAQ is clicked
 
 # Display the answer to the selected FAQ in the main window
 if st.session_state.faq_answer:
     st.write(st.session_state.faq_answer)
 
 
+