Update app.py

app.py

@@ -123,3 +123,4 @@ with col3:
         for event in graph.stream(None, st.session_state.state, stream_mode="values"):
             if "messages" in event:
                 st.session_state.rewritten_text = extract_display_text(event["messages"][-1].content)
+