Update app.py

app.py

@@ -54,7 +54,7 @@ show_image = st.sidebar.checkbox("📖 Show Document Image", value=True)
 
 # ------------------ Assistant Query Function ------------------
 def query_assistant(prompt):
-    st.session_state.messages.insert(0, {"role": "user", "content": prompt})  # insert at top
+    st.session_state.messages.insert(0, {"role": "user", "content": prompt})  # Insert at top
 
     try:
         if st.session_state.thread_id is None:
@@ -116,7 +116,7 @@ with center:
         prompt = st.text_input("💡 Ask a question about the document:", key="chat_input")
         if prompt:
             query_assistant(prompt)
-            st.experimental_rerun()
+            st.rerun()  # ✅ Updated to current Streamlit standard
 
     # Show messages: latest at top
     for message in st.session_state.messages: