Update app.py

app.py

@@ -344,47 +344,65 @@ def main_app():
     realism = st.checkbox("Realism", value=False)
 
     if st.button("Generate Image"):
-        if prompt:
-            with st.spinner("Generating Image..."):
-                image_result = generate_image(prompt, aspect_ratio, realism)
-                if isinstance(image_result, tuple) and len(image_result) == 2 and image_result[0] is not None:
-                    image, image_url = image_result
-
-                    # 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(cloud_storage_url, use_column_width=True)  # Display high-res image from URL
-                                    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.")
+    if prompt:
+        with st.spinner("Generating Image..."):
+            image_result = generate_image(prompt, aspect_ratio, realism)
+            if isinstance(image_result, tuple) and len(image_result) == 2 and image_result[0] is not None:
+                image, image_url = image_result
+
+                # Define the boundary size
+                preview_size = 400
+
+                # Get original image dimensions
+                original_width, original_height = image.size
+
+                # Calculate scaling factor to fit within the boundary
+                width_ratio = preview_size / original_width
+                height_ratio = preview_size / original_height
+
+                scaling_factor = min(width_ratio, height_ratio)
+
+                # Calculate new dimensions
+                new_width = int(original_width * scaling_factor)
+                new_height = int(original_height * scaling_factor)
+
+                # Resize the image
+                resized_image = image.resize((new_width, new_height), Image.LANCZOS)
+
+                # Upload the high-resolution image
+                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(resized_image, use_column_width=False)  # Display the resized image
+                                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 create thumbnail")
+                            st.error("Failed to upload thumbnail to cloud storage.")
                     else:
-                        st.error("Failed to upload image to cloud storage.")
-
+                        st.error("Failed to create thumbnail")
                 else:
-                    st.error(f"Image generation failed: {image_result}")
-        else:
-            st.warning("Please enter a prompt to generate an image.")
+                    st.error("Failed to upload image to cloud storage.")
 
+            else:
+                st.error(f"Image generation failed: {image_result}")
+    else:
+        st.warning("Please enter a prompt to generate an image.")
     st.header("Your Generated Images")
      # Initialize the current window, if it doesn't exist in session state
     if "current_window_start" not in st.session_state: