Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import time
 
 # FIRST: Set page config before ANY other Streamlit command
 st.set_page_config(page_title="Indian Spiritual RAG")
@@ -54,25 +55,30 @@ if not st.session_state.initialized:
 # Preload the model to avoid session state issues
 if not st.session_state.initialized:
     try:
-        with st.spinner("Hang in there! We are setting the system up for you. 😊"):
-            # Force model loading at startup to avoid session state issues
-            load_model()
-            st.session_state.initialized = True
-            st.success("System initialized successfully!")
+        init_message = st.empty()
+        init_message.info("Hang in there! We are setting the system up for you. 😊")
+        
+        # Force model loading at startup to avoid session state issues
+        load_model()
+        
+        # Keep the message for 2 seconds before replacing it
+        time.sleep(2)
+        init_message.success("System initialized successfully!")
+        st.session_state.initialized = True
     except Exception as e:
         st.error(f"Error initializing: {str(e)}")
 
 # Function to process when enter is pressed or button is clicked
 def process_input():
-    st.session_state.last_query = st.session_state.query
-    st.session_state.query = ""
+    st.session_state.last_query = st.session_state.query_input
+    st.session_state.query_input = ""
     st.session_state.submit_clicked = True
 
 # Query input with callback for Enter key
 query = st.text_input(
     "Ask your question:",
-    key="query",
-    on_change=process_input if st.session_state.query and st.session_state.query.strip() else None
+    key="query_input",
+    on_change=process_input
 )
 
 # Display the current question if there is one