Refactor prompt enrichment logic in generate_equirect.py

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)