diff --git a/generate_equirect.py b/generate_equirect.py index 8c60a36..904f193 100644 --- a/generate_equirect.py +++ b/generate_equirect.py @@ -846,12 +846,14 @@ def main(): if args.allow_downloads: os.environ["SKYMAP_ALLOW_DOWNLOADS"] = "1" + prompt = enrich_prompt(args.prompt) + if args.postprocess_input: if not args.postprocess_output or not args.postprocess_tempdir: parser.error("--postprocess-input requires --postprocess-output and --postprocess-tempdir") try: result_path = postprocess_image( - args.prompt, + prompt, os.path.abspath(args.postprocess_input), os.path.abspath(args.postprocess_output), os.path.abspath(args.postprocess_tempdir), @@ -869,7 +871,6 @@ def main(): print(f"Generation failed: {e}") raise - prompt = enrich_prompt(args.prompt) base = sanitize_name(prompt) target = args.output or next_filename(args.output_dir, base, args.width, args.height) output_abs = os.path.abspath(target)