import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
from _lib import fal_common  # noqa: E402

a, p = fal_common.parse_args()
if not p.get("prompt"):
    print("ERROR: prompt is required")
    sys.exit(1)
fal_common.run(
    endpoint="fal-ai/ideogram/v3",
    arguments={"prompt": p["prompt"], "image_size": p.get("image_size", "square_hd")},
    outdir=a.outdir,
    extra_args=p.get("extra_args"),
    out_stem="generated",
    collect="images",
)