Update app.py

app.py

@@ -15,10 +15,8 @@ if 'last_query' not in st.session_state:
     st.session_state.last_query = ""
 if 'submit_clicked' not in st.session_state:
     st.session_state.submit_clicked = False
-if 'success_shown' not in st.session_state:
-    st.session_state.success_shown = False
-if 'init_timestamp' not in st.session_state:
-    st.session_state.init_timestamp = None
+if 'init_time' not in st.session_state:
+    st.session_state.init_time = None
 
 # THEN: Import your modules
 from rag_engine import process_query, load_model
@@ -27,7 +25,7 @@ from utils import setup_all_auth
 # Create a placeholder for our initialization message
 init_message = st.empty()
 
-# Custom styling without any JavaScript (we'll use pure HTML)
+# Custom styling (pure CSS)
 st.markdown("""
 <style>
 .main-title {
@@ -63,6 +61,13 @@ div[data-baseweb="input"] > div {
 <div class="main-title">Indian Spiritual Texts Q&A</div>
 """, unsafe_allow_html=True)
 
+# Function to process query (only on explicit submission)
+def process_input():
+    if st.session_state.query_input:  # Only process if input is not empty
+        st.session_state.last_query = st.session_state.query_input
+        st.session_state.query_input = ""  # Clear the input field
+        st.session_state.submit_clicked = True
+
 # Handle initialization and success message timing
 if not st.session_state.initialized:
     # Show loading message
@@ -72,39 +77,45 @@ if not st.session_state.initialized:
         # Setup authentication
         setup_all_auth()
         
-        # Load the model (this will store it in st.session_state)
+        # Load the model
         load_model()
         
-        # Mark as initialized and set timestamp
+        # Mark as initialized and record time
         st.session_state.initialized = True
-        st.session_state.success_shown = True
-        st.session_state.init_timestamp = time.time()
+        st.session_state.init_time = time.time()
         
         # Show success message
         init_message.success("System initialized successfully!")
+        
+        # Force rerun to start the timer for removing the message
+        time.sleep(0.1)  # Small delay to ensure message appears
+        st.experimental_rerun()
+        
     except Exception as e:
         init_message.error(f"Error initializing: {str(e)}")
-else:
-    # Check if we need to hide the success message
-    if st.session_state.success_shown and st.session_state.init_timestamp is not None:
-        time_elapsed = time.time() - st.session_state.init_timestamp
-        if time_elapsed >= 2.0:
-            init_message.empty()
-            st.session_state.success_shown = False
-
-# Function to process when enter is pressed or button is clicked
-def process_input():
-    st.session_state.last_query = st.session_state.query_input
-    st.session_state.query_input = ""
-    st.session_state.submit_clicked = True
+# Check if we need to hide the success message (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:
+        init_message.empty()
+        st.session_state.init_time = None  # Reset timer after clearing
 
-# Query input with callback for Enter key
+# Query input with callback ONLY for Enter key
 query = st.text_input(
     "Ask your question:",
     key="query_input",
-    on_change=process_input
+    on_change=None  # Don't trigger on any change, only on enter
 )
 
+# Only process enter key press here
+if query and query != st.session_state.get('previous_query', ''):
+    st.session_state.previous_query = query
+    st.session_state.last_query = query
+    st.session_state.query_input = ""
+    st.session_state.submit_clicked = True
+    # Force refresh to update UI immediately
+    st.experimental_rerun()
+
 # Display the current question if there is one
 if st.session_state.last_query:
     st.markdown("### Current Question:")
@@ -117,25 +128,31 @@ with col1:
 with col2:
     word_limit = st.slider("Word limit:", 50, 500, 200)
 
-# Process button
-if st.button("Get Answer") or st.session_state.submit_clicked:
-    if st.session_state.last_query:
-        # Reset the submit flag
-        st.session_state.submit_clicked = False
-        
-        # Single processing spinner for the entire operation
-        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)
-            except Exception as e:
-                st.error(f"Error processing query: {str(e)}")
-    else:
-        st.warning("Please enter a question first.")
+# Process button - separate from the input field's behavior
+if st.button("Get Answer"):
+    if query:  # If there's text in the input field
+        st.session_state.last_query = query
+        st.session_state.query_input = ""
+        st.session_state.submit_clicked = True
+        # Force refresh
+        st.experimental_rerun()
+
+# Only process the query if explicitly submitted
+if st.session_state.submit_clicked and st.session_state.last_query:
+    # Reset the submit flag
+    st.session_state.submit_clicked = False
+    
+    # Process the query
+    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)
+        except Exception as e:
+            st.error(f"Error processing query: {str(e)}")
 
 # Add helpful information
 st.markdown("---")