Update app.py

app.py

@@ -14,7 +14,7 @@ if os.path.exists("data.txt"):
 else:
     saved_data = {"messages": [], "reasons": [], "images": []}
 
-# Password logic (for simulation only - NOT secure)
+# Password logic (using session state)
 if 'password_entered' not in st.session_state:
     st.session_state.password_entered = False
 
@@ -24,9 +24,9 @@ if not st.session_state.password_entered:
 
     if password_entered == correct_password:
         st.session_state.password_entered = True
-        st.experimental_rerun()  # Rerun to hide password input
+        st.experimental_rerun()
 
-if st.session_state.password_entered:  # Show content only after correct password
+if st.session_state.password_entered:
     st.title("Happy Valentine's Day, My Love! ❤️")
 
     # --- Styling ---
@@ -34,18 +34,18 @@ if st.session_state.password_entered:  # Show content only after correct passwor
         """
         <style>
         body {
-            background-color: #f8f0e3; /* Light beige */
+            background-color: #f8f0e3;
             font-family: 'Arial', sans-serif;
         }
         .stButton>button {
-            background-color: #f08080 !important; /* Light coral */
+            background-color: #f08080 !important;
             color: white !important;
             border: none;
             padding: 10px 20px;
             border-radius: 5px;
         }
         .stTextInput, .stTextArea, .stSelectbox {
-            border: 2px solid #f08080; /* Light coral */
+            border: 2px solid #f08080;
             border-radius: 5px;
         }
         </style>
@@ -53,8 +53,28 @@ if st.session_state.password_entered:  # Show content only after correct passwor
         unsafe_allow_html=True,
     )
 
-    # --- Heart Animation (using HTML and CSS) ---
-    # ... (same heart animation code as before)
+    # --- Heart Animation ---
+    st.markdown(
+        """
+        <div style="position: fixed; top: 0; left: 0; width: 100%; height: 100%; z-index: -1;">
+            <div class="heart"></div>
+            <div class="heart" style="left: 10%; animation-delay: 2s;"></div>
+            <div class="heart" style="left: 20%; animation-delay: 1s;"></div>
+            <div class="heart" style="left: 30%; animation-delay: 3s;"></div>
+            <div class="heart" style="left: 40%; animation-delay: 0.5s;"></div>
+            <div class="heart" style="left: 50%; animation-delay: 2.5s;"></div>
+            <div class="heart" style="left: 60%; animation-delay: 1.5s;"></div>
+            <div class="heart" style="left: 70%; animation-delay: 3.5s;"></div>
+            <div class="heart" style="left: 80%; animation-delay: 0.8s;"></div>
+            <div class="heart" style="left: 90%; animation-delay: 2.2s;"></div>
+        </div>
+
+        <style>
+        .heart { /* ... (same heart animation CSS as before) ... */ }
+        </style>
+        """,
+        unsafe_allow_html=True,
+    )
 
     # Tabs
     tabs = ["Memories", "Messages", "Reasons"]
@@ -71,7 +91,6 @@ if st.session_state.password_entered:  # Show content only after correct passwor
                     image = Image.open(uploaded_file)
                     st.image(image, caption=uploaded_file.name, use_column_width=True)
 
-                    # Save image to static folder and update data.txt
                     image_path = os.path.join("static", uploaded_file.name)
                     with open(image_path, "wb") as f:
                         f.write(uploaded_file.getvalue())
@@ -83,8 +102,7 @@ if st.session_state.password_entered:  # Show content only after correct passwor
                 except Exception as e:
                     st.error(f"Error opening image {uploaded_file.name}: {e}")
 
-        # Display previously uploaded images (always)
-        if saved_data["images"]:
+        if saved_data["images"]:  # Always display previously uploaded images
             for image_name in saved_data["images"]:
                 try:
                     image_path = os.path.join("static", image_name)
@@ -96,7 +114,38 @@ if st.session_state.password_entered:  # Show content only after correct passwor
                     st.error(f"Error loading previously saved image {image_name}: {e}")
 
     elif selected_tab == "Messages":
-        # ... (same as before)
+        st.write("## Sweet Messages")
+        new_message = st.text_area("Add a new message:", height=100)
+        if st.button("Save Message"):
+            if new_message:
+                saved_data["messages"].append(new_message)
+                with open("data.txt", "w") as f:
+                    f.write(str(saved_data))
+                st.success("Message saved!")
+                st.rerun()
+
+        if saved_data["messages"]:
+            for message in saved_data["messages"]:
+                st.write(f"- {message}")
 
     elif selected_tab == "Reasons":
-        # ... (same as before)
+        st.write("## Reasons I Love You")
+
+        if saved_data["reasons"]:
+            for reason in saved_data["reasons"]:
+                st.write(f"- {reason}")
+
+        new_reason = st.text_input("Add a reason:")
+        if st.button("Save Reason"):
+            if new_reason:
+                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:  # This should never be reached now
+    st.error("Incorrect secret word. Try again.")
+
+else:
+    st.info("Enter the secret word to unlock the surprises!")