Update app.py

app.py

@@ -34,10 +34,11 @@ if password_entered == correct_password:
             for uploaded_file in uploaded_files:
                 try:
                     image = Image.open(uploaded_file)
-                    st.image(image, caption=uploaded_file.name, use_column_width=True)
+                    st.image(image, caption=uploaded_file.name, use_container_width=True)
 
                     # Save image to static folder and update data.txt
-                    with open(os.path.join("static", uploaded_file.name), "wb") as f:
+                    image_path = os.path.join("static", uploaded_file.name)
+                    with open(image_path, "wb") as f:
                         f.write(uploaded_file.getvalue())
 
                     saved_data["images"].append(uploaded_file.name)
@@ -46,11 +47,12 @@ if password_entered == correct_password:
 
                 except Exception as e:
                     st.error(f"Error opening image {uploaded_file.name}: {e}")
+
         elif saved_data["images"]:  # Show previously uploaded images
             for image_name in saved_data["images"]:
                 try:
-                    image_path = os.path.join("static", image_name)
-                    image = Image.open(image_path)
+                    image_path = os.path.join("static", image_name)  # Correct path!
+                    image = Image.open(image_path)  # Open from correct path
                     st.image(image, caption=image_name, use_column_width=True)
                 except FileNotFoundError:
                     st.warning(f"Could not load previously saved image: {image_name}. It might have been deleted.")
@@ -66,7 +68,7 @@ if password_entered == correct_password:
                 with open("data.txt", "w") as f:
                     f.write(str(saved_data))
                 st.success("Message saved!")
-                st.rerun()  # Corrected: st.rerun()
+                st.rerun()
 
         if saved_data["messages"]:  # Display previously saved messages
             for message in saved_data["messages"]:
@@ -75,7 +77,6 @@ if password_entered == correct_password:
     elif selected_tab == "Reasons":
         st.write("## Reasons I Love You")
 
-        # Display existing reasons first
         if saved_data["reasons"]:
             for reason in saved_data["reasons"]:
                 st.write(f"- {reason}")
@@ -87,7 +88,7 @@ if password_entered == correct_password:
                 with open("data.txt", "w") as f:
                     f.write(str(saved_data))
                 st.success("Reason saved!")
-                st.rerun()  # Corrected: st.rerun()
+                st.rerun()
 
 elif password_entered:
     st.error("Incorrect secret word. Try again.")