Update app.py

app.py

@@ -64,8 +64,19 @@ 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):
+    reason = st.text_input("{text}")
+    if st.button("Submit"):
+        resturn text
+    if st.button("Cancel"):
+        return ""
+
 def edit_message(position):
     st.toast("try to edit  message no: " + str(position))
+    text = edit_text(st.session_state.messages[position])
+    st.toast("result: " + text)
+    
 
 def remove_message(position):
     st.toast("try to remove message no: " + str(position-1) + " and "+ str(position))