Update app.py

app.py

@@ -122,9 +122,10 @@ form small {
 
 # Function to handle query selection
 def set_query(query):
-    st.session_state.last_query = query
-    st.session_state.submit_clicked = True
-    st.rerun()
+    if not st.session_state.processing:
+        st.session_state.last_query = query
+        st.session_state.submit_clicked = True
+        st.rerun()
 
 # Function to group questions into rows based on length
 def group_buttons(questions, max_chars_per_row=100):
@@ -199,12 +200,12 @@ 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.processing):
                 set_query(q)
 
 # Function to handle form submission
 def handle_form_submit():
-    if st.session_state.query_input and st.session_state.query_input.strip():
+    if st.session_state.query_input and st.session_state.query_input.strip() and not st.session_state.processing:
         st.session_state.last_query = st.session_state.query_input.strip()
         st.session_state.submit_clicked = True
         # Increment the form key to force a reset
@@ -214,7 +215,7 @@ def handle_form_submit():
 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)
+    submit_button = st.form_submit_button("Get Answer", on_click=handle_form_submit, disabled=st.session_state.processing)
 
 # Display the current question if there is one
 if st.session_state.last_query: