Update app.py

app.py

@@ -349,37 +349,37 @@ def main_app():
                 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)
@@ -398,7 +398,7 @@ def main_app():
                             st.error("Failed to create thumbnail")
                     else:
                         st.error("Failed to upload image to cloud storage.")
-        
+
                 else:
                     st.error(f"Image generation failed: {image_result}")
         else: