Various improvements

submodules/InstantPageUI/Sources/InstantPageGalleryController.swift

@@ -372,6 +372,8 @@ public class InstantPageGalleryController: ViewController, StandalonePresentable
                 strongSelf.replaceRootController(controller, ready)
             }
         }, editMedia: { _ in
+        }, controller: { [weak self] in
+            return self
         })
         self.displayNode = GalleryControllerNode(controllerInteraction: controllerInteraction)
         self.displayNodeDidLoad()