You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
imaginAIry/imaginairy/cli/edit_demo.py

46 lines
1.1 KiB
Python

"""Command for applying preset image edits"""
import click
@click.command("edit-demo")
@click.argument("image_paths", metavar="PATH|URL", required=True, nargs=-1)
@click.option(
"--outdir",
default="./outputs",
show_default=True,
type=click.Path(),
help="Where to write results to.",
)
@click.option(
"-h",
"--height",
default=512,
show_default=True,
type=int,
help="Image height. Should be multiple of 8.",
)
@click.option(
"-w",
"--width",
default=512,
show_default=True,
type=int,
help="Image width. Should be multiple of 8.",
)
def edit_demo_cmd(image_paths, outdir, height, width):
"""Make some fun pre-set edits to input photos."""
from imaginairy.cli.shared import imaginairy_click_context
from imaginairy.utils.surprise_me import create_surprise_me_images
with imaginairy_click_context():
for image_path in image_paths:
create_surprise_me_images(
image_path,
outdir=outdir,
make_gif=True,
width=width,
height=height,
seed=1,
)