Insert AI-generated blocks into the page

Telegram/SourceFiles/boxes/create_ai_box.cpp

@@ -423,6 +423,7 @@ void CreateAiBox(not_null<Ui::GenericBox*> box, CreateAiBoxArgs &&args) {
 					if (state->applyToPage) {
 						state->applyToPage(state->page);
 					}
+					box->closeBox();
 				});
 			pill->setFullRadius(true);
 

Telegram/SourceFiles/iv/editor/iv_editor_box.cpp

@@ -1553,9 +1553,15 @@ void WindowHost::Impl::setupWindow(ShowWindowDescriptor &&descriptor) {
 					u"rich_message"_q);
 				return;
 			}
+			const auto editor = _editor;
 			ShowCreateAiBox(_show, {
 				.session = session,
-				.applyToPage = [](std::shared_ptr<const RichPage>) {},
+				.applyToPage = [editor](std::shared_ptr<const RichPage> page) {
+					if (!editor || !page || page->blocks.empty()) {
+						return;
+					}
+					editor->insertPreparedBlocks(page->blocks);
+				},
 			});
 		});
 		addBottomAiStar(button, session);