Update app.py

app.py

@@ -74,10 +74,11 @@ def edit_text(text):
 
 def edit_message(position):
     st.toast("try to edit  message no: " + str(position-1))
-    message = st.session_state.messages[position-1]
-    # text = edit_text(message["content"])
-    text = message["content"]
+    text = edit_text(st.session_state.messages[position-1]["content"])
+    #text = message["content"]
     st.toast("result: " + text)
+    st.session_state.messages[position-1]["content"] = text
+    st.rerun()
     
 
 def remove_message(position):