Update app.py

app.py

@@ -148,7 +148,7 @@ if st.session_state.password_entered:
             for image_name in saved_data["images"]:
                 try:
                     image_path = os.path.join("static", image_name)
-                    with open(image_path, "rb") as f:  # Open in binary read mode
+                    with open(image_path, "rb") as f:
                         image = Image.open(f)
                         st.image(image, caption=image_name, use_column_width=True)
                 except FileNotFoundError:
@@ -158,9 +158,18 @@ if st.session_state.password_entered:
 
     elif selected_tab == "Messages":
         st.write("## Sweet Messages")
+
         if saved_data["messages"]:
             for i, message in enumerate(saved_data["messages"]):
-                st.markdown(f"<div style='background-color: #ffe6f2; padding: 10px; border-radius: 5px; margin-bottom: 10px;'>{message}</div>", unsafe_allow_html=True)
+                col1, col2 = st.columns([0.8, 0.2])
+                with col1:
+                    st.markdown(f"<div style='background-color: #ffe6f2; padding: 10px; border-radius: 5px; margin-bottom: 10px;'>{message}</div>", unsafe_allow_html=True)
+                with col2:
+                    if st.button("Delete", key=f"delete_message_{i}"):
+                        del saved_data["messages"][i]
+                        with open("data.txt", "w") as f:
+                            f.write(str(saved_data))
+                        st.experimental_rerun()
 
         new_message = st.text_area("Add a new message:", height=100)
         if st.button("Save Message"):
@@ -176,7 +185,15 @@ if st.session_state.password_entered:
 
         if saved_data["reasons"]:
             for i, reason in enumerate(saved_data["reasons"]):
-                st.write(f"{i+1}. {reason}")
+                col1, col2 = st.columns([0.8, 0.2])
+                with col1:
+                    st.write(f"{i+1}. {reason}")
+                with col2:
+                    if st.button("Delete", key=f"delete_reason_{i}"):
+                        del saved_data["reasons"][i]
+                        with open("data.txt", "w") as f:
+                            f.write(str(saved_data))
+                        st.experimental_rerun()
 
         new_reason = st.text_input("Add a reason:")
         if st.button("Save Reason"):
@@ -184,11 +201,4 @@ if st.session_state.password_entered:
                 saved_data["reasons"].append(new_reason)
                 with open("data.txt", "w") as f:
                     f.write(str(saved_data))
-                st.success("Reason saved!")
-                st.rerun()
-
-elif password_entered:
-    st.error("Incorrect secret word. Try again.")
-
-else:
-    st.info("Enter the secret word to unlock the surprises!")
+                st.success("Reason saved!")