Update app.py

app.py

@@ -347,57 +347,60 @@ def main_app():
         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 create thumbnail")
+                if isinstance(image_result, tuple) and len(image_result) == 2:
+                    image, image_url = image_result
+                    if isinstance(image, Image.Image):
+                      # 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 create thumbnail")
+                      else:
+                          st.error("Failed to upload image to cloud storage.")
                     else:
-                        st.error("Failed to upload image to cloud storage.")
+                      st.error(f"Image generation failed: {image}")
 
                 else:
                     st.error(f"Image generation failed: {image_result}")