diff --git a/ai_diffusion/ai_diffusion.action b/ai_diffusion/ai_diffusion.action index b644e756..f137edb1 100644 --- a/ai_diffusion/ai_diffusion.action +++ b/ai_diffusion/ai_diffusion.action @@ -75,6 +75,18 @@ false + + + Create region + + + + 10000 + 0 + + false + + Switch workspace: Generate diff --git a/ai_diffusion/extension.py b/ai_diffusion/extension.py index e5cb6673..97ae0e8b 100644 --- a/ai_diffusion/extension.py +++ b/ai_diffusion/extension.py @@ -50,6 +50,7 @@ def createActions(self, window): self._create_action(window, "cancel_all", actions.cancel_all) self._create_action(window, "toggle_preview", actions.toggle_preview) self._create_action(window, "apply", actions.apply) + self._create_action(window, "create_region", actions.create_region) self._create_action( window, "switch_workspace_generation", actions.set_workspace(Workspace.generation) ) diff --git a/ai_diffusion/ui/actions.py b/ai_diffusion/ui/actions.py index 265ec217..3f23f393 100644 --- a/ai_diffusion/ui/actions.py +++ b/ai_diffusion/ui/actions.py @@ -40,6 +40,11 @@ def apply(): model.apply_result(*model.jobs.selection) +def create_region(): + if model := root.model_for_active_document(): + model.regions.create_region(group=model.workspace is not Workspace.live) + + def set_workspace(workspace: Workspace): def action(): if model := root.model_for_active_document():