Update app.py

app.py

@@ -22,6 +22,9 @@ if 'form_key' not in st.session_state:
 # New variable for debouncing: whether processing is in progress
 if 'is_processing' not in st.session_state:
     st.session_state.is_processing = False
+# Store the answer so that it persists on screen
+if 'last_answer' not in st.session_state:
+    st.session_state.last_answer = None
 
 # THEN: Import your modules
 from rag_engine import process_query, load_model
@@ -231,23 +234,26 @@ with col1:
 with col2:
     word_limit = st.slider("Word limit:", 50, 500, 200)
 
-# Only process the query if explicitly submitted
+# Process the query only if explicitly submitted
 if st.session_state.submit_clicked and st.session_state.last_query:
     st.session_state.submit_clicked = False
 
     with st.spinner("Processing your question..."):
         try:
             result = process_query(st.session_state.last_query, top_k=top_k, word_limit=word_limit)
-            st.subheader("Answer:")
-            st.write(result["answer_with_rag"])
-            st.subheader("Sources:")
-            for citation in result["citations"].split("\n"):
-                st.write(citation)
+            st.session_state.last_answer = result  # Store the result in session state
         except Exception as e:
-            st.error(f"Error processing query: {str(e)}")
-    # Reset debouncing after the answer has been produced
+            st.session_state.last_answer = {"answer_with_rag": f"Error processing query: {str(e)}", "citations": ""}
+    # Reset debouncing after processing
     st.session_state.is_processing = False
-    st.experimental_rerun()
+
+# Display the answer if available
+if st.session_state.last_answer is not None:
+    st.subheader("Answer:")
+    st.write(st.session_state.last_answer["answer_with_rag"])
+    st.subheader("Sources:")
+    for citation in st.session_state.last_answer["citations"].split("\n"):
+        st.write(citation)
 
 # Add helpful information
 st.markdown("---")