Update app.py

app.py

@@ -105,14 +105,6 @@ def register_callback():
     except Exception as e:
         st.error(f"Registration failed: {e}")
 
-# Callback for logout
-def logout_callback():
-    st.session_state.logged_in = False
-    st.session_state.current_user = None
-    st.session_state.display_name = None
-    st.session_state.selected_image = None
-    st.info("Logged out successfully!")
-
 # Callback for login
 def login_callback():
     login_identifier = st.session_state.login_identifier
@@ -168,6 +160,14 @@ def login_callback():
     except Exception as e:
         st.error(f"Login failed: {e}")
 
+# Callback for logout
+def logout_callback():
+    st.session_state.logged_in = False
+    st.session_state.current_user = None
+    st.session_state.display_name = None
+    st.session_state.selected_image = None
+    st.info("Logged out successfully!")
+
 # Function to get image from url
 def get_image_from_url(url):
     """
@@ -266,7 +266,7 @@ def upload_image_to_storage(image, user_id, is_thumbnail = False):
             if is_thumbnail:
                  file_path = f"user_images/{user_id}/thumbnails/{image_id}.png" # path for thumbnail
             else:
-                 file_path = f"images/{user_id}/{image_id}.png" # path for high resolution images
+                 file_path = f"user_images/{user_id}/{image_id}.png" # path for high resolution images
             blob = bucket.blob(file_path)
 
             # Convert PIL Image to BytesIO object
@@ -372,7 +372,7 @@ def main_app():
                                 st.write(f"**Realism:** {realism}")
                                 download_path = download_image(image_url)
                                 if download_path:
-                                  st.download_button(label="Download Image", data = open(download_path, "rb"), file_name = f"image.png")
+                                  st.download_button(label="Download Image", data = open(download_path, "rb"), file_name = f"image.png", key=f"download_high_res_{uuid.uuid4()}")
                           else:
                                st.error("Failed to upload image to cloud storage.")
                       else:
@@ -449,7 +449,7 @@ def main_app():
            st.write(f"**Realism:** {st.session_state.selected_image['realism']}")
            download_path = download_image(st.session_state.selected_image['image_url'])
            if download_path:
-              st.download_button(label="Download Image", data = open(download_path, "rb"), file_name = f"image.png")
+              st.download_button(label="Download Image", data = open(download_path, "rb"), file_name = f"image.png", key=f"download_overlay_{uuid.uuid4()}")
 
         if st.button("Close"):
           st.session_state.selected_image = None # close the modal when "close" is clicked