Update app.py

app.py

@@ -110,12 +110,6 @@ div.stInfo {
     min-width: 120px;
     margin: 0 5px;
 }
-/* Hide "Press Enter to submit form" text */
-.stTextInput label + div + div small,
-[data-testid="stForm"] small,
-form small {
-    display: none !important;
-}
 </style>
 <div class="main-title">Spirituality Q&A</div>
 """, unsafe_allow_html=True)
@@ -123,6 +117,7 @@ form small {
 # Function to handle query selection
 def set_query(query):
     if not st.session_state.processing:
+        st.session_state.processing = True
         st.session_state.last_query = query
         st.session_state.submit_clicked = True
         st.rerun()
@@ -206,6 +201,7 @@ for row_idx, row in enumerate(question_rows):
 # Function to handle form submission
 def handle_form_submit():
     if st.session_state.query_input and st.session_state.query_input.strip() and not st.session_state.processing:
+        st.session_state.processing = True
         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
@@ -215,7 +211,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, disabled=st.session_state.processing)
+    submit_button = st.form_submit_button("Get Answer", on_click=handle_form_submit)
 
 # Display the current question if there is one
 if st.session_state.last_query:
@@ -229,34 +225,22 @@ with col1:
 with col2:
     word_limit = st.slider("Word limit:", 50, 500, 200)
 
-# Only process the query if explicitly submitted and not already processing
-if st.session_state.submit_clicked and st.session_state.last_query and not st.session_state.processing:
+# Only process the query if explicitly submitted
+if st.session_state.submit_clicked and st.session_state.last_query:
     st.session_state.submit_clicked = False
-    st.session_state.processing = True
 
     with st.spinner("Processing your question..."):
         try:
-            progress_bar = st.progress(0)
-            progress_bar.progress(25, "Generating embeddings...")
-            
-            # Call processing function with progress tracking
-            progress_bar.progress(50, "Retrieving relevant passages...")
-            
             result = process_query(st.session_state.last_query, top_k=top_k, word_limit=word_limit)
-            
-            progress_bar.progress(75, "Generating answer...")
-            time.sleep(0.5)  # Small delay for visual feedback
-            progress_bar.progress(100)
-            
             st.subheader("Answer:")
             st.write(result["answer_with_rag"])
             st.subheader("Sources:")
             for citation in result["citations"].split("\n"):
                 st.write(citation)
-                
         except Exception as e:
             st.error(f"Error processing query: {str(e)}")
         finally:
+            # Reset the processing flag when done
             st.session_state.processing = False
 
 # Add helpful information