Update app.py

app.py

@@ -223,24 +223,20 @@ display_conversation()
 
 # --- Chat Input Form ---
 with st.form("chat_form", clear_on_submit=True):
-    # Use st.text_area for multi-line text input with text wrapping
     user_input = st.text_area("Enter your message:", height=150)
     submitted = st.form_submit_button("Send")
     if submitted and user_input:
         st.session_state.chat_history.append(("user", user_input))
-        st.experimental_rerun()
+        # No need to force a rerun—Streamlit re-runs automatically on widget interaction.
+
 
 # --- Generate Assistant Response ---
 if st.session_state.chat_history and st.session_state.chat_history[-1][0] == "user":
     inputs = {"messages": st.session_state.chat_history}
-
-    # Placeholder for real-time streaming of the response.
     response_placeholder = st.empty()
     assistant_message = ""
 
-    # Stream the agent's response chunk-by-chunk.
     for s in graph.stream(inputs, stream_mode="values"):
-        # Extract the last message from the messages list.
         message = s["messages"][-1]
         if isinstance(message, tuple):
             role, text = message
@@ -249,6 +245,5 @@ if st.session_state.chat_history and st.session_state.chat_history[-1][0] == "us
         assistant_message += text
         response_placeholder.markdown(f"**Assistant:** {assistant_message}")
 
-    # Append the complete assistant response to the chat history.
     st.session_state.chat_history.append(("assistant", assistant_message))
-    st.experimental_rerun()
+    # No rerun needed here either.