Integrate Stable Diffusion with the Imaginate layer (#784)

* Add AI Artist layer

* WIP add a button to download the rendered folder under an AI Artist layer

* Successfully download the correct image

* Break out image downloading JS into helper function

* Change file download from using data URLs to blob URLs

* WIP rasterize to blob

* Remove dimensions from AI Artist layer

* Successfully draw rasterized image on layer after calculation

* Working txt2img generation based on user prompt

* Add img2img and the main parameters

* Fix ability to rasterize multi-depth documents with blob URL images by switching them to base64

* Fix test

* Rasterize with artboard background color

* Allow aspect ratio stretch of AI Artist images

* Add automatic resolution choosing

* Add a terminate button, and make the lifecycle more robust

* Add negative prompt

* Add range bounds for parameter inputs

* Add seed

* Add tiling and restore faces

* Add server status check, server hostname customization, and resizing layer to fit AI Artist resolution

* Fix background color of infinite canvas rasterization

* Escape prompt text sent in the JSON

* Revoke blob URLs when cleared/replaced to reduce memory leak

* Fix welcome screen logo color

* Add PreferencesMessageHandler

* Add persistent storage of preferences

* Fix crash introduced in previous commit when moving mouse on page load

* Add tooltips to the AI Artist layer properties

* Integrate AI Artist tool into the raster section of the tool shelf

* Add a refresh button to the connection status

* Fix crash when generating and switching to a different document tab

* Add persistent image storage to AI Artist layers and fix duplication bugs

* Add a generate with random seed button

* Simplify and standardize message names

* Majorly improve robustness of networking code

* Fix race condition causing default server hostname to show disconnected when app loads with AI Artist layer selected (probably, not confirmed fixed)

* Clean up messages and function calls by changing arguments into structs

* Update API to more recent server commit

* Add support for picking the sampling method

* Add machinery for filtering selected layers with type

* Replace placeholder button icons

* Improve the random icon by tilting the dice

* Use selected_layers() instead of repeating that code

* Fix borrow error

* Change message flow in progress towards fixing #797

* Allow loading image on non-active document (fixes #797)

* Reduce code duplication with rasterization

* Add AI Artist tool and layer icons, and remove ugly node layer icon style

* Rename "AI Artist" codename to "Imaginate" feature name

Co-authored-by: otdavies <oliver@psyfer.io>
Co-authored-by: 0hypercube <0hypercube@gmail.com>

editor/src/dispatcher.rs

@@ -22,6 +22,7 @@ struct DispatcherMessageHandlers {
 	input_preprocessor_message_handler: InputPreprocessorMessageHandler,
 	layout_message_handler: LayoutMessageHandler,
 	portfolio_message_handler: PortfolioMessageHandler,
+	preferences_message_handler: PreferencesMessageHandler,
 	tool_message_handler: ToolMessageHandler,
 	workspace_message_handler: WorkspaceMessageHandler,
 }
@@ -92,14 +93,16 @@ impl Dispatcher {
 				NoOp => {}
 				#[remain::unsorted]
 				Init => {
+					// Load persistent data from the browser database
+					queue.push_back(FrontendMessage::TriggerLoadAutoSaveDocuments.into());
+					queue.push_back(FrontendMessage::TriggerLoadPreferences.into());
+
 					// Display the menu bar at the top of the window
-					let message = MenuBarMessage::SendLayout.into();
-					queue.push_back(message);
+					queue.push_back(MenuBarMessage::SendLayout.into());
 
 					// Load the default font
 					let font = Font::new(DEFAULT_FONT_FAMILY.into(), DEFAULT_FONT_STYLE.into());
-					let message = FrontendMessage::TriggerFontLoad { font, is_default: true }.into();
-					queue.push_back(message);
+					queue.push_back(FrontendMessage::TriggerFontLoad { font, is_default: true }.into());
 				}
 
 				Broadcast(message) => self.message_handlers.broadcast_message_handler.process_message(message, (), &mut queue),
@@ -107,9 +110,11 @@ impl Dispatcher {
 					self.message_handlers.debug_message_handler.process_message(message, (), &mut queue);
 				}
 				Dialog(message) => {
-					self.message_handlers
-						.dialog_message_handler
-						.process_message(message, &self.message_handlers.portfolio_message_handler, &mut queue);
+					self.message_handlers.dialog_message_handler.process_message(
+						message,
+						(&self.message_handlers.portfolio_message_handler, &self.message_handlers.preferences_message_handler),
+						&mut queue,
+					);
 				}
 				Frontend(message) => {
 					// Handle these messages immediately by returning early
@@ -145,9 +150,14 @@ impl Dispatcher {
 					self.message_handlers.layout_message_handler.process_message(message, action_input_mapping, &mut queue);
 				}
 				Portfolio(message) => {
-					self.message_handlers
-						.portfolio_message_handler
-						.process_message(message, &self.message_handlers.input_preprocessor_message_handler, &mut queue);
+					self.message_handlers.portfolio_message_handler.process_message(
+						message,
+						(&self.message_handlers.input_preprocessor_message_handler, &self.message_handlers.preferences_message_handler),
+						&mut queue,
+					);
+				}
+				Preferences(message) => {
+					self.message_handlers.preferences_message_handler.process_message(message, (), &mut queue);
 				}
 				Tool(message) => {
 					if let Some(document) = self.message_handlers.portfolio_message_handler.active_document() {
@@ -155,8 +165,9 @@ impl Dispatcher {
 							message,
 							(
 								document,
+								self.message_handlers.portfolio_message_handler.active_document_id().unwrap(),
 								&self.message_handlers.input_preprocessor_message_handler,
-								self.message_handlers.portfolio_message_handler.font_cache(),
+								&self.message_handlers.portfolio_message_handler.persistent_data,
 							),
 							&mut queue,
 						);