Update app.py

app.py

@@ -19,6 +19,9 @@ if 'init_time' not in st.session_state:
     st.session_state.init_time = None
 if 'form_key' not in st.session_state:
     st.session_state.form_key = 0  # This will help us reset the form
+# New variable for debouncing: whether processing is in progress
+if 'is_processing' not in st.session_state:
+    st.session_state.is_processing = False
 
 # THEN: Import your modules
 from rag_engine import process_query, load_model
@@ -112,11 +115,15 @@ div.stInfo {
 <div class="main-title">Spirituality Q&A</div>
 """, unsafe_allow_html=True)
 
-# Function to handle query selection
+# Function to handle query selection from the common questions buttons
 def set_query(query):
+    # If already processing, ignore further input
+    if st.session_state.is_processing:
+        return
     st.session_state.last_query = query
     st.session_state.submit_clicked = True
-    st.rerun()
+    st.session_state.is_processing = True
+    st.experimental_rerun()
 
 # Function to group questions into rows based on length
 def group_buttons(questions, max_chars_per_row=100):
@@ -159,7 +166,6 @@ if not st.session_state.initialized:
         init_message.error(f"Error initializing: {str(e)}")
 
 # Handle timing of success message disappearance
-# This will run on each rerender after initialization
 elif st.session_state.init_time is not None:
     elapsed_time = time.time() - st.session_state.init_time
     if elapsed_time >= 2.0:
@@ -186,27 +192,32 @@ st.markdown("### Common questions to try:")
 # Group questions into rows
 question_rows = group_buttons(common_questions, max_chars_per_row=80)
 
-# Create buttons for each row
+# Create buttons for each row. They are disabled if a query is processing.
 for row_idx, row in enumerate(question_rows):
     cols = st.columns(len(row))
     for i, (col, q) in enumerate(zip(cols, row)):
         with col:
-            if st.button(q, key=f"r{row_idx}_q{i}", use_container_width=True):
+            if st.button(q, key=f"r{row_idx}_q{i}", use_container_width=True, disabled=st.session_state.is_processing):
                 set_query(q)
 
 # Function to handle form submission
 def handle_form_submit():
+    # If already processing, ignore further input
+    if st.session_state.is_processing:
+        return
     if st.session_state.query_input and st.session_state.query_input.strip():
         st.session_state.last_query = st.session_state.query_input.strip()
         st.session_state.submit_clicked = True
+        st.session_state.is_processing = True
         # Increment the form key to force a reset
         st.session_state.form_key += 1
 
-# Create a form with dynamic key for resetting
+# Create a form with dynamic key for resetting.
+# The form's submit button is also disabled when processing.
 with st.form(key=f"query_form_{st.session_state.form_key}"):
     query = st.text_input("Ask your question:", key="query_input", 
-                         placeholder="Press enter to submit your question")
-    submit_button = st.form_submit_button("Get Answer", on_click=handle_form_submit)
+                         placeholder="Press enter to submit your question", disabled=st.session_state.is_processing)
+    submit_button = st.form_submit_button("Get Answer", on_click=handle_form_submit, disabled=st.session_state.is_processing)
 
 # Display the current question if there is one
 if st.session_state.last_query:
@@ -234,6 +245,8 @@ if st.session_state.submit_clicked and st.session_state.last_query:
                 st.write(citation)
         except Exception as e:
             st.error(f"Error processing query: {str(e)}")
+    # Reset debouncing after the answer has been produced
+    st.session_state.is_processing = False
 
 # Add helpful information
 st.markdown("---")