Update app.py

app.py

@@ -97,9 +97,9 @@ pos = 0
 for message in st.session_state.messages:
     pos=+1
     with st.chat_message(message["role"]):
-        cols = st.columns([9,1])
-        cols[1].markdown(message["content"])
-        cols[2].button("remove", key=button_remove_message, args=[pos], on_click=remove_message)
+        col1, col2 = st.columns([9,1])
+        col1.markdown(message["content"])
+        col2.button("remove", key=button_remove_message, args=[pos], on_click=remove_message)
 
 
 if "remove" not in st.session_state: