Update app.py

app.py

@@ -344,65 +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
-
-                # 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()}")
+        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 upload thumbnail to cloud storage.")
                         else:
-                            st.error("Failed to upload thumbnail to cloud storage.")
+                            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("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.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: