Update app.py

app.py

@@ -64,20 +64,18 @@ st.sidebar.button('Reset Chat', on_click=reset_conversation) #Reset button
 st.sidebar.write(f"You're now chatting with **{selected_model}**")
 st.sidebar.markdown("*Generated content may be inaccurate or false.*")
 
-@st.dialog("Edit Message")
-def edit_text(text):
-    returnText = st.text_area("message:", value = text)
     if st.button("Submit"):
         return returnText
     if st.button("Cancel"):
         return ""
 
+@st.dialog("Edit Message")
 def edit_message(position):
-    st.toast("try to edit  message no: " + str(position-1))
-    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
+    returnText = st.text_area("message:", value = st.session_state.messages[position-1]["content"])
+    if st.button("Submit"):
+        st.session_state.messages[position-1]["content"] = returnText
+    if st.button("Cancel"):
+        pass
     st.rerun()