Update app.py

app.py

@@ -352,35 +352,35 @@ def main_app():
                 if isinstance(image_result, tuple) and len(image_result) == 2 and image_result[0] is not None:
                     image, image_url = image_result
 
-                    #create thumbnail
-                    thumbnail = create_thumbnail(image)
-
-                    if thumbnail:
-                      # Upload thumbnail to cloud storage and store url
-                      thumbnail_url = upload_image_to_storage(thumbnail, st.session_state.current_user, is_thumbnail=True)
-
-                      if thumbnail_url:
-                          # Upload image to cloud storage and store url
-                          cloud_storage_url = upload_image_to_storage(image, st.session_state.current_user, is_thumbnail=False)
-
-                          if cloud_storage_url:
-                              # Store image data in database
-                              store_image_data_in_db(st.session_state.current_user, prompt, aspect_ratio, realism, cloud_storage_url, thumbnail_url)
-                              st.success("Image stored to database successfully!")
-                              with st.container(border=True):
-                                st.image(image, use_column_width=True)  # Display high res image on top
-                                st.write(f"**Prompt:** {prompt}")
-                                st.write(f"**Aspect Ratio:** {aspect_ratio}")
-                                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", key=f"download_high_res_{uuid.uuid4()}")
-                          else:
-                               st.error("Failed to upload image to cloud storage.")
-                      else:
-                         st.error("Failed to upload thumbnail to cloud storage.")
+                    # Upload the high-resolution image first
+                    cloud_storage_url = upload_image_to_storage(image, st.session_state.current_user, is_thumbnail=False)
+
+                    if cloud_storage_url:
+                        # Create thumbnail from the high-resolution image
+                        thumbnail = create_thumbnail(image)
+
+                        if thumbnail:
+                            # Upload thumbnail to cloud storage and store url
+                            thumbnail_url = upload_image_to_storage(thumbnail, st.session_state.current_user, is_thumbnail=True)
+
+                            if thumbnail_url:
+                                # Store image data in database
+                                store_image_data_in_db(st.session_state.current_user, prompt, aspect_ratio, realism, cloud_storage_url, thumbnail_url)
+                                st.success("Image stored to database successfully!")
+                                with st.container(border=True):
+                                    st.image(image, use_column_width=True)  # Display the (now thumbnailed) image here - consider fetching original for display
+                                    st.write(f"**Prompt:** {prompt}")
+                                    st.write(f"**Aspect Ratio:** {aspect_ratio}")
+                                    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", key=f"download_high_res_{uuid.uuid4()}")
+                            else:
+                                st.error("Failed to upload thumbnail to cloud storage.")
+                        else:
+                            st.error("Failed to create thumbnail")
                     else:
-                        st.error("Failed to create thumbnail")
+                        st.error("Failed to upload image to cloud storage.")
 
                 else:
                     st.error(f"Image generation failed: {image_result}")