Update app.py

app.py

@@ -238,17 +238,27 @@ with st.form("chat_form", clear_on_submit=True):
 # --- 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}
+
+    # Create a container for streaming output
     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
         else:
             text = message.content
+
+        # Append new text and update the markdown display.
         assistant_message += text
-        response_placeholder.markdown(f"**Assistant:** {assistant_message}")
 
+        # Update the placeholder with the current assistant message.
+        # Using unsafe_allow_html=True ensures that any markdown (or embedded HTML) is rendered.
+        response_placeholder.markdown(assistant_message, unsafe_allow_html=True)
+
+    # Once complete, add the full assistant response to the chat history.
     st.session_state.chat_history.append(("assistant", assistant_message))
-    # No rerun needed here either.
+