Update app.py

app.py

@@ -68,7 +68,7 @@ if not st.session_state.initialized:
         st.session_state.init_time = time.time()
         init_message.success("System initialized successfully!")
         time.sleep(0.1)
-        st.experimental_rerun()
+        st.rerun()  # Ensure the UI updates correctly
     except Exception as e:
         init_message.error(f"Error initializing: {str(e)}")
 
@@ -84,6 +84,7 @@ def handle_form_submit():
         st.session_state.last_query = st.session_state.query_input
         st.session_state.submit_clicked = True
         st.session_state.query_input = ""
+        st.rerun()  # Forces UI to refresh, ensuring input box turns green
 
 # Create a form for handling the user input
 with st.form(key="query_form"):